Update app.py

app.py

@@ -21,54 +21,42 @@ initialize_image_llms()
 
 # --- Get API Readiness Status ---
 GEMINI_TEXT_IS_READY = is_gemini_text_ready()
-HF_TEXT_IS_READY = is_hf_text_ready() # For text fallback
+HF_TEXT_IS_READY = is_hf_text_ready()
 DALLE_IMAGE_IS_READY = is_dalle_ready() 
-HF_IMAGE_IS_READY = is_hf_image_api_ready() # For image fallback
+HF_IMAGE_IS_READY = is_hf_image_api_ready()
 
 # --- Application Configuration (Models, Defaults) ---
 TEXT_MODELS = {}
 UI_DEFAULT_TEXT_MODEL_KEY = None
-
-# Define your Gemini Preview Model IDs here (REPLACE PLACEHOLDERS WITH ACTUAL IDs IF DIFFERENT)
-GEMINI_2_5_PRO_PREVIEW_ID = "gemini-2.5-pro-preview-05-06" # From your screenshot
-GEMINI_2_5_FLASH_PREVIEW_ID = "gemini-2.5-flash-preview-04-17" # From your screenshot
-
 if GEMINI_TEXT_IS_READY:
-    # Add the new Preview models with high priority
-    TEXT_MODELS[f"🚀 Gemini 2.5 Pro Preview (Narrate)"] = {"id": GEMINI_2_5_PRO_PREVIEW_ID, "type": "gemini"}
-    TEXT_MODELS[f"⚡ Gemini 2.5 Flash Preview (Narrate)"] = {"id": GEMINI_2_5_FLASH_PREVIEW_ID, "type": "gemini"}
     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"}
-    
-    # Set the default to your newest preferred model if available
-    if f"🚀 Gemini 2.5 Pro Preview (Narrate)" in TEXT_MODELS: UI_DEFAULT_TEXT_MODEL_KEY = f"🚀 Gemini 2.5 Pro Preview (Narrate)"
-    elif f"⚡ Gemini 2.5 Flash Preview (Narrate)" in TEXT_MODELS: UI_DEFAULT_TEXT_MODEL_KEY = f"⚡ Gemini 2.5 Flash Preview (Narrate)"
-    elif "✨ Gemini 1.5 Flash (Narrate)" in TEXT_MODELS: UI_DEFAULT_TEXT_MODEL_KEY = "✨ Gemini 1.5 Flash (Narrate)"
-    else: UI_DEFAULT_TEXT_MODEL_KEY = "Legacy Gemini 1.0 Pro (Narrate)" # Further fallback
-    print(f"INFO: app.py - Gemini text models populated. Default set to: {UI_DEFAULT_TEXT_MODEL_KEY}")
-
-elif HF_TEXT_IS_READY: # Fallback to HF if no Gemini is ready
-    TEXT_MODELS["Mistral 7B (Narrate via HF - Fallback)"] = {"id": "mistralai/Mistral-7B-Instruct-v0.2", "type": "hf_text"}
-    TEXT_MODELS["Gemma 2B (Narrate via HF - Fallback)"] = {"id": "google/gemma-2b-it", "type": "hf_text"}
-    UI_DEFAULT_TEXT_MODEL_KEY = "Mistral 7B (Narrate via HF - Fallback)"
-    print(f"INFO: app.py - HF text models populated as fallback. Default set to: {UI_DEFAULT_TEXT_MODEL_KEY}")
-    
-if not TEXT_MODELS: # If neither is ready
+if HF_TEXT_IS_READY: # This will be used if Gemini is not 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: # Determine default text model
+    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: # Fallback if preferred defaults are somehow not in the populated list
+        UI_DEFAULT_TEXT_MODEL_KEY = list(TEXT_MODELS.keys())[0]
+else: # No text models configured at all
     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
 if DALLE_IMAGE_IS_READY: 
     IMAGE_PROVIDERS["🖼️ OpenAI DALL-E 3"] = "dalle_3" 
     IMAGE_PROVIDERS["🖼️ OpenAI DALL-E 2 (Legacy)"] = "dalle_2" 
     UI_DEFAULT_IMAGE_PROVIDER_KEY = "🖼️ OpenAI DALL-E 3"
