Update app.py

app.py

@@ -1,30 +1,43 @@
-# storyverse_weaver_streamlit/app_st.py (example name)
-import streamlit as st
-from PIL import Image, ImageDraw, ImageFont
+# app.py (Streamlit version for StoryVerse Weaver)
+import streamlit as st # FIRST STREAMLIT IMPORT
+from PIL import Image, ImageDraw, ImageFont # For creating placeholder images
 import os
 import time
 import random
+import traceback # For better error display
 
-# --- Assuming your core logic is in a sibling 'core' directory ---
-# You might need to adjust sys.path if running locally vs. deployed
+# --- Page Configuration (MUST BE THE VERY FIRST STREAMLIT COMMAND) ---
+st.set_page_config(
+    page_title="✨ StoryVerse Weaver ✨",
+    page_icon="😻", 
+    layout="wide", 
+    initial_sidebar_state="expanded" 
+)
+# --- END OF PAGE CONFIG ---
+
+# --- Add project root to sys.path if core modules are in a subdirectory ---
+# This might be needed depending on your exact file structure when running locally.
+# For Hugging Face Spaces, if 'core' and 'prompts' are at the same level as app.py,
+# direct imports usually work.
 # import sys
-# sys.path.append(os.path.join(os.path.dirname(__file__), '..')) # If core is one level up
+# SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
+# sys.path.append(os.path.dirname(SCRIPT_DIR)) # If app.py is in a subdir like 'src'
+# sys.path.append(SCRIPT_DIR) # If core/prompts are subdirs of where app.py is
 
-from core.llm_services import initialize_text_llms, is_gemini_text_ready, is_hf_text_ready, generate_text_gemini, generate_text_hf
+# --- Core Logic Imports (NOW AFTER st.set_page_config) ---
+from core.llm_services import initialize_text_llms, is_gemini_text_ready, is_hf_text_ready, generate_text_gemini, generate_text_hf, LLMTextResponse
 from core.image_services import initialize_image_llms, is_dalle_ready, is_hf_image_api_ready, generate_image_dalle, generate_image_hf_model, ImageGenResponse
-from core.story_engine import Story, Scene # Your existing Story and Scene classes
+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
 from core.utils import basic_text_cleanup
 
-# --- Initialize Services ONCE ---
-# Use Streamlit's caching for resource-heavy initializations if they don't depend on session state
-@st.cache_resource # Caches the result across sessions/reruns if inputs don't change
-def load_ai_services():
+# --- Initialize AI Services (Cached) ---
+@st.cache_resource 
+def load_ai_services_config():
     print("--- Initializing AI Services (Streamlit Cache Resource) ---")
     initialize_text_llms()
     initialize_image_llms()
-    # Return status flags to be stored in session_state or used directly
     return {
         "gemini_text_ready": is_gemini_text_ready(),
         "hf_text_ready": is_hf_text_ready(),
@@ -32,29 +45,117 @@ def load_ai_services():
         "hf_image_ready": is_hf_image_api_ready()
     }
 
-ai_services_status = load_ai_services()
+AI_SERVICES_STATUS = load_ai_services_config()
 
 # --- Application Configuration (Models, Defaults) ---
-# (Similar logic to your Gradio app.py for populating TEXT_MODELS, IMAGE_PROVIDERS etc.)
 TEXT_MODELS = {}
 UI_DEFAULT_TEXT_MODEL_KEY = None
