63 lines
2.0 KiB
Python
63 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
ComfyUI_PromptManager Gallery System Restart Script.
|
|
|
|
This script reinitializes and restarts the automatic image gallery monitoring system
|
|
for ComfyUI_PromptManager. Use this script after code modifications or when the
|
|
gallery system needs to be reloaded.
|
|
|
|
The script performs the following operations:
|
|
1. Initializes the database connection
|
|
2. Creates a PromptTracker instance for linking prompts to images
|
|
3. Sets up the ImageMonitor for real-time file system monitoring
|
|
4. Starts monitoring ComfyUI output directories
|
|
5. Reports system status and readiness
|
|
|
|
Usage:
|
|
python restart_gallery.py
|
|
|
|
This is particularly useful during development when you need to restart the
|
|
gallery monitoring system without restarting the entire ComfyUI application.
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
|
|
# Add current directory to Python path
|
|
current_dir = os.path.dirname(os.path.abspath(__file__))
|
|
sys.path.insert(0, current_dir)
|
|
|
|
try:
|
|
# Import and initialize the gallery system
|
|
from database.operations import PromptDatabase
|
|
from utils.prompt_tracker import get_prompt_tracker
|
|
from utils.image_monitor import get_image_monitor
|
|
|
|
print("[RESTART] Restarting PromptManager Gallery System...")
|
|
|
|
# Initialize components using singleton getters
|
|
db = PromptDatabase()
|
|
tracker = get_prompt_tracker(db)
|
|
monitor = get_image_monitor(db, tracker)
|
|
|
|
print("[SUCCESS] Components initialized successfully")
|
|
|
|
# Test image monitoring directories
|
|
status = monitor.get_status()
|
|
print(f"[INFO] Monitoring status: {status}")
|
|
|
|
# Start monitoring
|
|
monitor.start_monitoring()
|
|
print("[SUCCESS] Image monitoring restarted")
|
|
|
|
print("\n[READY] Gallery system restart completed!")
|
|
print("Generate some images now to test the automatic linking.")
|
|
|
|
except ImportError as e:
|
|
print(f"[ERROR] Import error: {e}")
|
|
print("Make sure you're running this from the PromptManager directory")
|
|
|
|
except Exception as e:
|
|
print(f"[ERROR] Error: {e}")
|
|
import traceback
|
|
traceback.print_exc() |