Update app.py

app.py

@@ -9,7 +9,7 @@ import traceback
 
 # --- Core Logic Imports ---
 from core.llm_services import initialize_text_llms, is_gemini_text_ready, is_hf_text_ready, generate_text_gemini, generate_text_hf
-from core.image_services import initialize_image_llms, STABILITY_API_CONFIGURED, OPENAI_DALLE_CONFIGURED, generate_image_stabilityai, generate_image_dalle, ImageGenResponse
+from core.image_services import initialize_image_llms, STABILITY_API_CONFIGURED, OPENAI_DALLE_CONFIGURED, HF_IMAGE_API_CONFIGURED, generate_image_stabilityai, generate_image_dalle, generate_image_hf_model, ImageGenResponse
 from core.story_engine import Story, Scene
 from prompts.narrative_prompts import get_narrative_system_prompt, format_narrative_user_prompt
 from prompts.image_style_prompts import STYLE_PRESETS, COMMON_NEGATIVE_PROMPTS, format_image_generation_prompt
@@ -21,9 +21,10 @@ initialize_image_llms()
 
 # --- Get API Readiness Status ---
 GEMINI_TEXT_IS_READY = is_gemini_text_ready()
-HF_TEXT_IS_READY = is_hf_text_ready()
+HF_TEXT_IS_READY = is_hf_text_ready() # For text models
 STABILITY_API_IS_READY = STABILITY_API_CONFIGURED
 OPENAI_DALLE_IS_READY = OPENAI_DALLE_CONFIGURED
+HF_IMAGE_IS_READY = HF_IMAGE_API_CONFIGURED # For image models via HF API
 
 # --- Application Configuration (Models, Defaults) ---
 TEXT_MODELS = {}
@@ -31,35 +32,38 @@ UI_DEFAULT_TEXT_MODEL_KEY = None
 if GEMINI_TEXT_IS_READY:
     TEXT_MODELS["✨ Gemini 1.5 Flash (Narrate)"] = {"id": "gemini-1.5-flash-latest", "type": "gemini"}
     TEXT_MODELS["Legacy Gemini 1.0 Pro (Narrate)"] = {"id": "gemini-1.0-pro-latest", "type": "gemini"}
-if HF_TEXT_IS_READY: # This condition was correct
-    TEXT_MODELS["Mistral 7B (Narrate)"] = {"id": "mistralai/Mistral-7B-Instruct-v0.2", "type": "hf_text"}
-    TEXT_MODELS["Gemma 2B (Narrate)"] = {"id": "google/gemma-2b-it", "type": "hf_text"}
+if HF_TEXT_IS_READY:
+    TEXT_MODELS["Mistral 7B (Narrate via HF)"] = {"id": "mistralai/Mistral-7B-Instruct-v0.2", "type": "hf_text"}
+    TEXT_MODELS["Gemma 2B (Narrate via HF)"] = {"id": "google/gemma-2b-it", "type": "hf_text"}
 
 if TEXT_MODELS:
-    # Prioritize based on readiness and preference
-    if GEMINI_TEXT_IS_READY and "✨ Gemini 1.5 Flash (Narrate)" in TEXT_MODELS:
-        UI_DEFAULT_TEXT_MODEL_KEY = "✨ Gemini 1.5 Flash (Narrate)"
-    elif HF_TEXT_IS_READY and "Mistral 7B (Narrate)" in TEXT_MODELS:
-        UI_DEFAULT_TEXT_MODEL_KEY = "Mistral 7B (Narrate)"
-    elif TEXT_MODELS: # Fallback to first available
-        UI_DEFAULT_TEXT_MODEL_KEY = list(TEXT_MODELS.keys())[0]
+    if GEMINI_TEXT_IS_READY and "✨ Gemini 1.5 Flash (Narrate)" in TEXT_MODELS: UI_DEFAULT_TEXT_MODEL_KEY = "✨ Gemini 1.5 Flash (Narrate)"
+    elif HF_TEXT_IS_READY and "Mistral 7B (Narrate via HF)" in TEXT_MODELS: UI_DEFAULT_TEXT_MODEL_KEY = "Mistral 7B (Narrate via HF)"
+    elif TEXT_MODELS: UI_DEFAULT_TEXT_MODEL_KEY = list(TEXT_MODELS.keys())[0]
 else:
     TEXT_MODELS["No Text Models Configured"] = {"id": "dummy_text_error", "type": "none"}
     UI_DEFAULT_TEXT_MODEL_KEY = "No Text Models Configured"
 