-# ... (Populate based on ai_services_status["gemini_text_ready"], ai_services_status["hf_text_ready"]) ...
-if ai_services_status["gemini_text_ready"]: TEXT_MODELS["✨ Gemini 1.5 Flash (Narrate)"] = {"id": "gemini-1.5-flash-latest", "type": "gemini"} # etc.
-if ai_services_status["hf_text_ready"]: TEXT_MODELS["Mistral 7B (Narrate via HF)"] = {"id": "mistralai/Mistral-7B-Instruct-v0.2", "type": "hf_text"} # etc.
-if TEXT_MODELS: UI_DEFAULT_TEXT_MODEL_KEY = list(TEXT_MODELS.keys())[0] # Simplified default
+if AI_SERVICES_STATUS["gemini_text_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 AI_SERVICES_STATUS["hf_text_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:
+    if AI_SERVICES_STATUS["gemini_text_ready"] and "✨ Gemini 1.5 Flash (Narrate)" in TEXT_MODELS: UI_DEFAULT_TEXT_MODEL_KEY = "✨ Gemini 1.5 Flash (Narrate)"
+    elif AI_SERVICES_STATUS["hf_text_ready"] and "Mistral 7B (Narrate via HF)" in TEXT_MODELS: UI_DEFAULT_TEXT_MODEL_KEY = "Mistral 7B (Narrate via HF)"
+    else: 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
-# ... (Populate based on ai_services_status["dalle_image_ready"], ai_services_status["hf_image_ready"]) ...
-if ai_services_status["dalle_image_ready"]: IMAGE_PROVIDERS["🖼️ DALL-E 3"] = "dalle_3" #etc.
-if ai_services_status["hf_image_ready"]: IMAGE_PROVIDERS["🎡 HF - SDXL Base"] = "hf_sdxl_base" #etc.
-if IMAGE_PROVIDERS: UI_DEFAULT_IMAGE_PROVIDER_KEY = list(IMAGE_PROVIDERS.keys())[0] # Simplified default
+if AI_SERVICES_STATUS["dalle_image_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 AI_SERVICES_STATUS["hf_image_ready"]: 
+    IMAGE_PROVIDERS["🎡 HF - Stable Diffusion XL Base"] = "hf_sdxl_base" 
+    IMAGE_PROVIDERS["🎠 HF - OpenJourney"] = "hf_openjourney"
+    IMAGE_PROVIDERS["🌌 HF - Stable Diffusion v1.5"] = "hf_sd_1_5" 
+    UI_DEFAULT_IMAGE_PROVIDER_KEY = "🎡 HF - Stable Diffusion XL Base"
+    
+if not IMAGE_PROVIDERS: 
+    IMAGE_PROVIDERS["No Image Providers Configured"] = "none"
+    UI_DEFAULT_IMAGE_PROVIDER_KEY = "No Image Providers Configured"
+elif not UI_DEFAULT_IMAGE_PROVIDER_KEY and IMAGE_PROVIDERS : 
+    UI_DEFAULT_IMAGE_PROVIDER_KEY = list(IMAGE_PROVIDERS.keys())[0]
+
+# --- Custom CSS for Dark Theme "WOW" ---
+streamlit_omega_css = """
+<style>
+    body { color: #D0D0E0; background-color: #0F0F1A; font-family: 'Lexend Deca', sans-serif;}
+    .stApp { background-color: #0F0F1A; } 
+    h1, h2, h3, h4, h5, h6 { color: #C080F0; } 
+    /* Main containers */
+    .main .block-container { 
+        padding-top: 2rem; padding-bottom: 2rem; padding-left: 2rem; padding-right: 2rem; 
+        max-width: 1400px; margin: auto;
+        background-color: #1A1A2E; /* Panel background for main content area */
+        border-radius: 15px;
+        box-shadow: 0 8px 24px rgba(0,0,0,0.15);
+    }
+    /* Sidebar styling */
+    [data-testid="stSidebar"] {
+        background-color: #131325; /* Slightly different dark for sidebar */
+        border-right: 1px solid #2A2A4A;
+    }
+    [data-testid="stSidebar"] .stMarkdown h3 { /* Sidebar headers */
+        color: #D0D0FF !important;
+        font-size: 1.5em;
+        border-bottom: 2px solid #7F00FF;
+        padding-bottom: 5px;
+    }
+    /* Input elements */
+    .stTextInput > div > div > input, .stTextArea > div > div > textarea, 
+    .stSelectbox > div > div > div[data-baseweb="select"] > div,
+    div[data-baseweb="input"] > input /* For text_input */
+     { 
+        background-color: #2A2A4A !important; color: #E0E0FF !important; 
+        border: 1px solid #4A4A6A !important; border-radius: 8px !important;
+    }
+    /* Buttons */
+    .stButton > button {
+        background: linear-gradient(135deg, #7F00FF 0%, #E100FF 100%) !important;
+        color: white !important; border: none !important; border-radius: 8px !important;
+        padding: 0.7em 1.3em !important; font-weight: 600 !important;
+        box-shadow: 0 4px 8px rgba(0,0,0,0.15) !important;
+        transition: all 0.2s ease-in-out;
+        width: 100%; /* Make form submit button full width */
+    }
+    .stButton > button:hover { transform: scale(1.03) translateY(-1px); box-shadow: 0 8px 16px rgba(127,0,255,0.3) !important; }
+    .stButton > button:disabled { background: #4A4A6A !important; color: #8080A0 !important; cursor: not-allowed; }
+    
+    /* Secondary button style (if you add more buttons) */
+    /* You might need to assign a class and target it if Streamlit doesn't have variants like Gradio */
+    button[kind="secondary"] { /* Conceptual, Streamlit buttons don't have 'kind' directly in CSS */
+        background-color: #4A4A6A !important; color: #E0E0FF !important;
+    }
 
+    /* Image display */
+    .stImage > img { border-radius: 12px; box-shadow: 0 6px 15px rgba(0,0,0,0.25); max-height: 550px; margin: auto; display: block;}
+    /* Expander styling */
+    .stExpander { background-color: #161628; border: 1px solid #2A2A4A; border-radius: 12px; margin-bottom: 1em;}
+    .stExpander header { font-size: 1.1em; font-weight: 500; color: #D0D0FF;}
+    /* Note/Alert styling */
+    .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;}
+    /* Custom class for status messages */
+    .status-message { padding: 10px; border-radius: 6px; margin-top: 10px; text-align: center; font-weight: 500; }
+    .status-success { background-color: #104010; color: #B0FFB0; border: 1px solid #208020; }
+    .status-error { background-color: #401010; color: #FFB0B0; border: 1px solid #802020; }
+    .status-processing { background-color: #102040; color: #B0D0FF; border: 1px solid #204080; }
+    /* Gallery columns */
+    .gallery-col img { border-radius: 8px; box-shadow: 0 2px 6px rgba(0,0,0,0.15); margin-bottom: 5px;}
+    .gallery-col .stCaption { font-size: 0.85em; text-align: center; }
+</style>
+"""
+st.markdown(streamlit_omega_css, unsafe_allow_html=True)
 
-# --- Helper: Placeholder Image (can be same as before) ---
-@st.cache_data # Cache placeholder images
+# --- Helper: Placeholder Image Creation ---
+@st.cache_data 
 def create_placeholder_image_st(text="Processing...", size=(512, 512), color="#23233A", text_color="#E0E0FF"):
-    # ... (same PIL logic 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 
@@ -68,102 +169,78 @@ def create_placeholder_image_st(text="Processing...", size=(512, 512), color="#2
 if 'story_object' not in st.session_state:
     st.session_state.story_object = Story()
 if 'current_log' not in st.session_state:
-    st.session_state.current_log = ["Welcome to StoryVerse Weaver (Streamlit Edition)!"]
-if 'latest_scene_image' not in st.session_state:
-    st.session_state.latest_scene_image = None
+    st.session_state.current_log = ["Welcome to StoryVerse Omega (Streamlit Edition)!"]
+if 'latest_scene_image_pil' not in st.session_state:
+    st.session_state.latest_scene_image_pil = None
 if 'latest_scene_narrative' not in st.session_state:
-    st.session_state.latest_scene_narrative = "Describe your first scene to begin!"
+    st.session_state.latest_scene_narrative = "## ✨ A New Story Begins ✨\nDescribe your first scene idea in the panel on the left!"
+if 'status_message' not in st.session_state:
+    st.session_state.status_message = {"text": "Ready to weave your first masterpiece!", "type": "processing"}
 if 'processing_scene' not in st.session_state:
     st.session_state.processing_scene = False
-
-# --- Page Configuration (Do this ONCE at the top) ---
-st.set_page_config(
-    page_title="✨ StoryVerse Weaver ✨",
-    page_icon="🌌",
-    layout="wide", # "wide" or "centered"
-    initial_sidebar_state="expanded" # "auto", "expanded", "collapsed"
-)
-
-# --- Custom CSS for Dark Theme "WOW" ---
-# (You'd inject this using st.markdown(..., unsafe_allow_html=True) or a separate CSS file)
-streamlit_omega_css = """
-<style>
-    /* Base dark theme */
-    body { color: #D0D0E0; background-color: #0F0F1A; }
-    .stApp { background-color: #0F0F1A; }
-    h1, h2, h3, h4, h5, h6 { color: #C080F0; }
-    .stTextInput > div > div > input, .stTextArea > div > div > textarea, .stSelectbox > div > div > select {
-        background-color: #2A2A4A; color: #E0E0FF; border: 1px solid #4A4A6A; border-radius: 8px;
-    }
-    .stButton > button {
-        background: linear-gradient(135deg, #7F00FF 0%, #E100FF 100%) !important;
-        color: white !important; border: none !important; border-radius: 8px !important;
-        padding: 0.5em 1em !important; font-weight: 600 !important;
-        box-shadow: 0 4px 8px rgba(0,0,0,0.15) !important;
-    }
-    .stButton > button:hover { transform: scale(1.03) translateY(-1px); box-shadow: 0 8px 16px rgba(127,0,255,0.3) !important; }
-    /* Add more specific styles for sidebar, expanders, image display etc. */
-    .main .block-container { padding-top: 2rem; padding-bottom: 2rem; padding-left: 3rem; padding-right: 3rem; max-width: 1400px; margin: auto;}
-    .stImage > img { border-radius: 12px; box-shadow: 0 6px 15px rgba(0,0,0,0.25); max-height: 600px;}
-    .stExpander { background-color: #1A1A2E; border: 1px solid #2A2A4A; border-radius: 12px; margin-bottom: 1em;}
-    .stExpander header { font-size: 1.1em; font-weight: 500; color: #D0D0FF;}
-    .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;}
-</style>
-"""
-st.markdown(streamlit_omega_css, unsafe_allow_html=True)
+if 'form_scene_prompt' not in st.session_state: st.session_state.form_scene_prompt = ""
+if 'form_image_style' not in st.session_state: st.session_state.form_image_style = "Default (Cinematic Realism)"
+if 'form_artist_style' not in st.session_state: st.session_state.form_artist_style = ""
 
 
 # --- Main App UI & Logic ---
-st.markdown("<div align='center'><h1>✨ StoryVerse Weaver ✨</h1>\n<h3>Craft Immersive Multimodal Worlds with AI</h3></div>", unsafe_allow_html=True)
+st.markdown("<div align='center'><h1>✨ StoryVerse Omega ✨</h1>\n<h3>Craft Immersive Multimodal Worlds with AI</h3></div>", unsafe_allow_html=True)
 st.markdown("<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 your environment/secrets!</div>", unsafe_allow_html=True)
 
 
 # --- Sidebar for Inputs & Configuration ---
 with st.sidebar:
-    st.header("🎨 Scene Weaver Panel")
+    st.markdown("### 💡 **Craft Your Scene**")
     
     with st.form("scene_input_form"):
-        scene_prompt_text = st.text_area(
+        scene_prompt_text_input = st.text_area( # Unique key for widget
             "Scene Vision (Description, Dialogue, Action):", 
             height=200,
-            placeholder="e.g., Amidst swirling cosmic dust, Captain Eva pilots her damaged starfighter..."
+            value=st.session_state.form_scene_prompt, 
+            placeholder="e.g., Amidst swirling cosmic dust..."
         )
-        
-        st.subheader("Visual Style")
+        st.markdown("#### 🎨 Visual Style")
         col_style1, col_style2 = st.columns(2)
         with col_style1:
-            image_style_dropdown = st.selectbox("Style Preset:", options=["Default (Cinematic Realism)"] + sorted(list(STYLE_PRESETS.keys())), index=0)
+            image_style_dropdown_input = st.selectbox("Style Preset:", options=["Default (Cinematic Realism)"] + sorted(list(STYLE_PRESETS.keys())), 
+                                                index= (["Default (Cinematic Realism)"] + sorted(list(STYLE_PRESETS.keys()))).index(st.session_state.form_image_style) if st.session_state.form_image_style in (["Default (Cinematic Realism)"] + sorted(list(STYLE_PRESETS.keys()))) else 0, 
+                                                key="sel_image_style")
         with col_style2:
-            artist_style_text = st.text_input("Artistic Inspiration (Optional):", placeholder="e.g., Moebius")
+            artist_style_text_input = st.text_input("Artistic Inspiration (Optional):", placeholder="e.g., Moebius", value=st.session_state.form_artist_style, key="sel_artist_style")
         
-        negative_prompt_text = st.text_area("Exclude from Image (Negative Prompt):", value=COMMON_NEGATIVE_PROMPTS, height=100)
+        negative_prompt_text_input = st.text_area("Exclude from Image (Negative Prompt):", value=COMMON_NEGATIVE_PROMPTS, height=100, key="sel_negative_prompt")
 
         with st.expander("⚙️ Advanced AI Configuration", expanded=False):
-            text_model_key = st.selectbox("Narrative AI Engine:", options=list(TEXT_MODELS.keys()), index=0 if UI_DEFAULT_TEXT_MODEL_KEY in TEXT_MODELS else (list(TEXT_MODELS.keys()).index(UI_DEFAULT_TEXT_MODEL_KEY) if UI_DEFAULT_TEXT_MODEL_KEY else 0) )
-            image_provider_key = st.selectbox("Visual AI Engine:", options=list(IMAGE_PROVIDERS.keys()), index=0 if UI_DEFAULT_IMAGE_PROVIDER_KEY in IMAGE_PROVIDERS else (list(IMAGE_PROVIDERS.keys()).index(UI_DEFAULT_IMAGE_PROVIDER_KEY) if UI_DEFAULT_IMAGE_PROVIDER_KEY else 0) )
-            narrative_length = st.selectbox("Narrative Detail:", options=["Short (1 paragraph)", "Medium (2-3 paragraphs)", "Detailed (4+ paragraphs)"], index=1)
-            image_quality = st.selectbox("Image Detail/Style:", options=["Standard", "High Detail", "Sketch Concept"], index=0)
+            text_model_key_input = st.selectbox("Narrative AI Engine:", options=list(TEXT_MODELS.keys()), 
+                                          index=list(TEXT_MODELS.keys()).index(UI_DEFAULT_TEXT_MODEL_KEY) if UI_DEFAULT_TEXT_MODEL_KEY in TEXT_MODELS else 0, 
+                                          key="sel_text_model")
+            image_provider_key_input = st.selectbox("Visual AI Engine:", options=list(IMAGE_PROVIDERS.keys()), 
+                                              index=list(IMAGE_PROVIDERS.keys()).index(UI_DEFAULT_IMAGE_PROVIDER_KEY) if UI_DEFAULT_IMAGE_PROVIDER_KEY in IMAGE_PROVIDERS else 0, 
+                                              key="sel_image_provider")
+            narrative_length_input = st.selectbox("Narrative Detail:", options=["Short (1 paragraph)", "Medium (2-3 paragraphs)", "Detailed (4+ paragraphs)"], index=1, key="sel_narr_length")
+            image_quality_input = st.selectbox("Image Detail/Style:", options=["Standard", "High Detail", "Sketch Concept"], index=0, key="sel_img_quality")
         
         submit_scene_button = st.form_submit_button("🌌 Weave This Scene!", use_container_width=True, type="primary", disabled=st.session_state.processing_scene)
 
-    if st.button("🎲 Surprise Me!", use_container_width=True, disabled=st.session_state.processing_scene):
-        sur_prompt, sur_style, sur_artist = surprise_me_func() # Assuming this is defined as before
-        # Need to update the actual input widget values; Streamlit doesn't directly map outputs to inputs like Gradio's Examples
-        # This requires a more involved way to update widget states, or just display the suggestion.
-        # For simplicity, we'll just show what it would generate. A real app might use st.experimental_rerun or callbacks.
-        st.info(f"Surprise Idea: Prompt='{sur_prompt}', Style='{sur_style}', Artist='{sur_artist}'\n(Copy these into the fields above!)")
+    if st.button("🎲 Surprise Me!", use_container_width=True, disabled=st.session_state.processing_scene, key="surprise_btn_sidebar"):
+        sur_prompt, sur_style, sur_artist = surprise_me_func()
+        st.session_state.form_scene_prompt = sur_prompt
+        st.session_state.form_image_style = sur_style
+        st.session_state.form_artist_style = sur_artist
+        st.experimental_rerun()
 
-
-    if st.button("🗑️ New Story", use_container_width=True, disabled=st.session_state.processing_scene):
+    if st.button("🗑️ New Story", use_container_width=True, disabled=st.session_state.processing_scene, key="clear_btn_sidebar"):
         st.session_state.story_object = Story()
         st.session_state.current_log = ["Story Cleared. Ready for a new verse!"]
-        st.session_state.latest_scene_image = None
+        st.session_state.latest_scene_image_pil = None
         st.session_state.latest_scene_narrative = "## ✨ A New Story Begins ✨\nDescribe your first scene!"
-        st.experimental_rerun() # Rerun the script to refresh the UI
+        st.session_state.status_message = {"text": "📜 Story Cleared. A fresh canvas awaits!", "type": "processing"}
+        st.session_state.form_scene_prompt = "" # Clear form input too
+        st.experimental_rerun() 
 
     with st.expander("🔧 AI Services Status", expanded=False):
-        text_llm_ok, image_gen_ok = (ai_services_status["gemini_text_ready"] or ai_services_status["hf_text_ready"]), \
-                                    (ai_services_status["dalle_image_ready"] or ai_services_status["hf_image_ready"])
+        text_llm_ok, image_gen_ok = (AI_SERVICES_STATUS["gemini_text_ready"] or AI_SERVICES_STATUS["hf_text_ready"]), \
+                                    (AI_SERVICES_STATUS["dalle_image_ready"] or AI_SERVICES_STATUS["hf_image_ready"])
         if not text_llm_ok and not image_gen_ok: st.error("CRITICAL: NO AI SERVICES CONFIGURED.")
         else:
             if text_llm_ok: st.success("Text Generation Service(s) Ready.")
@@ -173,113 +250,125 @@ with st.sidebar:
 
 
 # --- Main Display Area ---
-st.markdown("---")
-st.markdown("### 🖼️ **Your Evolving StoryVerse**", unsafe_allow_html=True) # For potential custom class via CSS
+col_main_left, col_main_right = st.columns([2,3]) # Define columns for latest scene and gallery
 
-if st.session_state.processing_scene:
-    st.info("🌌 Weaving your scene... Please wait.")
-    # Could use st.spinner("Weaving your scene...")
-
-# Display Latest Scene
-if st.session_state.latest_scene_image or st.session_state.latest_scene_narrative != "Describe your first scene to begin!":
+with col_main_left:
     st.subheader("🌠 Latest Scene")
-    if st.session_state.latest_scene_image:
-        st.image(st.session_state.latest_scene_image, use_column_width=True, caption="Latest Generated Image")
-    st.markdown(st.session_state.latest_scene_narrative, unsafe_allow_html=True)
-    st.markdown("---")
-
+    if st.session_state.processing_scene and st.session_state.latest_scene_image_pil is None :
+        st.image(create_placeholder_image_st("🎨 Conjuring visuals..."), use_column_width="always")
+    elif st.session_state.latest_scene_image_pil:
+        st.image(st.session_state.latest_scene_image_pil, use_column_width="always", caption="Latest Generated Image")
+    else:
+        st.image(create_placeholder_image_st("Describe a scene to begin!", size=(512,300), color="#1A1A2E"), use_column_width="always")
+    
+    if st.session_state.processing_scene and "Musing narrative..." in st.session_state.latest_scene_narrative:
+         st.markdown(" Musing narrative...")
+    else:
+        st.markdown(st.session_state.latest_scene_narrative, unsafe_allow_html=True)
 
-# Display Story Scroll (Gallery)
-if st.session_state.story_object and st.session_state.story_object.scenes:
+with col_main_right:
     st.subheader("📚 Story Scroll")
-    # Streamlit doesn't have a direct "Gallery" like Gradio. We display images in columns.
-    num_columns = 3
-    cols = st.columns(num_columns)
-    scenes_for_gallery = st.session_state.story_object.get_all_scenes_for_gallery_display() # Ensure this returns (PIL.Image or None, caption)
-
-    for i, (img, caption) in enumerate(scenes_for_gallery):
-        with cols[i % num_columns]:
-            if img:
-                st.image(img, caption=caption if caption else f"Scene {i+1}", use_column_width=True)
-            elif caption: # If no image but caption (e.g. error)
-                st.caption(caption) # Display caption as text
-else:
-    st.caption("Your story scroll is empty. Weave your first scene!")
+    if st.session_state.story_object and st.session_state.story_object.scenes:
+        num_columns_gallery = 2 
+        gallery_cols = st.columns(num_columns_gallery)
+        scenes_for_gallery = st.session_state.story_object.get_all_scenes_for_gallery_display()
+        for i, (img, caption) in enumerate(scenes_for_gallery):
+            with gallery_cols[i % num_columns_gallery]:
+                if img: st.image(img, caption=caption if caption else f"Scene {i+1}", use_column_width="always", output_format="PNG") # Specify output format for PIL
+                elif caption: st.caption(caption)
+                else: st.caption(f"Scene {i+1} (No image)")
+    else:
+        st.caption("Your story scroll is empty. Weave your first scene in the panel on the left!")
+
+# Status Bar (displayed below main content)
+st.markdown("---")
+status_type = st.session_state.status_message.get("type", "processing")
+st.markdown(f"<p class='status-message status-{status_type}'>{st.session_state.status_message['text']}</p>", unsafe_allow_html=True)
 
 
 # Interaction Log
-with st.expander("⚙️ Interaction Log", expanded=False):
-    st.markdown("\n\n".join(st.session_state.current_log), unsafe_allow_html=True)
-
+with st.expander("⚙️ Interaction Log (Newest First)", expanded=False):
+    log_display = "\n\n---\n\n".join(st.session_state.current_log[::-1][:30]) 
+    st.markdown(log_display, unsafe_allow_html=True)
 
 # --- Logic for Form Submission ---
-if submit_scene_button and scene_prompt_text.strip(): # Check if form submitted and prompt is not empty
-    st.session_state.processing_scene = True
-    st.session_state.current_log.append(f"**🚀 New Scene Request - {time.strftime('%H:%M:%S')}**")
-    st.experimental_rerun() # Rerun to show "processing" state and disable button
-
-    # ---- This is where the main generation logic happens ----
-    # It's similar to add_scene_to_story_orchestrator but updates session_state
-    
-    # 1. Generate Narrative
-    current_narrative = f"Narrative Error: Init 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":
-        system_p = get_narrative_system_prompt("default")
-        prev_narrative = st.session_state.story_object.get_last_scene_narrative()
-        user_p = format_narrative_user_prompt(scene_prompt_text, prev_narrative)
-        st.session_state.current_log.append(f"  Narrative: Using {text_model_key} ({text_model_info['id']}).")
-        text_response = None
-        if text_model_info["type"] == "gemini" and ai_services_status["gemini_text_ready"]: 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" and ai_services_status["hf_text_ready"]: 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 submit_scene_button: 
+    if not scene_prompt_text_input.strip(): # Use the _input variable name
+        st.session_state.status_message = {"text": "Scene prompt cannot be empty!", "type": "error"}
+        # No explicit rerun here, warning will show, user can submit again.
+        # Or: st.warning("Scene prompt cannot be empty!"); st.stop()
+    else:
+        st.session_state.processing_scene = True
+        st.session_state.status_message = {"text": f"🌌 Weaving Scene {st.session_state.story_object.current_scene_number + 1}...", "type": "processing"}
+        st.session_state.current_log.append(f"**🚀 Scene {st.session_state.story_object.current_scene_number + 1} - {time.strftime('%H:%M:%S')}**")
+        # Values from widgets are automatically available by their keys
+        # e.g., scene_prompt_text_input, image_style_dropdown_input (from key="sel_image_style")
+        # However, for clarity, we re-fetch them here from their input widget keys.
         
-        if text_response and text_response.success: current_narrative = basic_text_cleanup(text_response.text); st.session_state.current_log.append(f"  Narrative: Success.")
-        elif text_response: current_narrative = f"**Narrative Error ({text_model_key}):** {text_response.error}"; st.session_state.current_log.append(f"  Narrative: FAILED - {text_response.error}")
-        else: st.session_state.current_log.append(f"  Narrative: FAILED - No response from {text_model_key}.")
-    else: current_narrative = "**Narrative Error:** Text model unavailable."; st.session_state.current_log.append(f"  Narrative: FAILED - Model '{text_model_key}' unavailable.")
-    
-    st.session_state.latest_scene_narrative = f"## Scene Idea: {scene_prompt_text}\n\n{current_narrative}"
-
-    # 2. Generate Image
-    generated_image_pil = None
-    image_gen_error = None
-    selected_image_provider_actual_type = IMAGE_PROVIDERS.get(image_provider_key)
-    image_content_prompt = current_narrative if current_narrative and "Error" not in current_narrative else scene_prompt_text
-    quality_kw = "ultra detailed, " if image_quality == "High Detail" else ("concept sketch, " if image_quality == "Sketch Concept" else "")
-    full_img_prompt = format_image_generation_prompt(quality_kw + image_content_prompt[:350], image_style_dropdown, artist_style_text)
-    st.session_state.current_log.append(f"  Image: Attempting with {image_provider_key} (type '{selected_image_provider_actual_type}').")
-
-    if selected_image_provider_actual_type and selected_image_provider_actual_type != "none":
-        img_response = None
-        if selected_image_provider_actual_type.startswith("dalle_") and ai_services_status["dalle_image_ready"]:
-            dalle_model = "dall-e-3" if selected_image_provider_actual_type == "dalle_3" else "dall-e-2"
-            img_response = generate_image_dalle(full_img_prompt, model=dalle_model, quality="hd" if image_quality=="High Detail" else "standard")
-        elif selected_image_provider_actual_type.startswith("hf_") and ai_services_status["hf_image_ready"]:
-            hf_model_id = "stabilityai/stable-diffusion-xl-base-1.0"; iw,ih=768,768
-            if selected_image_provider_actual_type == "hf_openjourney": hf_model_id="prompthero/openjourney";iw,ih=512,512
-            img_response = generate_image_hf_model(full_img_prompt, model_id=hf_model_id, negative_prompt=negative_prompt_text or COMMON_NEGATIVE_PROMPTS, width=iw, height=ih)
+        _scene_prompt = scene_prompt_text_input # From form
+        _image_style = image_style_dropdown_input # From form
+        _artist_style = artist_style_text_input # From form
+        _negative_prompt = negative_prompt_text_input # From form
+        _text_model = text_model_key_input # From form
+        _image_provider = image_provider_key_input # From form
+        _narr_length = narrative_length_input # From form
+        _img_quality = image_quality_input # From form
+
+        # ---- Main Generation Logic ----
+        current_narrative_text = f"Narrative Error: Init failed."
+        generated_image_pil = None
+        image_gen_error_msg = None
+        final_scene_error_msg = None
+
+        # 1. Generate Narrative
+        text_model_info = TEXT_MODELS.get(_text_model)
+        if text_model_info and text_model_info["type"] != "none":
+            system_p = get_narrative_system_prompt("default"); prev_narr = st.session_state.story_object.get_last_scene_narrative(); user_p = format_narrative_user_prompt(_scene_prompt, prev_narr)
+            st.session_state.current_log.append(f"  Narrative: Using {_text_model} ({text_model_info['id']}).")
+            text_resp = None
+            if text_model_info["type"] == "gemini" and AI_SERVICES_STATUS["gemini_text_ready"]: text_resp = generate_text_gemini(user_p, model_id=text_model_info["id"], system_prompt=system_p, max_tokens=768 if _narr_length.startswith("Detailed") else 400)
+            elif text_model_info["type"] == "hf_text" and AI_SERVICES_STATUS["hf_text_ready"]: text_resp = generate_text_hf(user_p, model_id=text_model_info["id"], system_prompt=system_p, max_tokens=768 if _narr_length.startswith("Detailed") else 400)
+            if text_resp and text_resp.success: current_narrative_text = basic_text_cleanup(text_resp.text); st.session_state.current_log.append("  Narrative: Success.")
+            elif text_resp: current_narrative_text = f"**Narrative Error:** {text_resp.error}"; st.session_state.current_log.append(f"  Narrative: FAILED - {text_resp.error}")
+            else: st.session_state.current_log.append(f"  Narrative: FAILED - No response.")
+        else: current_narrative_text = "**Narrative Error:** Model unavailable."; st.session_state.current_log.append(f"  Narrative: FAILED - Model '{_text_model}' unavailable.")
+        st.session_state.latest_scene_narrative = f"## Scene Idea: {_scene_prompt}\n\n{current_narrative_text}"
+
+        # 2. Generate Image
+        selected_img_prov_type = IMAGE_PROVIDERS.get(_image_provider)
+        img_content_prompt = current_narrative_text if current_narrative_text and "Error" not in current_narrative_text else _scene_prompt
+        quality_kw = "ultra detailed, " if _img_quality == "High Detail" else ("concept sketch, " if _img_quality == "Sketch Concept" else "")
+        full_img_prompt_for_gen = format_image_generation_prompt(quality_kw + img_content_prompt[:350], _image_style, _artist_style)
+        st.session_state.current_log.append(f"  Image: Using {_image_provider} (type '{selected_img_prov_type}').")
+        if selected_img_prov_type and selected_img_prov_type != "none":
+            img_resp = None
+            if selected_img_prov_type.startswith("dalle_") and AI_SERVICES_STATUS["dalle_image_ready"]:
+                dalle_model = "dall-e-3" if selected_img_prov_type == "dalle_3" else "dall-e-2"
+                img_resp = generate_image_dalle(full_img_prompt_for_gen, model=dalle_model)
+            elif selected_img_prov_type.startswith("hf_") and AI_SERVICES_STATUS["hf_image_ready"]:
+                hf_model_id = "stabilityai/stable-diffusion-xl-base-1.0" # Default
+                if selected_img_prov_type == "hf_openjourney": hf_model_id="prompthero/openjourney"
+                img_resp = generate_image_hf_model(full_img_prompt_for_gen, model_id=hf_model_id, negative_prompt=_negative_prompt)
+            if img_resp and img_resp.success: generated_image_pil = img_resp.image; st.session_state.current_log.append("  Image: Success.")
+            elif img_resp: image_gen_error_msg = f"**Image Error:** {img_resp.error}"; st.session_state.current_log.append(f"  Image: FAILED - {img_resp.error}")
+            else: image_gen_error_msg = "**Image Error:** No response."; st.session_state.current_log.append("  Image: FAILED - No response.")
+        else: image_gen_error_msg = "**Image Error:** No provider."; st.session_state.current_log.append(f"  Image: FAILED - No provider.")
+        st.session_state.latest_scene_image_pil = generated_image_pil if generated_image_pil else create_placeholder_image_st("Image Gen Failed", color="#401010")
+
+        # 3. Add Scene
+        if image_gen_error_msg and "**Narrative Error**" in current_narrative_text: final_scene_error_msg = f"{current_narrative_text}\n{image_gen_error_msg}" # etc.
+        elif "**Narrative Error**" in current_narrative_text: final_scene_error_msg = current_narrative_text
+        elif image_gen_error_msg: final_scene_error_msg = image_gen_error_msg
+        st.session_state.story_object.add_scene_from_elements(
+            user_prompt=_scene_prompt, narrative_text=current_narrative_text, image=generated_image_pil,
+            image_style_prompt=f"{_image_style}{f', by {_artist_style}' if _artist_style else ''}",
+            image_provider=_image_provider, error_message=final_scene_error_msg
+        )
+        st.session_state.current_log.append(f"  Scene {st.session_state.story_object.current_scene_number} processed.")
         
-        if img_response and img_response.success: generated_image_pil = img_response.image; st.session_state.current_log.append(f"  Image: Success from {img_response.provider}.")
-        elif img_response: image_gen_error = f"**Image Error:** {img_response.error}"; st.session_state.current_log.append(f"  Image: FAILED - {img_response.error}")
-        else: image_gen_error = "**Image Error:** No response/unknown issue."; st.session_state.current_log.append(f"  Image: FAILED - No response object.")
-    else: image_gen_error = "**Image Error:** No valid provider."; st.session_state.current_log.append(f"  Image: FAILED - No provider configured.")
-
-    st.session_state.latest_scene_image = generated_image_pil if generated_image_pil else create_placeholder_image("Image Gen Failed", color="#401010")
-
-    # 3. Add to Story Object
-    scene_err = None
-    if image_gen_error and "**Narrative Error**" in current_narrative: scene_err = f"{current_narrative}\n{image_gen_error}"
-    elif "**Narrative Error**" in current_narrative: scene_err = current_narrative
-    elif image_gen_error: scene_err = image_gen_error
-    
-    st.session_state.story_object.add_scene_from_elements(
-        user_prompt=scene_prompt_text, narrative_text=current_narrative, image=generated_image_pil,
-        image_style_prompt=f"{image_style_dropdown}{f', by {artist_style_text}' if artist_style_text else ''}",
-        image_provider=image_provider_key, error_message=scene_err
-    )
-    st.session_state.current_log.append(f"  Scene {st.session_state.story_object.current_scene_number} processed.")
-    st.session_state.processing_scene = False
-    st.experimental_rerun() # Rerun to update the main display with new scene and re-enable button
-
-elif submit_scene_button and not scene_prompt_text.strip(): # If form submitted but prompt is empty
-    st.warning("Please enter a scene vision/prompt!")
+        # 4. Set final status message
+        if final_scene_error_msg: st.session_state.status_message = {"text": f"Scene {st.session_state.story_object.current_scene_number} added with errors.", "type": "error"}
+        else: st.session_state.status_message = {"text": f"🌌 Scene {st.session_state.story_object.current_scene_number} woven!", "type": "success"}
+        
+        st.session_state.processing_scene = False
+        st.experimental_rerun()