-elif HF_IMAGE_IS_READY: 
-    IMAGE_PROVIDERS["🎡 HF - Stable Diffusion XL Base (Fallback)"] = "hf_sdxl_base" 
-    IMAGE_PROVIDERS["🎠 HF - OpenJourney (Fallback)"] = "hf_openjourney"
-    IMAGE_PROVIDERS["🌌 HF - Stable Diffusion v1.5 (Fallback)"] = "hf_sd_1_5" 
-    UI_DEFAULT_IMAGE_PROVIDER_KEY = "🎡 HF - Stable Diffusion XL Base (Fallback)"
+elif HF_IMAGE_IS_READY: # Fallback to HF if DALL-E not ready
+    IMAGE_PROVIDERS["🎡 HF - SDXL Base"] = "hf_sdxl_base" 
+    IMAGE_PROVIDERS["🎠 HF - OpenJourney"] = "hf_openjourney"
+    IMAGE_PROVIDERS["🌌 HF - SD v1.5"] = "hf_sd_1_5" 
+    UI_DEFAULT_IMAGE_PROVIDER_KEY = "🎡 HF - SDXL Base"
     
 if not IMAGE_PROVIDERS: 
     IMAGE_PROVIDERS["No Image Providers Configured"] = "none"
@@ -78,13 +66,53 @@ elif not UI_DEFAULT_IMAGE_PROVIDER_KEY and IMAGE_PROVIDERS :
 
 
 # --- Gradio UI Theme and CSS ---