+
 IMAGE_PROVIDERS = {}
 UI_DEFAULT_IMAGE_PROVIDER_KEY = None
+# Order of preference for default: Stability, then HF, then DALL-E (simulated)
 if STABILITY_API_IS_READY: IMAGE_PROVIDERS["🎨 Stability AI (SDXL)"] = "stability_ai"
+if HF_IMAGE_IS_READY: 
+    IMAGE_PROVIDERS["🎡 HF Model (SDXL Base)"] = "hf_image_sdxl_base" # Key for SDXL via HF
+    IMAGE_PROVIDERS["🎠 HF Model (OpenJourney)"] = "hf_image_openjourney" # Key for OpenJourney via HF
 if OPENAI_DALLE_IS_READY: IMAGE_PROVIDERS["🖼️ DALL-E 3 (Sim.)"] = "dalle"
 
 if IMAGE_PROVIDERS:
-    if "🎨 Stability AI (SDXL)" in IMAGE_PROVIDERS: UI_DEFAULT_IMAGE_PROVIDER_KEY = "🎨 Stability AI (SDXL)"
-    elif "🖼️ DALL-E 3 (Sim.)" in IMAGE_PROVIDERS: UI_DEFAULT_IMAGE_PROVIDER_KEY = "🖼️ DALL-E 3 (Sim.)"
+    if STABILITY_API_IS_READY and "🎨 Stability AI (SDXL)" in IMAGE_PROVIDERS: UI_DEFAULT_IMAGE_PROVIDER_KEY = "🎨 Stability AI (SDXL)"
+    elif HF_IMAGE_IS_READY and "🎡 HF Model (SDXL Base)" in IMAGE_PROVIDERS: UI_DEFAULT_IMAGE_PROVIDER_KEY = "🎡 HF Model (SDXL Base)"
+    elif OPENAI_DALLE_IS_READY and "🖼️ DALL-E 3 (Sim.)" in IMAGE_PROVIDERS: UI_DEFAULT_IMAGE_PROVIDER_KEY = "🖼️ DALL-E 3 (Sim.)"
     else: UI_DEFAULT_IMAGE_PROVIDER_KEY = list(IMAGE_PROVIDERS.keys())[0]
 else:
     IMAGE_PROVIDERS["No Image Providers Configured"] = "none"
     UI_DEFAULT_IMAGE_PROVIDER_KEY = "No Image Providers Configured"
 
