Update app.py

app.py

@@ -2,10 +2,10 @@
 import gradio as gr
 import os
 import time
-import json
-from PIL import Image, ImageDraw, ImageFont
+import json 
+from PIL import Image, ImageDraw, ImageFont 
 import random
-import traceback # For more detailed error logging in wrapper
+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
@@ -67,18 +67,47 @@ omega_theme = gr.themes.Base(
     button_primary_text_color="white", button_secondary_background_fill="#4A4A6A",
     button_secondary_text_color="#E0E0FF", slider_color="#A020F0"
 )
-omega_css = """ /* ... (Your full omega_css string from the "WOW UI" version) ... */ """
-# For brevity, I'll assume the full omega_css string is here. Copy it from the previous response.
-# Example start: body, .gradio-container { background-color: #0F0F1A !important; ... }
+omega_css = """
+body, .gradio-container { background-color: #0F0F1A !important; color: #D0D0E0 !important; }
+.gradio-container { max-width: 1400px !important; margin: auto !important; border-radius: 20px; box-shadow: 0 10px 30px rgba(0,0,0,0.2); padding: 25px !important; border: 1px solid #2A2A4A;}
+.gr-panel, .gr-box, .gr-accordion { background-color: #1A1A2E !important; border: 1px solid #2A2A4A !important; border-radius: 12px !important; box-shadow: 0 4px 15px rgba(0,0,0,0.1);}
+.gr-markdown h1 { font-size: 2.8em !important; text-align: center; color: transparent; background: linear-gradient(135deg, #A020F0 0%, #E040FB 100%); -webkit-background-clip: text; background-clip: text; margin-bottom: 5px !important; letter-spacing: -1px;}
+.gr-markdown h3 { color: #C080F0 !important; text-align: center; font-weight: 400; margin-bottom: 25px !important;}
+.input-section-header { font-size: 1.6em; font-weight: 600; color: #D0D0FF; margin-top: 15px; margin-bottom: 8px; border-bottom: 2px solid #7F00FF; padding-bottom: 5px;}
+.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: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; }
+.gallery_output { background-color: transparent !important; border: none !important; }
+.gallery_output .thumbnail-item { border-radius: 8px !important; box-shadow: 0 3px 8px rgba(0,0,0,0.2) !important; margin: 6px !important; transition: transform 0.2s ease; height: 180px !important; width: 180px !important;}
+.gallery_output .thumbnail-item:hover { transform: scale(1.05); }
+.status_text { font-weight: 500; padding: 12px 18px; text-align: center; border-radius: 8px; margin-top:12px; border: 1px solid transparent; font-size: 1.05em;}
+.error_text { background-color: #401010 !important; color: #FFB0B0 !important; border-color: #802020 !important; }
+.success_text { background-color: #104010 !important; color: #B0FFB0 !important; border-color: #208020 !important;}
+.processing_text { background-color: #102040 !important; color: #B0D0FF !important; border-color: #204080 !important;}
+.important-note { background-color: rgba(127,0,255,0.1); border-left: 5px solid #7F00FF; padding: 15px; margin-bottom:20px; color: #E0E0FF; border-radius: 6px;}
+.gr-tabitem { background-color: #1A1A2E !important; border-radius: 0 0 12px 12px !important; padding: 15px !important;}
+.gr-tab-button.selected { background-color: #2A2A4A !important; color: white !important; border-bottom: 3px solid #A020F0 !important; border-radius: 8px 8px 0 0 !important; font-weight: 600 !important;}
+.gr-tab-button { color: #A0A0C0 !important; border-radius: 8px 8px 0 0 !important;}
+.gr-accordion > .gr-block { border-top: 1px solid #2A2A4A !important; }
+.gr-markdown code { background-color: #2A2A4A !important; color: #C0C0E0 !important; padding: 0.2em 0.5em; border-radius: 4px; }
+.gr-markdown pre { background-color: #23233A !important; padding: 1em !important; border-radius: 6px !important; border: 1px solid #2A2A4A !important;}
+.gr-markdown pre > code { padding: 0 !important; background-color: transparent !important; }
+#surprise_button { background: linear-gradient(135deg, #ff7e5f 0%, #feb47b 100%) !important; font-weight:600 !important;}
+#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 ---
 def create_placeholder_image(text="Processing...", size=(512, 512), color="#23233A", text_color="#E0E0FF"):
-    # ... (Full implementation from previous "WOW UI" version) ...
     img = Image.new('RGB', size, color=color); draw = ImageDraw.Draw(img)
-    try: font = ImageFont.truetype("arial.ttf", 40) if os.path.exists("arial.ttf") else ImageFont.load_default()
+    try: font_path = "arial.ttf" if os.path.exists("arial.ttf") else None 
+    except: font_path = None # Handle potential OS errors accessing font list
+    try: font = ImageFont.truetype(font_path, 40) if font_path else ImageFont.load_default()
     except IOError: font = ImageFont.load_default()
     if hasattr(draw, 'textbbox'): bbox = draw.textbbox((0,0), text, font=font); tw, th = bbox[2]-bbox[0], bbox[3]-bbox[1]
-    else: tw, th = draw.textsize(text, font=font) # Fallback
+    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 ---
@@ -88,50 +117,64 @@ def add_scene_to_story_orchestrator(
     narrative_length: str, image_quality: str,
     progress=gr.Progress(track_tqdm=True)
 ):
-    # --- Define UI components for yielding (must match those in `with gr.Blocks...`) ---
-    # This mapping is conceptual if you yield component objects directly.
-    # If yielding dicts with string keys (like elem_id), ensure they match.
-    # For direct object yield:
-    # ui_status = output_status_bar 
-    # ui_latest_img = output_latest_scene_image
-    # ... etc.
-
-    if not current_story_obj: current_story_obj = Story()
+    if not current_story_obj: current_story_obj = Story() 
     
     log_accumulator = [f"**🚀 Scene {current_story_obj.current_scene_number + 1} - {time.strftime('%H:%M:%S')}**"]
-    # Initial UI update using direct component updates in yield
+    
+    # Initial UI update using direct component updates in yield by component variable name
     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..."), visible=True),
         output_latest_scene_narrative: gr.Markdown(value=" Musing narrative...", visible=True),
         engage_button: gr.Button(interactive=False),
         surprise_button: gr.Button(interactive=False),
-        output_interaction_log_markdown: gr.Markdown(value="\n".join(log_accumulator)) # Initial log
+        output_interaction_log_markdown: gr.Markdown(value="\n".join(log_accumulator))
+    }
+
+    # Prepare a dictionary to hold final return values for the .click() outputs list
+    # Initialize with current state or placeholders
+    final_return_dict = {
+        "story_state": current_story_obj,
+        "gallery": current_story_obj.get_all_scenes_for_gallery_display(), # Initial gallery
+        "latest_image": create_placeholder_image("Pending..."),
+        "latest_narrative": "## Pending...\nNarrative generation in progress.",
+        "status_bar": "<p class='processing_text status_text'>Processing...</p>",
+        "log": "\n".join(log_accumulator)
     }
 
     try:
+        start_time = time.time() 
+
         if not scene_prompt_text.strip():
             raise ValueError("Scene prompt cannot be empty!")
 
         # --- 1. Generate Narrative Text ---
         progress(0.1, desc="✍️ Crafting narrative...")
-        narrative_text_generated = f"Narrative Error: Generation failed for '{scene_prompt_text[:30]}...'." # Default error
+        narrative_text_generated = f"Narrative Error: Generation failed for '{scene_prompt_text[:30]}...'."
         text_model_info = TEXT_MODELS.get(text_model_key)
         if text_model_info and text_model_info["type"] != "none":
-            # ... (Full narrative generation logic as in your last working add_scene_to_story) ...
-            system_p = get_narrative_system_prompt("default") 
-            prev_narrative = current_story_obj.get_last_scene_narrative() 
+            system_p = get_narrative_system_prompt("default")
+            prev_narrative = current_story_obj.get_last_scene_narrative()
             user_p = format_narrative_user_prompt(scene_prompt_text, prev_narrative)
             log_accumulator.append(f"  Narrative: Using {text_model_key} ({text_model_info['id']}). Length aim: {narrative_length}")
             text_response = None
             if text_model_info["type"] == "gemini": text_response = generate_text_gemini(user_p, model_id=text_model_info["id"], system_prompt=system_p, max_tokens=768 if narrative_length.startswith("Detailed") else 400)
             elif text_model_info["type"] == "hf_text": text_response = generate_text_hf(user_p, model_id=text_model_info["id"], system_prompt=system_p, max_tokens=768 if narrative_length.startswith("Detailed") else 400)
-            if text_response and text_response.success: narrative_text_generated = basic_text_cleanup(text_response.text); log_accumulator.append(f"  Narrative: Success.")
-            elif text_response: narrative_text_generated = f"**Narrative Error ({text_model_key}):** {text_response.error}"; log_accumulator.append(f"  Narrative: FAILED - {text_response.error}")
-            else: log_accumulator.append(f"  Narrative: FAILED - No response from {text_model_key}.")
-        else: narrative_text_generated = "**Narrative Error:** Text model unavailable."; log_accumulator.append(f"  Narrative: FAILED - Model '{text_model_key}' unavailable.")
+            
+            if text_response and text_response.success:
+                narrative_text_generated = basic_text_cleanup(text_response.text)
+                log_accumulator.append(f"  Narrative: Success. (Snippet: {narrative_text_generated[:50]}...)")
+            elif text_response:
+                narrative_text_generated = f"**Narrative Error ({text_model_key}):** {text_response.error}"
+                log_accumulator.append(f"  Narrative: FAILED - {text_response.error}")
+            else:
+                 log_accumulator.append(f"  Narrative: FAILED - No response object from {text_model_key}.")
+        else:
+            narrative_text_generated = "**Narrative Error:** Selected text model not available or misconfigured."
+            log_accumulator.append(f"  Narrative: FAILED - Model '{text_model_key}' not available.")
         
-        yield { output_latest_scene_narrative: gr.Markdown(value=f"## Scene Idea: {scene_prompt_text}\n\n{narrative_text_generated}"),
+        final_return_dict["latest_narrative"] = f"## Scene Idea: {scene_prompt_text}\n\n{narrative_text_generated}"
+        yield { output_latest_scene_narrative: gr.Markdown(value=final_return_dict["latest_narrative"]),
                 output_interaction_log_markdown: gr.Markdown(value="\n".join(log_accumulator)) }
 
         # --- 2. Generate Image ---
@@ -145,25 +188,33 @@ def add_scene_to_story_orchestrator(
         log_accumulator.append(f"  Image: Using {image_provider_key}. Style: {image_style_dropdown}. Artist: {artist_style_text or 'N/A'}.")
 
         if selected_image_provider_type and selected_image_provider_type != "none":
-            # ... (Full image generation logic as in your last working add_scene_to_story) ...
             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")
-            if image_response and image_response.success: image_generated_pil = image_response.image; log_accumulator.append(f"  Image: Success.")
-            elif image_response: image_generation_error_message = f"**Image Error ({image_response.provider}):** {image_response.error}"; log_accumulator.append(f"  Image: FAILED - {image_response.error}")
-            else: image_generation_error_message = f"**Image Error:** No response from {image_provider_key}."; log_accumulator.append(f"  Image: FAILED - No response.")
-        else: image_generation_error_message = "**Image Error:** Image provider unavailable."; log_accumulator.append(f"  Image: FAILED - Provider '{image_provider_key}' unavailable.")
+            
+            if image_response and image_response.success:
+                image_generated_pil = image_response.image
+                log_accumulator.append(f"  Image: Success from {image_response.provider}.")
+            elif image_response:
+                image_generation_error_message = f"**Image Error ({image_response.provider}):** {image_response.error}"
+                log_accumulator.append(f"  Image: FAILED - {image_response.error}")
+            else:
+                image_generation_error_message = f"**Image Error:** No response object from {image_provider_key} service."
+                log_accumulator.append(f"  Image: FAILED - No response object from {image_provider_key}.")
+        else:
+            image_generation_error_message = "**Image Error:** Selected image provider not available or misconfigured."
+            log_accumulator.append(f"  Image: FAILED - Provider '{image_provider_key}' not available.")
 
-        yield { output_latest_scene_image: gr.Image(value=image_generated_pil if image_generated_pil else create_placeholder_image("Image Gen Failed", color="#401010")),
+        final_return_dict["latest_image"] = image_generated_pil if image_generated_pil else create_placeholder_image("Image Gen Failed", color="#401010")
+        yield { output_latest_scene_image: gr.Image(value=final_return_dict["latest_image"]),
                 output_interaction_log_markdown: gr.Markdown(value="\n".join(log_accumulator)) }
 
         # --- 3. Add Scene to Story Object ---
         final_scene_error = None
-        # ... (Logic to set final_scene_error based on narrative and image errors) ...
         if image_generation_error_message and "**Narrative Error**" in narrative_text_generated : final_scene_error = f"{narrative_text_generated}\n{image_generation_error_message}"
         elif "**Narrative Error**" in narrative_text_generated: final_scene_error = narrative_text_generated
         elif image_generation_error_message: final_scene_error = image_generation_error_message
-
+        
         current_story_obj.add_scene_from_elements(
             user_prompt=scene_prompt_text,
             narrative_text=narrative_text_generated if "**Narrative Error**" not in narrative_text_generated else "(Narrative gen failed)",
@@ -172,41 +223,40 @@ def add_scene_to_story_orchestrator(
             image_provider=image_provider_key if selected_image_provider_type != "none" else "N/A",
             error_message=final_scene_error
         )
-        log_accumulator.append(f"  Scene {current_story_obj.current_scene_number} processed.")
+        final_return_dict["story_state"] = current_story_obj # Update story state in final dict
+        log_accumulator.append(f"  Scene {current_story_obj.current_scene_number} processed and added to story object.")
         
         # --- 4. Prepare Final Outputs for Gradio ---
-        gallery_items_tuples = current_story_obj.get_all_scenes_for_gallery_display()
-        _ , latest_narr_for_display_final = current_story_obj.get_latest_scene_details_for_display()
+        final_return_dict["gallery"] = current_story_obj.get_all_scenes_for_gallery_display()
+        _ , final_return_dict["latest_narrative"] = current_story_obj.get_latest_scene_details_for_display() # Update narrative for final return
         
-        status_message_html = 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>"
+        final_return_dict["status_bar"] = 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>"
         
         progress(1.0, desc="Scene Complete!")
 
-        # Final return for the .click() handler's `outputs` list
-        return (
-            current_story_obj,
-            gallery_items_tuples,
-            image_generated_pil if image_generated_pil else create_placeholder_image("No Image" if not final_scene_error else "Image Error", color="#23233A"),
-            latest_narr_for_display_final,
-            gr.HTML(value=status_message_html),
-            "\n".join(log_accumulator)
-        )
-
-    except ValueError as ve: # Catch our specific input/config errors
+    except ValueError as ve:
         log_accumulator.append(f"\n**INPUT/CONFIG ERROR:** {ve}")
-        status_html_err = f"<p style='color: red;'>❌ CONFIGURATION ERROR: {ve}</p>"
-        # Return values for all outputs in case of error to prevent None issues
-        return current_story_obj, current_story_obj.get_all_scenes_for_gallery_display(), None, f"## Error\n{ve}", gr.HTML(value=status_html_err), "\n".join(log_accumulator)
+        final_return_dict["status_bar"] = f"<p class='error_text status_text'>❌ CONFIGURATION ERROR: {ve}</p>"
+        final_return_dict["latest_narrative"] = f"## Error\n{ve}"
     except Exception as e:
         log_accumulator.append(f"\n**UNEXPECTED RUNTIME ERROR:** {type(e).__name__} - {e}\n{traceback.format_exc()}")
-        status_html_err_unexp = f"<p style='color: red;'>❌ UNEXPECTED ERROR: {e}. Check logs.</p>"
-        return current_story_obj, current_story_obj.get_all_scenes_for_gallery_display(), None, "## Unexpected Error\nSee log for details.", gr.HTML(value=status_html_err_unexp), "\n".join(log_accumulator)
-    finally: # Ensure buttons are re-enabled even if an error occurs mid-yield
-        # This yield will be the last one if no error, or one of the last if error happened
-        # However, the final `return` is what matters for the outputs list of .click()
-        # We can also yield here to ensure buttons are re-enabled after processing.
+        final_return_dict["status_bar"] = f"<p class='error_text status_text'>❌ UNEXPECTED ERROR: {e}. Check logs.</p>"
+        final_return_dict["latest_narrative"] = f"## Unexpected Error\n{e}\nSee log for details."
+    finally:
+        log_accumulator.append(f"  Cycle ended at {time.strftime('%H:%M:%S')}. Total time: {time.time() - (start_time if 'start_time' in locals() else time.time()):.2f}s")
+        final_return_dict["log"] = "\n".join(log_accumulator)
+        # Final yield to re-enable buttons, this will be overridden by the return for other components
         yield { engage_button: gr.Button(interactive=True), surprise_button: gr.Button(interactive=True) }
 
+    # Final return statement for the .click() handler
+    return (
+        final_return_dict["story_state"],
+        final_return_dict["gallery"],
+        final_return_dict["latest_image"],
+        gr.Markdown(value=final_return_dict["latest_narrative"]),
+        gr.HTML(value=final_return_dict["status_bar"]),
+        gr.Markdown(value=final_return_dict["log"])
+    )
 
 def clear_story_state_ui_wrapper():
     new_story = Story()
@@ -214,34 +264,21 @@ def clear_story_state_ui_wrapper():
     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 a TUPLE in the order of the .click() outputs list
-    return (
-        new_story, 
-        cleared_gallery, 
-        None, # For latest_scene_image_output
-        gr.Markdown(value=initial_narrative), # For latest_scene_narrative_output
-        gr.HTML(value=status_msg), # For output_status_bar
-        "Log Cleared. Ready for a new adventure!", # For output_interaction_log_markdown
-        "" # For scene_prompt_input (to clear it)
-    )
+    return (new_story, cleared_gallery, None, gr.Markdown(value=initial_narrative), gr.HTML(value=status_msg), "Log Cleared. Ready for a new adventure!", "")
 
 def surprise_me_func():
-    # ... (Same surprise_me_func as before)
-    themes = ["Cosmic Horror", "Solarpunk Utopia", "Mythic Fantasy", "Noir Detective", "Silent Film Comedy"]
-    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"]
-    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"]
+    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", ""]*2)
+    artist = random.choice(["H.R. Giger", "Moebius", "Eyvind Earle", " Remedios Varo", "Alphonse Mucha", ""]*2)
     return prompt, style, artist
 
-
 # --- Gradio UI Definition ---
 with gr.Blocks(theme=omega_theme, css=omega_css, title="✨ StoryVerse Omega ✨ - AI Story & World Weaver") as story_weaver_demo:
     # Define output components that will be updated by the orchestrator
-    # story_state_output will be a gr.State component
-    story_state_output = gr.State(Story()) # Initial value
+    story_state_output = gr.State(Story()) 
 
     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>")
@@ -280,6 +317,7 @@ with gr.Blocks(theme=omega_theme, css=omega_css, title="✨ StoryVerse Omega ✨
                         image_quality_dropdown = gr.Dropdown(["Standard", "High Detail", "Sketch Concept"], value="Standard", label="Image Detail/Style")
             
             with gr.Row(elem_classes=["compact-row"], equal_height=True):
+                # Assign these buttons to Python variables to be updated by yield
                 engage_button = gr.Button("🌌 Weave This Scene!", variant="primary", scale=3, icon="✨")
                 surprise_button = gr.Button("🎲 Surprise Me!", variant="secondary", scale=1, icon="🎁")
                 clear_story_button = gr.Button("🗑️ New Story", variant="stop", scale=1, icon="♻️")
@@ -316,7 +354,6 @@ with gr.Blocks(theme=omega_theme, css=omega_css, title="✨ StoryVerse Omega ✨
             output_status_bar, 
             output_interaction_log_markdown
         ]
-        # Progressive updates are handled by `yield {component_variable: new_value}` inside the orchestrator
     )
     clear_story_button.click(
         fn=clear_story_state_ui_wrapper,
@@ -338,15 +375,13 @@ with gr.Blocks(theme=omega_theme, css=omega_css, title="✨ StoryVerse Omega ✨
         examples=[
             ["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 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], 
         label="🌌 Example Universes to Weave 🌌",
     )
     gr.HTML("<div style='text-align:center; margin-top:30px; padding-bottom:20px;'><p style='font-size:0.9em; color:#8080A0;'>✨ StoryVerse Omega™ - Weaving Worlds with Words and Pixels ✨</p></div>")
 
-# --- Entry Point for Running the Gradio App ---
+# --- Entry Point ---
 if __name__ == "__main__":
     print("="*80)
     print("✨ StoryVerse Omega™ - AI Story & World Weaver - Launching... ✨")
@@ -355,7 +390,7 @@ if __name__ == "__main__":
     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. Functionality will be severely limited or fail.")
+        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)