-# (omega_theme and omega_css definitions remain THE SAME as the last full app.py version)
-omega_theme = gr.themes.Base(font=[gr.themes.GoogleFont("Lexend Deca")], primary_hue=gr.themes.colors.purple, secondary_hue=gr.themes.colors.pink, neutral_hue=gr.themes.colors.slate).set(body_background_fill="#0F0F1A", block_background_fill="#1A1A2E", block_border_width="1px", block_border_color="#2A2A4A", block_label_background_fill="#2A2A4A", input_background_fill="#2A2A4A", input_border_color="#4A4A6A", button_primary_background_fill="linear-gradient(135deg, #7F00FF 0%, #E100FF 100%)", button_primary_text_color="white", button_secondary_background_fill="#4A4A6A", button_secondary_text_color="#E0E0FF", slider_color="#A020F0")
-omega_css = """ /* ... Paste your full omega_css string here ... */ """ # Make sure this is complete
+omega_theme = gr.themes.Base(
+    font=[gr.themes.GoogleFont("Lexend Deca"), "ui-sans-serif", "system-ui", "sans-serif"],
+    primary_hue=gr.themes.colors.purple, secondary_hue=gr.themes.colors.pink, neutral_hue=gr.themes.colors.slate
+).set(
+    body_background_fill="#0F0F1A", block_background_fill="#1A1A2E", block_border_width="1px",
+    block_border_color="#2A2A4A", block_label_background_fill="#2A2A4A", input_background_fill="#2A2A4A",
+    input_border_color="#4A4A6A", button_primary_background_fill="linear-gradient(135deg, #7F00FF 0%, #E100FF 100%)",
+    button_primary_text_color="white", button_secondary_background_fill="#4A4A6A",
+    button_secondary_text_color="#E0E0FF", slider_color="#A020F0"
+)
+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; display: flex; align-items: center; justify-content: center;}
+.gr-button span { white-space: nowrap !important; overflow: hidden; text-overflow: ellipsis; display: inline-block; max-width: 90%; line-height: normal !important; }
+.gr-button svg { width: 1.1em !important; height: 1.1em !important; margin-right: 4px !important; flex-shrink: 0;}
+.gr-button-primary { padding: 10px 15px !important; } /* Adjusted padding for potentially shorter 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; }
+.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 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 
@@ -107,13 +135,19 @@ def add_scene_to_story_orchestrator(
     log_accumulator = [f"**🚀 Scene {current_story_obj.current_scene_number + 1} - {time.strftime('%H:%M:%S')}**"]
     
     ret_story_state = current_story_obj
-    ret_gallery = current_story_obj.get_all_scenes_for_gallery_display()
+    # Initialize gallery with placeholders or current items to avoid errors if generation fails early
+    initial_gallery_items = current_story_obj.get_all_scenes_for_gallery_display()
+    if not initial_gallery_items: # Handle case where story is new and has no scenes
+        placeholder_img = create_placeholder_image("Waiting for first scene...", size=(180,180), color="#1A1A2E")
+        initial_gallery_items = [(placeholder_img, "Your StoryVerse awaits!")]
+    ret_gallery = initial_gallery_items
+
     ret_latest_image = None
     ret_latest_narrative_md_obj = gr.Markdown(value="## Processing...\nNarrative being woven...")
     ret_status_bar_html_obj = gr.HTML(value="<p class='processing_text status_text'>Processing...</p>")
     # ret_log_md will be built up
 
-    # Initial yield for UI updates (buttons handled by .then() chain)
+    # Initial yield for UI updates (buttons disabled by .then() chain)
     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...")),
@@ -170,8 +204,9 @@ def add_scene_to_story_orchestrator(
                 else: image_generation_error_message = "**Image Error:** DALL-E selected but API not ready."
             elif selected_image_provider_type.startswith("hf_"):
                 if HF_IMAGE_IS_READY:
-                    hf_model_id_to_call = "stabilityai/stable-diffusion-xl-base-1.0"; img_width, img_height = 768, 768
+                    hf_model_id_to_call = "stabilityai/stable-diffusion-xl-base-1.0"; img_width, img_height = 768, 768 # Defaults
                     if selected_image_provider_type == "hf_openjourney": hf_model_id_to_call = "prompthero/openjourney"; img_width,img_height = 512,512
+                    elif selected_image_provider_type == "hf_sdxl_base": hf_model_id_to_call = "stabilityai/stable-diffusion-xl-base-1.0"; # Redundant, but explicit
                     elif selected_image_provider_type == "hf_sd_1_5": hf_model_id_to_call = "runwayml/stable-diffusion-v1-5"; img_width,img_height = 512,512
                     image_response = generate_image_hf_model(full_image_prompt, model_id=hf_model_id_to_call, negative_prompt=negative_prompt_text or COMMON_NEGATIVE_PROMPTS, width=img_width, height=img_height)
                 else: image_generation_error_message = "**Image Error:** HF Image Model selected but API not ready."
@@ -208,6 +243,17 @@ def add_scene_to_story_orchestrator(
         
         # --- 4. Prepare Final Values for Return Tuple ---
         ret_gallery = current_story_obj.get_all_scenes_for_gallery_display()
+        # Ensure gallery items are PIL Images or None for errored/missing images
+        processed_gallery_tuples = []
+        for img_item, cap_text in ret_gallery:
+            if isinstance(img_item, Image.Image):
+                processed_gallery_tuples.append((img_item, cap_text))
+            else: # Assume it's an error or no image, create placeholder for gallery
+                gallery_placeholder = create_placeholder_image(f"S{cap_text.split(':')[0][1:]}\nError/NoImg", size=(180,180), color="#2A2A4A")
+                processed_gallery_tuples.append((gallery_placeholder, cap_text))
+        ret_gallery = processed_gallery_tuples
+
+
         _ , latest_narr_for_display_final_str_temp = current_story_obj.get_latest_scene_details_for_display()
         ret_latest_narrative_md_obj = gr.Markdown(value=latest_narr_for_display_final_str_temp)
         
@@ -229,6 +275,7 @@ def add_scene_to_story_orchestrator(
     log_accumulator.append(f"  Cycle ended at {time.strftime('%H:%M:%S')}. Total time: {current_total_time:.2f}s")
     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, 
         ret_latest_narrative_md_obj, ret_status_bar_html_obj, ret_log_md
@@ -236,38 +283,36 @@ def add_scene_to_story_orchestrator(
 
 def clear_story_state_ui_wrapper():
     new_story = Story(); ph_img = create_placeholder_image("Blank canvas...", color="#1A1A2E", text_color="#A0A0C0")
-    return (new_story, [(ph_img,"New StoryVerse...")], None, gr.Markdown("## ✨ New Story ✨"), gr.HTML("<p class='processing_text status_text'>📜 Story Cleared.</p>"), "Log Cleared.", "")
+    # Ensure gallery output for clear is also a list of (image, caption)
+    cleared_gallery_display = [(ph_img, "Your StoryVerse is new and untold...")]
+    initial_narrative = "## ✨ New Story ✨\nDescribe your first scene!"
+    status_msg = "<p class='processing_text status_text'>📜 Story Cleared.</p>"
+    return (new_story, cleared_gallery_display, None, gr.Markdown(initial_narrative), gr.HTML(status_msg), "Log Cleared.", "")
 
 def surprise_me_func():
-    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
+    print("DEBUG: surprise_me_func called") # For checking button functionality
+    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)
+    print(f"DEBUG: surprise_me_func returning: {prompt}, {style}, {artist}")
+    return prompt, style, artist
 
 def disable_buttons_for_processing():
+    print("DEBUG: Disabling buttons")
     return gr.Button(interactive=False), gr.Button(interactive=False)
 
 def enable_buttons_after_processing():
+    print("DEBUG: Enabling buttons")
     return gr.Button(interactive=True), gr.Button(interactive=True)
 
 # --- Gradio UI Definition ---
 with gr.Blocks(theme=omega_theme, css=omega_css, title="✨ StoryVerse Omega ✨") as story_weaver_demo:
+    # Define Python variables for UI components
     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>")
-
-    with gr.Accordion("🔧 AI Services Status & Info", open=False):
-        status_text_list = []; text_llm_ok = (GEMINI_TEXT_IS_READY or HF_TEXT_IS_READY); image_gen_ok = (DALLE_IMAGE_IS_READY or 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 Ready.</p>")
-            else: status_text_list.append("<p style='color:#FCD34D;'>⚠️ Text Generation NOT Ready.</p>")
-            if image_gen_ok: status_text_list.append("<p style='color:#A7F3D0;'>✅ Image Generation Ready.</p>")
-            else: status_text_list.append("<p style='color:#FCD34D;'>⚠️ Image Generation NOT Ready.</p>")
-        gr.HTML("".join(status_text_list))
-
-    with gr.Row(equal_height=False, variant="panel"):
+    
+    with gr.Row(equal_height=False, variant="panel"): # Main layout row
+        # Input Column
         with gr.Column(scale=7, min_width=450): 
             gr.Markdown("### 💡 **Craft Your Scene**", elem_classes="input-section-header")
-            with gr.Group(): scene_prompt_input = gr.Textbox(lines=7, label="Scene Vision:", placeholder="e.g., Amidst swirling cosmic dust...")
+            with gr.Group(): 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", allow_custom_value=True)
                 with gr.Column(scale=2): artist_style_input = gr.Textbox(label="Artistic Inspiration (Optional):", placeholder="e.g., Moebius...")
@@ -275,15 +320,17 @@ with gr.Blocks(theme=omega_theme, css=omega_css, title="✨ StoryVerse Omega ✨
             with gr.Accordion("⚙️ Advanced AI Configuration", open=False):
                  with gr.Group():
                     text_model_dropdown = gr.Dropdown(choices=list(TEXT_MODELS.keys()), value=UI_DEFAULT_TEXT_MODEL_KEY, label="Narrative AI Engine")
-                    image_provider_dropdown = gr.Dropdown(choices=list(IMAGE_PROVIDERS.keys()), value=UI_DEFAULT_IMAGE_PROVIDER_KEY, label="Visual AI Engine (DALL-E/HF)") 
+                    image_provider_dropdown = gr.Dropdown(choices=list(IMAGE_PROVIDERS.keys()), value=UI_DEFAULT_IMAGE_PROVIDER_KEY, label="Visual AI Engine")
                     with gr.Row():
                         narrative_length_dropdown = gr.Dropdown(["Short (1 paragraph)", "Medium (2-3 paragraphs)", "Detailed (4+ paragraphs)"], value="Medium (2-3 paragraphs)", label="Narrative Detail")
                         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):
-                engage_button = gr.Button("🌌 Weave Scene!", variant="primary", scale=3, icon="✨")
-                surprise_button = gr.Button("🎲 Surprise!", variant="secondary", scale=1, icon="🎁")
-                clear_story_button = gr.Button("🗑️ New", variant="stop", scale=1, icon="♻️")
-            output_status_bar = gr.HTML(value="<p class='processing_text status_text'>Ready to weave!</p>")
+                engage_button = gr.Button("🌌 Weave!", variant="primary", scale=3, icon="✨") # Shorter text
+                surprise_button = gr.Button("🎲 Surprise!", variant="secondary", scale=1, icon="🎁")# Shorter text
+                clear_story_button = gr.Button("🗑️ New", variant="stop", scale=1, icon="♻️") # Shorter text
+            output_status_bar = gr.HTML(value="<p class='processing_text status_text'>Ready to weave your first masterpiece!</p>")
+        
+        # Output Column
         with gr.Column(scale=10, min_width=700): 
             gr.Markdown("### 🖼️ **Your StoryVerse**", elem_classes="output-section-header")
             with gr.Tabs():
@@ -296,18 +343,18 @@ with gr.Blocks(theme=omega_theme, css=omega_css, title="✨ StoryVerse Omega ✨
                     with gr.Accordion("Interaction Log", open=False): 
                         output_interaction_log_markdown = gr.Markdown("Log...")
 
-    engage_button.click(fn=disable_buttons_for_processing, outputs=[engage_button, surprise_button], queue=False)\
-        .then(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=[story_state_output, output_gallery, output_latest_scene_image, output_latest_scene_narrative, output_status_bar, output_interaction_log_markdown])\
-        .then(fn=enable_buttons_after_processing, outputs=[engage_button, surprise_button], queue=False)
-    
-    clear_story_button.click(fn=clear_story_state_ui_wrapper, 
-                             outputs=[story_state_output, output_gallery, output_latest_scene_image, output_latest_scene_narrative, output_status_bar, output_interaction_log_markdown, scene_prompt_input])
-    
-    surprise_button.click(fn=surprise_me_func, 
-                            outputs=[scene_prompt_input, image_style_input, artist_style_input])
-    
+    # API Status (defined after main layout to ensure it's below everything)
+    with gr.Accordion("🔧 AI Services Status & Info", open=False, elem_id="api_status_accordion"):
+        status_text_list = []; text_llm_ok = (GEMINI_TEXT_IS_READY or HF_TEXT_IS_READY); image_gen_ok = (DALLE_IMAGE_IS_READY or 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 Ready.</p>")
+            else: status_text_list.append("<p style='color:#FCD34D;'>⚠️ Text Generation NOT Ready.</p>")
+            if image_gen_ok: status_text_list.append("<p style='color:#A7F3D0;'>✅ Image Generation Ready.</p>")
+            else: status_text_list.append("<p style='color:#FCD34D;'>⚠️ Image Generation NOT Ready.</p>")
+        gr.HTML("".join(status_text_list))
+
+    # Examples (defined after main layout)
     gr.Examples(
         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"],
@@ -320,13 +367,25 @@ with gr.Blocks(theme=omega_theme, css=omega_css, title="✨ StoryVerse Omega ✨
     )
     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>")
 
+    # Event Handlers
+    engage_event_actions = engage_button.click(fn=disable_buttons_for_processing, outputs=[engage_button, surprise_button], queue=False)\
+        .then(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=[story_state_output, output_gallery, output_latest_scene_image, output_latest_scene_narrative, output_status_bar, output_interaction_log_markdown])\
+        .then(fn=enable_buttons_after_processing, outputs=[engage_button, surprise_button], queue=False)
+    
+    clear_story_button.click(fn=clear_story_state_ui_wrapper, 
+                             outputs=[story_state_output, output_gallery, output_latest_scene_image, output_latest_scene_narrative, output_status_bar, output_interaction_log_markdown, scene_prompt_input])
+    
+    surprise_button.click(fn=surprise_me_func, 
+                            outputs=[scene_prompt_input, image_style_input, artist_style_input])
+    
 # --- Entry Point ---
 if __name__ == "__main__":
-    print("="*80); print("✨ StoryVerse Omega (DALL-E/Gemini/HF Focus) Launching... ✨")
+    print("="*80); print("✨ StoryVerse Omega (Full App with Fixes) Launching... ✨")
     print(f"  Gemini Text Ready: {GEMINI_TEXT_IS_READY}"); print(f"  HF Text Ready: {HF_TEXT_IS_READY}")
     print(f"  DALL-E Image Ready: {DALLE_IMAGE_IS_READY}"); print(f"  HF Image API Ready: {HF_IMAGE_IS_READY}")
-    if not (GEMINI_TEXT_IS_READY or HF_TEXT_IS_READY) or not (DALLE_IMAGE_IS_READY or HF_IMAGE_IS_READY): 
-        print("  🔴 WARNING: Not all primary/fallback AI services configured.")
+    if not (GEMINI_TEXT_IS_READY or HF_TEXT_IS_READY) or not (DALLE_IMAGE_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)
     story_weaver_demo.launch(debug=True, server_name="0.0.0.0", share=False)