+
 # --- Gradio UI Theme and CSS ---
 omega_theme = gr.themes.Base(
     font=[gr.themes.GoogleFont("Lexend Deca"), "ui-sans-serif", "system-ui", "sans-serif"],
@@ -81,6 +85,7 @@ body, .gradio-container { background-color: #0F0F1A !important; color: #D0D0E0 !
 .output-section-header { font-size: 1.8em; font-weight: 600; color: #D0D0FF; margin-top: 15px; margin-bottom: 12px;}
 .gr-input input, .gr-input textarea, .gr-dropdown select, .gr-textbox textarea { background-color: #2A2A4A !important; color: #E0E0FF !important; border: 1px solid #4A4A6A !important; border-radius: 8px !important; padding: 10px !important;}
 .gr-button { border-radius: 8px !important; font-weight: 500 !important; transition: all 0.2s ease-in-out !important;}
+.gr-button-primary { padding-top: 10px !important; padding-bottom: 10px !important; } /* Ensure enough padding for icons/text */
 .gr-button-primary:hover { transform: scale(1.03) translateY(-1px) !important; box-shadow: 0 8px 16px rgba(127,0,255,0.3) !important; }
 .panel_image { border-radius: 12px !important; overflow: hidden; box-shadow: 0 6px 15px rgba(0,0,0,0.25) !important; background-color: #23233A;}
 .panel_image img { max-height: 600px !important; }
@@ -103,8 +108,9 @@ body, .gradio-container { background-color: #0F0F1A !important; color: #D0D0E0 !
 #surprise_button:hover { transform: scale(1.03) translateY(-1px) !important; box-shadow: 0 8px 16px rgba(255,126,95,0.3) !important; }
 """
 
-# --- Helper: Placeholder Image Creation (Defined at global scope) ---
+# --- Helper: Placeholder Image Creation ---
 def create_placeholder_image(text="Processing...", size=(512, 512), color="#23233A", text_color="#E0E0FF"):
+    # ... (Full implementation as before) ...
     img = Image.new('RGB', size, color=color); draw = ImageDraw.Draw(img)
     try: font_path = "arial.ttf" if os.path.exists("arial.ttf") else None 
     except: font_path = None 
@@ -114,6 +120,7 @@ def create_placeholder_image(text="Processing...", size=(512, 512), color="#2323
     else: tw, th = draw.textsize(text, font=font) 
     draw.text(((size[0]-tw)/2, (size[1]-th)/2), text, font=font, fill=text_color); return img
 
+
 # --- StoryVerse Weaver Orchestrator ---
 def add_scene_to_story_orchestrator(
     current_story_obj: Story, scene_prompt_text: str, image_style_dropdown: str, artist_style_text: str,
@@ -126,21 +133,20 @@ def add_scene_to_story_orchestrator(
     
     log_accumulator = [f"**🚀 Scene {current_story_obj.current_scene_number + 1} - {time.strftime('%H:%M:%S')}**"]
     
-    # Initialize placeholders for the final return tuple values
+    # Placeholders for the final return tuple
     ret_story_state = current_story_obj
     ret_gallery = current_story_obj.get_all_scenes_for_gallery_display()
-    ret_latest_image = None
-    ret_latest_narrative_str = "## Processing...\nNarrative being woven..."
-    ret_status_bar_html = "<p class='processing_text status_text'>Processing...</p>"
-    # ret_log_md will be built up
+    ret_latest_image = create_placeholder_image("🎨 Conjuring visuals...") # Initial placeholder for image component
+    ret_latest_narrative_md_obj = gr.Markdown(value=" Musing narrative...")
+    ret_status_bar_html_obj = gr.HTML(value=f"<p class='processing_text status_text'>🌌 Weaving Scene {current_story_obj.current_scene_number + 1}...</p>")
+    # Log is built up
 
-    # Initial UI update using direct component updates in yield
-    # These component variables (output_status_bar etc.) must be defined in the `with gr.Blocks` scope
+    # Initial yield for UI updates handled by the .then() chain for buttons
+    # This first yield is for immediate feedback on other components
     yield {
-        output_status_bar: gr.HTML(value=f"<p class='processing_text status_text'>🌌 Weaving Scene {current_story_obj.current_scene_number + 1}...</p>"),
-        output_latest_scene_image: gr.Image(value=create_placeholder_image("🎨 Conjuring visuals...")), # Removed visible=True, Gradio handles it
-        output_latest_scene_narrative: gr.Markdown(value=" Musing narrative..."), # Removed visible=True
-        # engage_button and surprise_button will be handled by the .then() chaining in the UI definition
+        output_status_bar: ret_status_bar_html_obj,
+        output_latest_scene_image: gr.Image(value=ret_latest_image), # Show placeholder
+        output_latest_scene_narrative: ret_latest_narrative_md_obj,
         output_interaction_log_markdown: gr.Markdown(value="\n".join(log_accumulator))
     }
 
@@ -173,8 +179,9 @@ def add_scene_to_story_orchestrator(
             narrative_text_generated = "**Narrative Error:** Selected text model not available or misconfigured."
             log_accumulator.append(f"  Narrative: FAILED - Model '{text_model_key}' not available.")
         
-        ret_latest_narrative_str = f"## Scene Idea: {scene_prompt_text}\n\n{narrative_text_generated}"
-        yield { output_latest_scene_narrative: gr.Markdown(value=ret_latest_narrative_str),
+        ret_latest_narrative_str_content = f"## Scene Idea: {scene_prompt_text}\n\n{narrative_text_generated}"
+        ret_latest_narrative_md_obj = gr.Markdown(value=ret_latest_narrative_str_content)
+        yield { output_latest_scene_narrative: ret_latest_narrative_md_obj,
                 output_interaction_log_markdown: gr.Markdown(value="\n".join(log_accumulator)) }
 
         # --- 2. Generate Image ---
@@ -191,6 +198,8 @@ def add_scene_to_story_orchestrator(
             image_response = None
             if selected_image_provider_type == "stability_ai": image_response = generate_image_stabilityai(full_image_prompt, negative_prompt=negative_prompt_text or COMMON_NEGATIVE_PROMPTS, steps=40 if image_quality=="High Detail" else 25)
             elif selected_image_provider_type == "dalle": image_response = generate_image_dalle(full_image_prompt, quality="hd" if image_quality=="High Detail" else "standard")
+            elif selected_image_provider_type == "hf_image_sdxl_base": image_response = generate_image_hf_model(full_image_prompt, model_id="stabilityai/stable-diffusion-xl-base-1.0", negative_prompt=negative_prompt_text or COMMON_NEGATIVE_PROMPTS)
+            elif selected_image_provider_type == "hf_image_openjourney": image_response = generate_image_hf_model(full_image_prompt, model_id="prompthero/openjourney", negative_prompt=negative_prompt_text or COMMON_NEGATIVE_PROMPTS, width=512, height=512)
             
             if image_response and image_response.success:
                 image_generated_pil = image_response.image
@@ -229,55 +238,43 @@ def add_scene_to_story_orchestrator(
         # --- 4. Prepare Final Values for Return Tuple ---
         ret_gallery = current_story_obj.get_all_scenes_for_gallery_display()
         _ , latest_narr_for_display_final_str_temp = current_story_obj.get_latest_scene_details_for_display()
-        ret_latest_narrative_str = latest_narr_for_display_final_str_temp 
+        ret_latest_narrative_md_obj = gr.Markdown(value=latest_narr_for_display_final_str_temp)
         
         status_html_str_temp = f"<p class='error_text status_text'>Scene {current_story_obj.current_scene_number} added with errors.</p>" if final_scene_error else f"<p class='success_text status_text'>🌌 Scene {current_story_obj.current_scene_number} woven!</p>"
-        ret_status_bar_html = gr.HTML(value=status_html_str_temp) 
+        ret_status_bar_html_obj = gr.HTML(value=status_html_str_temp)
         
         progress(1.0, desc="Scene Complete!")
 
     except ValueError as ve:
         log_accumulator.append(f"\n**INPUT/CONFIG ERROR:** {ve}")
-        ret_status_bar_html = gr.HTML(value=f"<p class='error_text status_text'>❌ CONFIGURATION ERROR: {ve}</p>")
-        ret_latest_narrative_str = f"## Error\n{ve}"
+        ret_status_bar_html_obj = gr.HTML(value=f"<p class='error_text status_text'>❌ CONFIGURATION ERROR: {ve}</p>")
+        ret_latest_narrative_md_obj = gr.Markdown(value=f"## Error\n{ve}")
     except Exception as e:
         log_accumulator.append(f"\n**UNEXPECTED RUNTIME ERROR:** {type(e).__name__} - {e}\n{traceback.format_exc()}")
-        ret_status_bar_html = gr.HTML(value=f"<p class='error_text status_text'>❌ UNEXPECTED ERROR: {type(e).__name__}. Check logs.</p>")
-        ret_latest_narrative_str = f"## Unexpected Error\n{type(e).__name__}: {e}\nSee log for details."
-    # No `finally` block here for button updates; handled by `.then()`
+        ret_status_bar_html_obj = gr.HTML(value=f"<p class='error_text status_text'>❌ UNEXPECTED ERROR: {type(e).__name__}. Check logs.</p>")
+        ret_latest_narrative_md_obj = gr.Markdown(value=f"## Unexpected Error\n{type(e).__name__}: {e}\nSee log for details.")
+    # `finally` block is removed from here; button re-enabling is handled by .then()
 
     current_total_time = time.time() - start_time
     log_accumulator.append(f"  Cycle ended at {time.strftime('%H:%M:%S')}. Total time: {current_total_time:.2f}s")
-    ret_log_str = "\n".join(log_accumulator)
+    ret_log_md = gr.Markdown(value="\n".join(log_accumulator))
     
-    # This is the FINAL return. It must be a tuple matching the `outputs` list of engage_button.click()
     return (
-        ret_story_state,
-        ret_gallery,
-        ret_latest_image, 
-        gr.Markdown(value=ret_latest_narrative_str), 
-        ret_status_bar_html,     
-        gr.Markdown(value=ret_log_str)               
+        ret_story_state, ret_gallery, ret_latest_image, 
+        ret_latest_narrative_md_obj, ret_status_bar_html_obj, ret_log_md
     )
 
 def clear_story_state_ui_wrapper():
     new_story = Story()
     placeholder_img = create_placeholder_image("Your StoryVerse is a blank canvas...", color="#1A1A2E", text_color="#A0A0C0")
     cleared_gallery = [(placeholder_img, "Your StoryVerse is new and untold...")]
-    initial_narrative = "## ✨ A New Story Begins ✨\nDescribe your first scene idea in the panel to the left and let the AI help you weave your world!"
-    status_msg = "<p class='processing_text status_text'>📜 Story Cleared. A fresh canvas awaits your imagination!</p>"
-    return (new_story, cleared_gallery, None, gr.Markdown(value=initial_narrative), gr.HTML(value=status_msg), "Log Cleared. Ready for a new adventure!", "")
+    initial_narrative = "## ✨ A New Story Begins ✨\nDescribe your first scene idea..."
+    status_msg = "<p class='processing_text status_text'>📜 Story Cleared.</p>"
+    return (new_story, cleared_gallery, None, gr.Markdown(initial_narrative), gr.HTML(status_msg), "Log Cleared.", "")
 
 def surprise_me_func():
-    themes = ["Cosmic Horror", "Solarpunk Utopia", "Mythic Fantasy", "Noir Detective", "Silent Film Comedy", "Deep Sea Exploration", "Prehistoric Survival"]
-    actions = ["unearths an artifact of immense power", "negotiates with an interdimensional being", "solves an ancient riddle", "embarks on a perilous journey", "attends a secret festival", "witnesses a celestial event", "finds a hidden sanctuary"]
-    settings = ["on a rogue planet drifting through an empty void", "in a city built within a colossal, living tree", "within a library containing all possible books", "on a generation ship nearing its ancient destination", "in a dreamlike landscape where physics is suggestive", "at the bottom of a Mariana Trench-like abyss", "in a lush jungle teeming with dinosaurs"]
-    prompt = f"A protagonist {random.choice(actions)} {random.choice(settings)}. The overall theme is {random.choice(themes)}."
-    style = random.choice(list(STYLE_PRESETS.keys()))
-    artist = random.choice(["H.R. Giger", "Moebius", "Eyvind Earle", " Remedios Varo", "Alphonse Mucha", ""]*2)
-    return prompt, style, artist
-
-# --- Functions to control button interactivity ---
+    themes = ["Cosmic Horror", "Solarpunk Utopia", "Mythic Fantasy", "Noir Detective"]; actions = ["unearths an artifact", "negotiates"]; settings = ["on a rogue planet", "in a city in a tree"]; prompt = f"A protagonist {random.choice(actions)} {random.choice(settings)}. Theme: {random.choice(themes)}."; style = random.choice(list(STYLE_PRESETS.keys())); artist = random.choice(["H.R. Giger", "Moebius", ""]*2); return prompt, style, artist
+
 def disable_buttons_for_processing():
     return gr.Button(interactive=False), gr.Button(interactive=False)
 
@@ -285,15 +282,35 @@ def enable_buttons_after_processing():
     return gr.Button(interactive=True), gr.Button(interactive=True)
 
 # --- Gradio UI Definition ---
-with gr.Blocks(theme=omega_theme, css=omega_css, title="✨ StoryVerse Omega ✨ - AI Story & World Weaver") as story_weaver_demo:
+with gr.Blocks(theme=omega_theme, css=omega_css, title="✨ StoryVerse Omega ✨") as story_weaver_demo:
+    # Define Python variables for UI components that will be inputs or outputs
     story_state_output = gr.State(Story()) 
+    # Input Column Components
+    scene_prompt_input = gr.Textbox()      
+    image_style_input = gr.Dropdown()      
+    artist_style_input = gr.Textbox()      
+    negative_prompt_input = gr.Textbox()   
+    text_model_dropdown = gr.Dropdown()    
+    image_provider_dropdown = gr.Dropdown()
+    narrative_length_dropdown = gr.Dropdown() 
+    image_quality_dropdown = gr.Dropdown()    
+    # Output Column Components
+    output_gallery = gr.Gallery()
+    output_latest_scene_image = gr.Image()
+    output_latest_scene_narrative = gr.Markdown()
+    output_status_bar = gr.HTML()
+    output_interaction_log_markdown = gr.Markdown()
+    # Button components (updated via chained .then() events)
+    engage_button = gr.Button()
+    surprise_button = gr.Button()
+    clear_story_button = gr.Button()
 
+    # --- Layout the UI ---
     gr.Markdown("<div align='center'><h1>✨ StoryVerse Omega ✨</h1>\n<h3>Craft Immersive Multimodal Worlds with AI</h3></div>")
     gr.HTML("<div class='important-note'><strong>Welcome, Worldsmith!</strong> Describe your vision, choose your style, and let Omega help you weave captivating scenes with narrative and imagery. Ensure API keys (<code>STORYVERSE_...</code>) are correctly set in Space Secrets!</div>")
 
     with gr.Accordion("🔧 AI Services Status & Info", open=False):
-        status_text_list = []
-        text_llm_ok, image_gen_ok = (GEMINI_TEXT_IS_READY or HF_TEXT_IS_READY), (STABILITY_API_IS_READY or OPENAI_DALLE_IS_READY)
+        status_text_list = []; text_llm_ok, image_gen_ok = (GEMINI_TEXT_IS_READY or HF_TEXT_IS_READY), (STABILITY_API_IS_READY or OPENAI_DALLE_IS_READY or HF_IMAGE_IS_READY) # Added HF_IMAGE_IS_READY
         if not text_llm_ok and not image_gen_ok: status_text_list.append("<p style='color:#FCA5A5;font-weight:bold;'>⚠️ CRITICAL: NO AI SERVICES CONFIGURED.</p>")
         else:
             if text_llm_ok: status_text_list.append("<p style='color:#A7F3D0;'>✅ Text Generation Service(s) Ready.</p>")
@@ -309,7 +326,7 @@ with gr.Blocks(theme=omega_theme, css=omega_css, title="✨ StoryVerse Omega ✨
                 scene_prompt_input = gr.Textbox(lines=7, label="Scene Vision (Description, Dialogue, Action):", placeholder="e.g., Amidst swirling cosmic dust...")
             with gr.Row(elem_classes=["compact-row"]):
                 with gr.Column(scale=2):
-                    image_style_input = gr.Dropdown(choices=["Default (Cinematic Realism)"] + sorted(list(STYLE_PRESETS.keys())), value="Default (Cinematic Realism)", label="Visual Style Preset")
+                    image_style_input = gr.Dropdown(choices=["Default (Cinematic Realism)"] + sorted(list(STYLE_PRESETS.keys())), value="Default (Cinematic Realism)", label="Visual Style Preset", allow_custom_value=True) # allow_custom_value for examples
                 with gr.Column(scale=2):
                     artist_style_input = gr.Textbox(label="Artistic Inspiration (Optional):", placeholder="e.g., Moebius...")
             negative_prompt_input = gr.Textbox(lines=2, label="Exclude from Image (Negative Prompt):", value=COMMON_NEGATIVE_PROMPTS)
@@ -339,29 +356,28 @@ with gr.Blocks(theme=omega_theme, css=omega_css, title="✨ StoryVerse Omega ✨
                         output_interaction_log_markdown = gr.Markdown("Log will appear here...")
 
     # Chained event handling for engage_button
-    click_event_engage = engage_button.click(
-        fn=disable_buttons_for_processing, # Step 1: Disable buttons
+    engage_button.click(
+        fn=disable_buttons_for_processing,
         inputs=None,
         outputs=[engage_button, surprise_button],
-        queue=False # Run immediately without queuing for this UI update
+        queue=False 
     ).then(
-        fn=add_scene_to_story_orchestrator, # Step 2: Run the main orchestrator
+        fn=add_scene_to_story_orchestrator,
         inputs=[
             story_state_output, scene_prompt_input, 
             image_style_input, artist_style_input, negative_prompt_input,
             text_model_dropdown, image_provider_dropdown,
             narrative_length_dropdown, image_quality_dropdown
         ],
-        outputs=[ # These are updated by the FINAL RETURN of the orchestrator
+        outputs=[ 
             story_state_output, output_gallery, output_latest_scene_image, 
             output_latest_scene_narrative, output_status_bar, output_interaction_log_markdown
         ]
-        # Progressive updates within orchestrator happen via `yield {component_var: update}`
-    ).then(
-        fn=enable_buttons_after_processing, # Step 3: Re-enable buttons
+    ).then( 
+        fn=enable_buttons_after_processing,
         inputs=None,
         outputs=[engage_button, surprise_button],
-        queue=False # Run immediately after orchestrator finishes
+        queue=False 
     )
     
     clear_story_button.click(
@@ -379,10 +395,10 @@ with gr.Blocks(theme=omega_theme, css=omega_css, title="✨ StoryVerse Omega ✨
         outputs=[scene_prompt_input, image_style_input, artist_style_input]
     )
     gr.Examples(
-        examples=[
+        examples=[ # Ensure these styles are in your STYLE_PRESETS in image_style_prompts.py
             ["A lone, weary traveler on a mechanical steed crosses a vast, crimson desert under twin suns. Dust devils dance in the distance.", "Sci-Fi Western", "Moebius", "greenery, water, modern city"],
             ["Deep within an ancient, bioluminescent forest, a hidden civilization of sentient fungi perform a mystical ritual around a pulsating crystal.", "Psychedelic Fantasy", "Alex Grey", "technology, buildings, roads"],
-             ["A child sits on a crescent moon, fishing for stars in a swirling nebula. A friendly space whale swims nearby.", "Whimsical Cosmic", "James Jean", "realistic, dark, scary"],
+            ["A child sits on a crescent moon, fishing for stars in a swirling nebula. A friendly space whale swims nearby.", "Whimsical Cosmic", "James Jean", "realistic, dark, scary"],
             ["A grand, baroque library where the books fly freely and whisper forgotten lore to those who listen closely.", "Magical Realism", "Remedios Varo", "minimalist, simple, technology"]
         ],
         inputs=[scene_prompt_input, image_style_input, artist_style_input, negative_prompt_input], 
@@ -392,16 +408,10 @@ with gr.Blocks(theme=omega_theme, css=omega_css, title="✨ StoryVerse Omega ✨
 
 # --- Entry Point ---
 if __name__ == "__main__":
+    print("="*80); print("✨ StoryVerse Omega™ Launching... ✨")
+    print(f"  Gemini Text Ready: {GEMINI_TEXT_IS_READY}"); print(f"  HF Text Ready: {HF_TEXT_IS_READY}")
+    print(f"  Stability AI Ready: {STABILITY_API_IS_READY}"); print(f"  DALL-E Ready: {OPENAI_DALLE_IS_READY}"); print(f"  HF Image Ready: {HF_IMAGE_IS_READY}")
+    if not (GEMINI_TEXT_IS_READY or HF_TEXT_IS_READY) or not (STABILITY_API_IS_READY or OPENAI_DALLE_IS_READY or HF_IMAGE_IS_READY): print("  🔴 WARNING: Not all services configured.")
+    print(f"  Default Text Model: {UI_DEFAULT_TEXT_MODEL_KEY}"); print(f"  Default Image Provider: {UI_DEFAULT_IMAGE_PROVIDER_KEY}")
     print("="*80)
-    print("✨ StoryVerse Omega™ - AI Story & World Weaver - Launching... ✨")
-    print(f"  Text LLM Ready (Gemini): {GEMINI_TEXT_IS_READY}")
-    print(f"  Text LLM Ready (HF): {HF_TEXT_IS_READY}")
-    print(f"  Image Provider Ready (Stability AI): {STABILITY_API_IS_READY}")
-    print(f"  Image Provider Ready (DALL-E): {OPENAI_DALLE_IS_READY}")
-    if not (GEMINI_TEXT_IS_READY or HF_TEXT_IS_READY) or not (STABILITY_API_IS_READY or OPENAI_DALLE_IS_READY):
-        print("  🔴 WARNING: Not all required AI services are configured correctly.")
-    print(f"  Default Text Model: {UI_DEFAULT_TEXT_MODEL_KEY}")
-    print(f"  Default Image Provider: {UI_DEFAULT_IMAGE_PROVIDER_KEY}")
-    print("="*80)
-    
     story_weaver_demo.launch(debug=True, server_name="0.0.0.0", share=False)