diff --git a/ai_frame_image_server.py b/ai_frame_image_server.py index 872450a..23e40d7 100644 --- a/ai_frame_image_server.py +++ b/ai_frame_image_server.py @@ -39,7 +39,16 @@ from libs.comfyui import create_image def scheduled_task(): print(f"Executing scheduled task at {time.strftime('%Y-%m-%d %H:%M:%S')}") - create_image(None) + # Generate a random prompt using either OpenWebUI or OpenRouter + from libs.generic import create_prompt_with_random_model + prompt = create_prompt_with_random_model("Generate a random detailed prompt for stable diffusion.") + if prompt: + # Select a random model + import random + model = "Random Image Model" + create_image(prompt, model) + else: + print("Failed to generate a prompt for the scheduled task.") if user_config["frame"]["auto_regen"] == "True": if os.environ.get("WERKZEUG_RUN_MAIN") == "true": diff --git a/libs/comfyui.py b/libs/comfyui.py index 161a669..b274fca 100644 --- a/libs/comfyui.py +++ b/libs/comfyui.py @@ -145,8 +145,12 @@ def create_image(prompt: str | None = None, model: str = "Random Image Model") - """Generate an image with a chosen workflow (Random, FLUX*, or SDXL*).""" if prompt is None: - logging.error("No prompt provided.") - return + # Generate a random prompt using either OpenWebUI or OpenRouter + from libs.generic import create_prompt_with_random_model + prompt = create_prompt_with_random_model("Generate a random detailed prompt for stable diffusion.") + if not prompt: + logging.error("Failed to generate a prompt.") + return if not prompt: logging.error("No prompt generated.")