Update app.py

app.py

@@ -3,289 +3,316 @@ import streamlit as st
 from core.gemini_handler import GeminiHandler
 from core.visual_engine import VisualEngine 
 from core.prompt_engineering import (
-    create_story_breakdown_prompt,
-    create_image_prompt_from_scene_data,
-    create_narration_script_prompt, # New
-    create_scene_regeneration_prompt,
-    create_visual_regeneration_prompt
+    create_cinematic_treatment_prompt, # UPDATED
+    construct_dalle_prompt,           # UPDATED
+    create_narration_script_prompt_enhanced, # UPDATED
+    create_scene_regeneration_prompt,   # UPDATED
+    create_visual_regeneration_prompt   # UPDATED
 )
 import os
 
 # --- Configuration & Initialization ---
-st.set_page_config(page_title="CineGen AI Ultra", layout="wide", initial_sidebar_state="expanded")
+st.set_page_config(page_title="CineGen AI Ultra+", layout="wide", initial_sidebar_state="expanded")
 
 # --- Global State Variables & API Key Setup ---
-# Simplified API Key Loading
 def load_api_key(key_name_streamlit, key_name_env):
-    key = None
+    key = None; secrets_available = hasattr(st, 'secrets')
     try:
-        if key_name_streamlit in st.secrets: key = st.secrets[key_name_streamlit]
-    except AttributeError: print(f"st.secrets not available for {key_name_streamlit}.") # Local dev
-    except Exception as e: print(f"Error accessing st.secrets for {key_name_streamlit}: {e}")
+        if secrets_available and key_name_streamlit in st.secrets: key = st.secrets[key_name_streamlit]
+    except Exception: pass # Silently ignore if st.secrets access fails
     if not key and key_name_env in os.environ: key = os.environ[key_name_env]
     return key
 
-if 'GEMINI_API_KEY' not in st.session_state:
+# Initialize API Keys and handlers once
+if 'keys_loaded' not in st.session_state:
     st.session_state.GEMINI_API_KEY = load_api_key("GEMINI_API_KEY", "GEMINI_API_KEY")
-    if not st.session_state.GEMINI_API_KEY: st.error("Gemini API Key missing!"); st.stop()
-if 'gemini_handler' not in st.session_state:
+    st.session_state.OPENAI_API_KEY = load_api_key("OPENAI_API_KEY", "OPENAI_API_KEY")
+    st.session_state.ELEVENLABS_API_KEY = load_api_key("ELEVENLABS_API_KEY", "ELEVENLABS_API_KEY")
+    st.session_state.PEXELS_API_KEY = load_api_key("PEXELS_API_KEY", "PEXELS_API_KEY")
+    
+    if not st.session_state.GEMINI_API_KEY: st.error("Gemini API Key is essential and missing!"); st.stop()
+    
     st.session_state.gemini_handler = GeminiHandler(api_key=st.session_state.GEMINI_API_KEY)
-
-if 'visual_engine' not in st.session_state:
     st.session_state.visual_engine = VisualEngine(output_dir="temp_cinegen_media")
-    st.session_state.visual_engine.set_openai_api_key(load_api_key("OPENAI_API_KEY", "OPENAI_API_KEY"))
-    st.session_state.visual_engine.set_elevenlabs_api_key(load_api_key("ELEVENLABS_API_KEY", "ELEVENLABS_API_KEY"))
-    st.session_state.visual_engine.set_pexels_api_key(load_api_key("PEXELS_API_KEY", "PEXELS_API_KEY"))
+    st.session_state.visual_engine.set_openai_api_key(st.session_state.OPENAI_API_KEY)
+    st.session_state.visual_engine.set_elevenlabs_api_key(st.session_state.ELEVENLABS_API_KEY)
+    st.session_state.visual_engine.set_pexels_api_key(st.session_state.PEXELS_API_KEY)
+    st.session_state.keys_loaded = True
 
-# Initialize other session state variables (same as before)
+# Initialize other session state variables
 for key, default_val in [
-    ('story_scenes', []), ('scene_image_prompts', []), ('generated_images_paths', []),
-    ('video_path', None), ('character_definitions', {}), ('style_reference_description', ""),
-    ('overall_narration_path', None), ('narration_script_for_display', "")
+    ('story_treatment_scenes', []), ('scene_dalle_prompts', []), ('generated_visual_paths', []),
+    ('video_path', None), ('character_definitions', {}), ('global_style_additions', ""),
+    ('overall_narration_audio_path', None), ('narration_script_display', "")
 ]:
     if key not in st.session_state: st.session_state[key] = default_val
-# --- End State & API Key Setup ---
-
 
-def initialize_new_story(): # Remains same
-    st.session_state.story_scenes = []
-    st.session_state.scene_image_prompts = []
-    st.session_state.generated_images_paths = []
+# --- Helper Functions ---
+def initialize_new_project():
+    st.session_state.story_treatment_scenes = []
+    st.session_state.scene_dalle_prompts = []
+    st.session_state.generated_visual_paths = []
     st.session_state.video_path = None
-    st.session_state.overall_narration_path = None
-    st.session_state.narration_script_for_display = ""
+    st.session_state.overall_narration_audio_path = None
+    st.session_state.narration_script_display = ""
 
-
-def generate_visual_for_scene_wrapper(scene_index, scene_data, is_regeneration=False, version_count=1):
-    # ... (This helper remains the same, but now `generate_image_visual` in engine has Pexels fallback) ...
-    # Pass scene_data for Pexels query generation
-    scene_num_for_log = scene_data.get('scene_number', scene_index + 1)
-    textual_image_prompt = ""
-    if is_regeneration and scene_index < len(st.session_state.scene_image_prompts) and st.session_state.scene_image_prompts[scene_index]:
-        textual_image_prompt = st.session_state.scene_image_prompts[scene_index]
-    else:
-        textual_image_prompt = create_image_prompt_from_scene_data(
-            scene_data, st.session_state.character_definitions, st.session_state.style_reference_description)
-    if not textual_image_prompt: return False
-    if scene_index >= len(st.session_state.scene_image_prompts):
-        while len(st.session_state.scene_image_prompts) <= scene_index: st.session_state.scene_image_prompts.append("")
-    st.session_state.scene_image_prompts[scene_index] = textual_image_prompt
-    image_filename = f"scene_{scene_num_for_log}_visual_v{version_count}.png" # DALL-E saves as PNG
-    
-    # Pass scene_data to visual engine for Pexels query if DALL-E fails
-    generated_image_path = st.session_state.visual_engine.generate_image_visual(
-        textual_image_prompt, scene_data, image_filename 
+def generate_visual_for_scene_core(scene_index, scene_data, version=1):
+    # scene_data here is one scene from story_treatment_scenes
+    dalle_prompt = construct_dalle_prompt(
+        scene_data, 
+        st.session_state.character_definitions, 
+        st.session_state.global_style_additions
     )
-    while len(st.session_state.generated_images_paths) <= scene_index: st.session_state.generated_images_paths.append(None)
-    if generated_image_path and os.path.exists(generated_image_path):
-        st.session_state.generated_images_paths[scene_index] = generated_image_path; return True
+    if not dalle_prompt: return False
+    
+    # Ensure lists are long enough
+    while len(st.session_state.scene_dalle_prompts) <= scene_index: st.session_state.scene_dalle_prompts.append("")
+    while len(st.session_state.generated_visual_paths) <= scene_index: st.session_state.generated_visual_paths.append(None)
+        
+    st.session_state.scene_dalle_prompts[scene_index] = dalle_prompt
+    
+    filename = f"scene_{scene_data.get('scene_number', scene_index+1)}_visual_v{version}.png"
+    # Pass the full scene_data to visual_engine for Pexels query construction if needed
+    img_path = st.session_state.visual_engine.generate_image_visual(dalle_prompt, scene_data, filename)
+    
+    if img_path and os.path.exists(img_path):
+        st.session_state.generated_visual_paths[scene_index] = img_path; return True
     else:
-        st.session_state.generated_images_paths[scene_index] = None; return False
-
+        st.session_state.generated_visual_paths[scene_index] = None; return False
 
 # --- UI Sidebar ---
 with st.sidebar:
-    st.title("🎬 CineGen AI Ultra")
-    st.markdown("### Creative Controls")
-    # ... (user_idea, genre, mood, num_scenes - same) ...
-    user_idea = st.text_area("Core story idea:", "A lone astronaut discovers a glowing alien artifact on Mars, a sense of wonder and slight dread.", height=100, key="user_idea_input")
-    genre = st.selectbox("Genre:", ["Cyberpunk", "Sci-Fi", "Fantasy", "Noir", "Thriller"], index=0, key="genre_select")
-    mood = st.selectbox("Mood:", ["Wonder", "Suspenseful", "Mysterious", "Gritty", "Epic"], index=0, key="mood_select")
-    num_scenes_val = st.slider("Number of Scenes:", 1, 3, 1, key="num_scenes_slider")
+    st.title("🎬 CineGen AI Ultra+")
+    st.markdown("### Creative Seed")
+    user_idea = st.text_area("Core Story Idea / Theme:", "A lone wanderer searches for a mythical oasis in a vast, post-apocalyptic desert, haunted by mirages and mechanical scavengers.", height=120, key="user_idea_main")
+    genre = st.selectbox("Primary Genre:", ["Cyberpunk", "Sci-Fi", "Fantasy", "Noir", "Thriller", "Western", "Post-Apocalyptic"], index=6, key="genre_main")
+    mood = st.selectbox("Overall Mood:", ["Hopeful yet Desperate", "Mysterious & Eerie", "Gritty & Tense", "Epic & Awe-Inspiring", "Melancholy & Reflective"], index=0, key="mood_main")
+    num_scenes = st.slider("Number of Key Scenes:", 1, 4, 2, key="num_scenes_main") # Max 3-4 for API cost/time
 
+    creative_guidance_options = {"Standard": "standard", "More Artistic": "more_artistic", "Experimental Narrative": "experimental_narrative"}
+    selected_creative_guidance = st.selectbox("AI Creative Guidance Level:", options=list(creative_guidance_options.keys()), key="creative_guidance_select")
 
-    if st.button("🚀 Generate Cinematic Masterpiece", type="primary", key="generate_masterpiece_btn", use_container_width=True):
-        initialize_new_story()
-        if not user_idea.strip(): st.warning("Please enter a story idea.")
+    if st.button("🌌 Generate Cinematic Treatment", type="primary", key="generate_treatment_btn", use_container_width=True):
+        initialize_new_project()
+        if not user_idea.strip(): st.warning("Please provide a story idea.")
         else:
-            with st.status("Crafting your cinematic vision...", expanded=True) as status_main:
-                st.write("Phase 1: Gemini is drafting the script... 📜")
-                # ... (Story generation - same) ...
-                story_prompt_text = create_story_breakdown_prompt(user_idea, genre, mood, num_scenes_val)
+            with st.status("AI Director is envisioning your masterpiece...", expanded=True) as status:
+                status.write("Phase 1: Gemini crafting cinematic treatment (scenes, style, camera, sound)... 📜")
+                treatment_prompt = create_cinematic_treatment_prompt(user_idea, genre, mood, num_scenes, creative_guidance_options[selected_creative_guidance])
                 try:
-                    st.session_state.story_scenes = st.session_state.gemini_handler.generate_story_breakdown(story_prompt_text)
-                    status_main.update(label="Script complete! ✅ Generating visuals...", state="running")
-                    num_actual_scenes = len(st.session_state.story_scenes)
-                    st.session_state.scene_image_prompts = [""] * num_actual_scenes
-                    st.session_state.generated_images_paths = [None] * num_actual_scenes
-                except Exception as e: status_main.update(label=f"Script gen failed: {e}", state="error"); st.stop()
-                
-                if st.session_state.story_scenes:
-                    st.write("Phase 2: Creating visuals (DALL-E/Pexels)... 🖼️")
-                    visual_success_count = 0
-                    for i, scene_data in enumerate(st.session_state.story_scenes):
-                        st.write(f"  Processing Scene {scene_data.get('scene_number', i + 1)} visuals...")
-                        if generate_visual_for_scene_wrapper(i, scene_data, version_count=1): visual_success_count +=1
+                    st.session_state.story_treatment_scenes = st.session_state.gemini_handler.generate_story_breakdown(treatment_prompt) # Re-use for JSON list
+                    num_gen_scenes = len(st.session_state.story_treatment_scenes)
+                    st.session_state.scene_dalle_prompts = [""] * num_gen_scenes
+                    st.session_state.generated_visual_paths = [None] * num_gen_scenes
+                    status.update(label="Treatment complete! Generating visuals...", state="running")
+
+                    visual_successes = 0
+                    for i_scene, scene_content in enumerate(st.session_state.story_treatment_scenes):
+                        status.write(f"  Creating visual for Scene {scene_content.get('scene_number', i_scene+1)}: {scene_content.get('scene_title','')}...")
+                        if generate_visual_for_scene_core(i_scene, scene_content, version=1): visual_successes += 1
                     
-                    if visual_success_count == 0 : 
-                        status_main.update(label="Visual gen failed for all scenes.", state="error", expanded=False); st.stop()
+                    if visual_successes == 0 and num_gen_scenes > 0:
+                        status.update(label="Visual generation failed for all scenes. Check API keys/quota.", state="error", expanded=False); st.stop()
                     
-                    st.write("Phase 3: Generating narration script with Gemini... 🎤")
-                    narration_prompt = create_narration_script_prompt(st.session_state.story_scenes, mood, genre)
-                    try:
-                        narration_script = st.session_state.gemini_handler.generate_image_prompt(narration_prompt) # Re-use for text gen
-                        st.session_state.narration_script_for_display = narration_script
-                        status_main.update(label="Narration script ready! Generating voice...", state="running")
-                        
-                        st.write("Phase 4: Synthesizing voice with ElevenLabs... 🔊")
-                        st.session_state.overall_narration_path = st.session_state.visual_engine.generate_narration_audio(
-                            narration_script, "overall_narration.mp3"
-                        )
-                        if st.session_state.overall_narration_path:
-                             status_main.update(label="Voiceover generated! ✨", state="running")
-                        else:
-                            status_main.update(label="Voiceover failed, proceeding without.", state="warning")
-                    except Exception as e:
-                        status_main.update(label=f"Narration/Voice gen failed: {e}", state="warning")
+                    status.update(label="Visuals ready! Generating narration script...", state="running")
+                    narration_prompt = create_narration_script_prompt_enhanced(st.session_state.story_treatment_scenes, mood, genre, st.session_state.get("selected_voice_style", "cinematic_trailer"))
+                    narr_script = st.session_state.gemini_handler.generate_image_prompt(narration_prompt) # generate_image_prompt is just text gen
+                    st.session_state.narration_script_display = narr_script
+                    status.update(label="Narration script ready! Synthesizing voice...", state="running")
+                    
+                    st.session_state.overall_narration_audio_path = st.session_state.visual_engine.generate_narration_audio(narr_script)
+                    if st.session_state.overall_narration_audio_path: status.update(label="Voiceover ready! ✨", state="running")
+                    else: status.update(label="Voiceover failed. Video will be silent.", state="warning")
+
+                    status.update(label="All components ready! View storyboard. 🚀", state="complete", expanded=False)
 
-                    status_main.update(label="All major components generated! Storyboard ready. 🎉", state="complete", expanded=False)
+                except Exception as e: status.update(label=f"Error during generation: {e}", state="error", expanded=True); st.stop()
 
-    st.markdown("---") # Advanced Options (Character, Style - same as previous full app.py)
-    # ... (Character Consistency expander - same)
-    with st.expander("Character Consistency", expanded=False):
-        char_name_input = st.text_input("Character Name (e.g., Jax)", key="char_name_adv_input")
-        char_desc_input = st.text_area("Character Description (e.g., 'male astronaut, rugged, dark short hair, blue eyes, wearing a worn white and orange spacesuit with a cracked visor helmet slung at his hip')", key="char_desc_adv_input", height=120)
-        if st.button("Add/Update Character", key="add_char_adv_btn"):
-            if char_name_input and char_desc_input: st.session_state.character_definitions[char_name_input.strip().lower()] = char_desc_input.strip(); st.success(f"Character '{char_name_input.strip()}' defined.")
-            else: st.warning("Provide both name and description.")
+    st.markdown("---")
+    st.markdown("### Fine-Tuning Options")
+    # Character Definitions Expander (same as previous)
+    with st.expander("Define Characters", expanded=False):
+        # ... (UI for char_name_input, char_desc_input, add button - same as previous, ensure lowercase keys for definitions)
+        char_name_input = st.text_input("Character Name", key="char_name_adv_input_ultra")
+        char_desc_input = st.text_area("Detailed Visual Description", key="char_desc_adv_input_ultra", height=100, placeholder="e.g., Jax: rugged male astronaut, mid-40s, salt-and-pepper short hair, cybernetic left eye glowing faintly blue, wearing a patched-up crimson flight suit.")
+        if st.button("Save Character", key="add_char_adv_btn_ultra"):
+            if char_name_input and char_desc_input: st.session_state.character_definitions[char_name_input.strip().lower()] = char_desc_input.strip(); st.success(f"Character '{char_name_input.strip()}' saved.")
+            else: st.warning("Provide name and description.")
         if st.session_state.character_definitions:
-            st.caption("Defined Characters:")
-            for char_key, desc_val in st.session_state.character_definitions.items(): st.markdown(f"**{char_key.title()}:** _{desc_val}_")
-    # ... (Style Controls expander - same)
-    with st.expander("Style Controls", expanded=True):
-        predefined_styles = { "Default (Cinematic Photorealism)": "", "Gritty Neo-Noir": "neo-noir aesthetic, gritty realism, deep shadows...", "Epic Fantasy Matte Painting": "epic fantasy matte painting style...", "Impressionistic Dream": "impressionistic oil painting style...", "Vintage Sci-Fi (70s/80s Film)": "retro sci-fi aesthetic, analog film look...", "Modern Anime Key Visual": "high-detail modern anime key visual style..." }
-        selected_preset = st.selectbox("Base Style Preset:", options=list(predefined_styles.keys()), key="style_preset_select_adv")
-        custom_keywords = st.text_area("Add Custom Style Keywords:", key="custom_style_keywords_adv_input", height=80, placeholder="e.g., 'Dutch angle, fisheye lens'")
-        current_style_desc = st.session_state.style_reference_description
-        if st.button("Apply & Set Styles", key="apply_styles_adv_btn"):
+            st.caption("Current Characters:")
+            for k,v in st.session_state.character_definitions.items(): st.markdown(f"**{k.title()}:** _{v}_")
+
+    # Style Controls Expander (same as previous)
+    with st.expander("Global Style Overrides", expanded=False):
+        # ... (UI for predefined_styles, custom_style_keywords, apply button - same as previous) ...
+        predefined_styles = { "Default (Director's Choice)": "", "Hyper-Realistic Gritty Noir": "hyper-realistic gritty neo-noir, extreme detail, deep dynamic shadows, complex reflections on wet surfaces, cinematic film grain, desaturated palette with isolated vibrant neon accents (e.g. red, cyan), anamorphic lens distortion, atmospheric haze.", "Surreal Dreamscape Fantasy": "surreal dreamscape, epic fantasy elements, painterly with photorealistic details, impossible architecture, bioluminescent flora, otherworldly color palette (e.g., magenta skies, turquoise rivers), style of Roger Dean meets Zdzisław Beksiński.", "Vintage Analog Sci-Fi": "70s/80s analog sci-fi film aesthetic, tangible practical effects look, subtle light leaks, lens flares, warm filmic tones mixed with cool blues, detailed retro-futuristic technology with chunky buttons and CRT screens."}
+        selected_preset = st.selectbox("Base Style Preset:", options=list(predefined_styles.keys()), key="style_preset_select_adv_ultra")
+        custom_keywords = st.text_area("Additional Custom Style Keywords:", key="custom_style_keywords_adv_input_ultra", height=80, placeholder="e.g., 'shot with Arri Alexa, shallow depth of field, golden hour tones'")
+        current_style_desc = st.session_state.global_style_additions
+        if st.button("Apply Global Styles", key="apply_styles_adv_btn_ultra"):
             final_desc = predefined_styles[selected_preset]; 
             if custom_keywords.strip(): final_desc = f"{final_desc}, {custom_keywords.strip()}" if final_desc else custom_keywords.strip()
-            st.session_state.style_reference_description = final_desc.strip(); current_style_desc = final_desc.strip()
-            if current_style_desc: st.success("Styles applied!")
-            else: st.info("Default style (no specific additions).")
-        if current_style_desc: st.caption(f"Active style additions: \"{current_style_desc}\"")
-        else: st.caption("No specific style prompt additions active.")
-    # Conceptual Voice selection for ElevenLabs
+            st.session_state.global_style_additions = final_desc.strip(); current_style_desc = final_desc.strip()
+            if current_style_desc: st.success("Global styles applied!")
+            else: st.info("Global style additions cleared (using Director's per-scene choice).")
+        if current_style_desc: st.caption(f"Active global style additions: \"{current_style_desc}\"")
+
+    # Voice Customization Expander (same as previous)
     with st.expander("Voice Customization (ElevenLabs)", expanded=False):
-        # In a real app, you'd fetch voice list from ElevenLabs API if key is present
-        available_voices_conceptual = ["Rachel", "Adam", "Bella", "Antoni", "Elli", "Josh", "Arnold", "Callum"]
-        selected_voice = st.selectbox("Choose Narrator Voice:", available_voices_conceptual, 
-                                      index=available_voices_conceptual.index(st.session_state.visual_engine.elevenlabs_voice_id) if st.session_state.visual_engine.elevenlabs_voice_id in available_voices_conceptual else 0,
-                                      key="elevenlabs_voice_select")
-        if st.button("Set Voice", key="set_voice_btn"):
-            st.session_state.visual_engine.elevenlabs_voice_id = selected_voice
-            st.success(f"Narrator voice set to: {selected_voice}")
+        # ... (UI for available_voices_conceptual, selected_voice, set button - same) ...
+        # In a real app, fetch from ElevenLabs API if key is present. This is a placeholder list.
+        elevenlabs_voices = ["Rachel", "Adam", "Bella", "Antoni", "Elli", "Josh", "Arnold", "Domi", "Fin", "Sarah"] 
+        current_el_voice = st.session_state.visual_engine.elevenlabs_voice_id if hasattr(st.session_state, 'visual_engine') else "Rachel"
+        selected_el_voice = st.selectbox("Narrator Voice:", elevenlabs_voices, index=elevenlabs_voices.index(current_el_voice) if current_el_voice in elevenlabs_voices else 0, key="el_voice_select_ultra")
+        if st.button("Set Narrator Voice", key="set_voice_btn_ultra"):
+            if hasattr(st.session_state, 'visual_engine'): st.session_state.visual_engine.elevenlabs_voice_id = selected_el_voice
+            st.success(f"Narrator voice set to: {selected_el_voice}")
 
 
 # --- Main Content Area ---
-# ... (Header, Scene Iteration, Details, Image Display, Edit Popovers - mostly the same as previous full app.py)
-# Key change: Pass character_definitions and style_reference_description to create_visual_regeneration_prompt
+st.header("🎬 Cinematic Storyboard & Treatment")
 
-st.header("📝 Cinematic Storyboard")
-if st.session_state.narration_script_for_display: # Display narration script if generated
-    with st.expander("View Generated Narration Script", expanded=False):
-        st.markdown(st.session_state.narration_script_for_display)
+# Display Narration Script if available
+if st.session_state.narration_script_display:
+    with st.expander("📜 View Full Narration Script", expanded=False):
+        st.markdown(f"> _{st.session_state.narration_script_display}_")
 
-if not st.session_state.story_scenes:
-    st.info("Enter your idea in the sidebar and click '🚀 Generate Cinematic Masterpiece' to begin.")
+if not st.session_state.story_treatment_scenes:
+    st.info("Use the sidebar to generate your cinematic treatment.")
 else:
-    for i, scene_data_display in enumerate(st.session_state.story_scenes):
-        # ... (scene_num_display, unique_key_part, subheader, columns, Scene Details, Image Display - same as previous)
-        scene_num_display = scene_data_display.get('scene_number', i + 1); action_summary = scene_data_display.get('key_action', f"scene{i}"); key_part_raw = ''.join(e for e in action_summary if e.isalnum() or e.isspace()); unique_key_part = key_part_raw.replace(" ", "_")[:20]
-        st.subheader(f"Scene {scene_num_display}: {scene_data_display.get('emotional_beat', 'Untitled Scene')}")
-        col1, col2 = st.columns([2, 3])
-        with col1: 
-            with st.expander("Scene Details", expanded=True):
-                st.markdown(f"**Setting:** {scene_data_display.get('setting_description', 'N/A')}"); st.markdown(f"**Characters:** {', '.join(scene_data_display.get('characters_involved', []))}"); st.markdown(f"**Key Action:** {scene_data_display.get('key_action', 'N/A')}"); st.markdown(f"**Dialogue Snippet:** `\"{scene_data_display.get('dialogue_snippet', '...')}\"`"); st.markdown(f"**Visual Style Suggestion (Gemini):** {scene_data_display.get('visual_style_suggestion', 'N/A')}"); st.markdown(f"**Camera Suggestion (Gemini):** {scene_data_display.get('camera_angle_suggestion', 'N/A')}")
-                current_textual_prompt = st.session_state.scene_image_prompts[i] if i < len(st.session_state.scene_image_prompts) else None
-                if current_textual_prompt:
-                     with st.popover("View Full Image Prompt"): st.markdown(f"**Full Textual Prompt Sent to DALL-E:**"); st.code(current_textual_prompt, language='text')
-        with col2: 
-            current_image_path = st.session_state.generated_images_paths[i] if i < len(st.session_state.generated_images_paths) else None
-            if current_image_path and os.path.exists(current_image_path): st.image(current_image_path, caption=f"Visual Concept for Scene {scene_num_display}") 
+    for i_main, scene_content_display in enumerate(st.session_state.story_treatment_scenes):
+        scene_num = scene_content_display.get('scene_number', i_main + 1)
+        scene_title = scene_content_display.get('scene_title', 'Untitled Scene')
+        unique_key_base = f"scene_{scene_num}_{''.join(filter(str.isalnum, scene_title[:10]))}" # For unique widget keys
+
+        st.subheader(f"SCENE {scene_num}: {scene_title.upper()}")
+        
+        # Display Director's Notes if any (for experimental narrative)
+        if "director_note" in scene_content_display:
+            st.info(f"🎬 Director's Note for Scene {scene_num}: {scene_content_display['director_note']}")
+
+        col_details, col_visual = st.columns([0.45, 0.55]) # Adjust ratio
+
+        with col_details:
+            with st.expander("📝 Scene Treatment Details", expanded=True):
+                st.markdown(f"**Emotional Beat:** {scene_content_display.get('emotional_beat', 'N/A')}")
+                st.markdown(f"**Setting:** {scene_content_display.get('setting_description', 'N/A')}")
+                st.markdown(f"**Characters:** {', '.join(scene_content_display.get('characters_involved', ['N/A']))}")
+                st.markdown(f"**Character Focus Moment:** _{scene_content_display.get('character_focus_moment', 'N/A')}_")
+                st.markdown(f"**Key Plot Beat:** {scene_content_display.get('key_plot_beat', 'N/A')}")
+                st.markdown(f"**Dialogue Hook:** `\"{scene_content_display.get('suggested_dialogue_hook', '...')}\"`")
+                st.markdown("---")
+                st.markdown(f"**🎬 Director's Visual Style:** {scene_content_display.get('PROACTIVE_visual_style_감독', 'N/A')}")
+                st.markdown(f"**🎥 Director's Camera Work:** {scene_content_display.get('PROACTIVE_camera_work_감독', 'N/A')}")
+                st.markdown(f"**🔊 Director's Sound Design:** {scene_content_display.get('PROACTIVE_sound_design_감독', 'N/A')}")
+                
+                current_dalle_prompt = st.session_state.scene_dalle_prompts[i_main] if i_main < len(st.session_state.scene_dalle_prompts) else None
+                if current_dalle_prompt:
+                     with st.popover("👁️ View DALL-E Prompt"): 
+                        st.markdown(f"**Full DALL-E Prompt:**"); st.code(current_dalle_prompt, language='text')
+                
+                pexels_query_display = scene_content_display.get('pexels_search_query_감독', None)
+                if pexels_query_display:
+                    st.caption(f"Suggested Pexels Query: `{pexels_query_display}`")
+
+
+        with col_visual:
+            current_img_path = st.session_state.generated_visual_paths[i_main] if i_main < len(st.session_state.generated_visual_paths) else None
+            if current_img_path and os.path.exists(current_img_path):
+                st.image(current_img_path, caption=f"Visual Concept for Scene {scene_num}: {scene_title}", use_column_width='always')
             else:
-                if st.session_state.story_scenes: st.caption("Visual for this scene is pending or failed.")
-            with st.popover(f"✏️ Edit Scene {scene_num_display} Script"): 
-                # ... (Script edit logic - same, but uses updated generate_visual_for_scene_wrapper)
-                feedback_script = st.text_area("Describe script changes:", key=f"script_feedback_{unique_key_part}_{i}", height=100) 
-                if st.button(f"🔄 Update Scene {scene_num_display} Script", key=f"regen_script_btn_{unique_key_part}_{i}"): 
-                    if feedback_script:
-                        with st.status(f"Updating Scene {scene_num_display}...", expanded=True) as status_script_regen:
-                            regen_prompt = create_scene_regeneration_prompt(scene_data_display, feedback_script, st.session_state.story_scenes)
+                st.caption("Visual concept pending or failed.")
+
+            # Edit Popovers - logic for calling regeneration needs to use the new prompt structures
+            with st.popover(f"✏️ Edit Scene {scene_num} Treatment"): 
+                feedback_script_edit = st.text_area("Describe changes to treatment details:", key=f"treat_feed_{unique_key_base}", height=150) 
+                if st.button(f"🔄 Update Scene {scene_num} Treatment", key=f"regen_treat_btn_{unique_key_base}"): 
+                    if feedback_script_edit:
+                        with st.status(f"Updating Scene {scene_num} Treatment...", expanded=True) as status_treat_regen:
+                            regen_prompt_text = create_scene_regeneration_prompt(scene_content_display, feedback_script_edit, st.session_state.story_treatment_scenes)
                             try:
-                                updated_scene_data = st.session_state.gemini_handler.regenerate_scene_script_details(regen_prompt); st.session_state.story_scenes[i] = updated_scene_data 
-                                status_script_regen.update(label="Script updated! Regenerating visual...", state="running")
-                                version = 1
-                                if current_image_path: try: base, _ = os.path.splitext(os.path.basename(current_image_path)); version = int(base.split('_v')[-1]) + 1 if '_v' in base else 2 except: version = 2
-                                if generate_visual_for_scene_wrapper(i, updated_scene_data, is_regeneration=False, version_count=version): status_script_regen.update(label="Scene script & visual updated! 🎉", state="complete", expanded=False)
-                                else: status_script_regen.update(label="Script updated, visual failed.", state="warning", expanded=False)
+                                updated_scene_data = st.session_state.gemini_handler.regenerate_scene_script_details(regen_prompt_text) # Assumes this returns one scene obj
+                                st.session_state.story_treatment_scenes[i_main] = updated_scene_data 
+                                status_treat_regen.update(label="Treatment updated! Regenerating visual & DALL-E prompt...", state="running")
+                                version_num = 1
+                                if current_img_path: try: base,_=os.path.splitext(os.path.basename(current_img_path)); version_num = int(base.split('_v')[-1])+1 if '_v' in base else 2 except: version_num=2
+                                
+                                # Regenerate visual for the updated scene data
+                                if generate_visual_for_scene_core(i_main, updated_scene_data, version=version_num):
+                                    status_treat_regen.update(label="Scene Treatment & Visual Updated! 🎉", state="complete", expanded=False)
+                                else: status_treat_regen.update(label="Treatment updated, visual failed.", state="warning", expanded=False)
                                 st.rerun()
-                            except Exception as e: status_script_regen.update(label=f"Error: {e}", state="error")
-                    else: st.warning("Please provide feedback.")
-            with st.popover(f"🎨 Edit Scene {scene_num_display} Visuals"): 
-                # ... (Visual edit logic - same, but passes more context to create_visual_regeneration_prompt)
-                prompt_for_edit_visuals = st.session_state.scene_image_prompts[i] if i < len(st.session_state.scene_image_prompts) else "No prompt."
-                st.caption("Current Full Image Prompt:"); st.code(prompt_for_edit_visuals, language='text')
-                feedback_visual = st.text_area("Describe visual changes for the prompt:", key=f"visual_feedback_{unique_key_part}_{i}", height=100) 
-                if st.button(f"🔄 Update Scene {scene_num_display} Visuals", key=f"regen_visual_btn_{unique_key_part}_{i}"): 
-                    if feedback_visual:
-                        with st.status(f"Refining prompt & regenerating visual...", expanded=True) as status_visual_regen:
-                            prompt_refinement_request = create_visual_regeneration_prompt(
-                                prompt_for_edit_visuals, feedback_visual, scene_data_display, 
-                                st.session_state.character_definitions, st.session_state.style_reference_description # Pass these
+                            except Exception as e: status_treat_regen.update(label=f"Error: {e}", state="error")
+                    else: st.warning("Please provide feedback for treatment regeneration.")
+
+            with st.popover(f"🎨 Edit Scene {scene_num} Visual Prompt"): 
+                dalle_prompt_to_edit = st.session_state.scene_dalle_prompts[i_main] if i_main < len(st.session_state.scene_dalle_prompts) else "No DALL-E prompt."
+                st.caption("Current DALL-E Prompt:"); st.code(dalle_prompt_to_edit, language='text')
+                feedback_visual_edit = st.text_area("Describe changes for the DALL-E prompt:", key=f"visual_feed_{unique_key_base}", height=150) 
+                if st.button(f"🔄 Update Scene {scene_num} Visual Prompt & Image", key=f"regen_visual_btn_{unique_key_base}"): 
+                    if feedback_visual_edit:
+                        with st.status(f"Refining DALL-E prompt & regenerating visual...", expanded=True) as status_visual_edit_regen:
+                            # Gemini refines the DALL-E prompt
+                            refinement_req_prompt = create_visual_regeneration_prompt(
+                                dalle_prompt_to_edit, feedback_visual_edit, scene_content_display, 
+                                st.session_state.character_definitions, st.session_state.global_style_additions
                             )
                             try:
-                                refined_textual_image_prompt = st.session_state.gemini_handler.regenerate_image_prompt_from_feedback(prompt_refinement_request)
-                                st.session_state.scene_image_prompts[i] = refined_textual_image_prompt 
-                                status_visual_regen.update(label="Prompt refined! Regenerating visual...", state="running")
-                                version = 1
-                                if current_image_path: try: base, _ = os.path.splitext(os.path.basename(current_image_path)); version = int(base.split('_v')[-1]) + 1 if '_v' in base else 2 except: version = 2
-                                if generate_visual_for_scene_wrapper(i, scene_data_display, is_regeneration=True, version_count=version): status_visual_regen.update(label="Visual updated! 🎉", state="complete", expanded=False)
-                                else: status_visual_regen.update(label="Prompt refined, visual failed.", state="warning", expanded=False)
+                                refined_dalle_prompt = st.session_state.gemini_handler.generate_image_prompt(refinement_req_prompt) # generate_image_prompt is just text gen
+                                st.session_state.scene_dalle_prompts[i_main] = refined_dalle_prompt # Update stored DALL-E prompt
+                                status_visual_edit_regen.update(label="DALL-E prompt refined! Regenerating visual...", state="running")
+                                version_num = 1
+                                if current_img_path: try: base,_=os.path.splitext(os.path.basename(current_img_path)); version_num = int(base.split('_v')[-1])+1 if '_v' in base else 2 except: version_num=2
+                                
+                                # Generate visual with the NEW refined DALL-E prompt
+                                if generate_visual_for_scene_core(i_main, scene_content_display, version=version_num): # scene_content_display is still relevant context
+                                    status_visual_edit_regen.update(label="Visual Updated! 🎉", state="complete", expanded=False)
+                                else: status_visual_edit_regen.update(label="Prompt refined, visual failed.", state="warning", expanded=False)
                                 st.rerun()
-                            except Exception as e: status_visual_regen.update(label=f"Error: {e}", state="error")
-                    else: st.warning("Please provide feedback.")
+                            except Exception as e: status_visual_edit_regen.update(label=f"Error: {e}", state="error")
+                    else: st.warning("Please provide feedback for visual prompt regeneration.")
         st.markdown("---")
     
-    # Video Generation - Pass overall_narration_path
-    if st.session_state.story_scenes and any(p for p in st.session_state.generated_images_paths if p is not None):
-        if st.button("🎬 Assemble Narrated Masterpiece", key="assemble_narrated_video_btn", type="primary", use_container_width=True):
-            with st.status("Assembling narrated animatic video...", expanded=True) as status_video:
-                image_data_for_video = []
-                # ... (image_data_for_video assembly - same) ...
-                for idx, scene_info in enumerate(st.session_state.story_scenes):
-                    img_path = st.session_state.generated_images_paths[idx] if idx < len(st.session_state.generated_images_paths) else None
-                    if img_path and os.path.exists(img_path):
-                        image_data_for_video.append({'path':img_path, 'scene_num':scene_info.get('scene_number',idx+1), 'key_action':scene_info.get('key_action','')})
-                        st.write(f"Adding Scene {scene_info.get('scene_number', idx + 1)} to video.")
+    # Video Generation Button
+    if st.session_state.story_treatment_scenes and any(p for p in st.session_state.generated_visual_paths if p is not None):
+        if st.button("🎬 Assemble Narrated Cinematic Animatic", key="assemble_ultra_video_btn", type="primary", use_container_width=True):
+            with st.status("Assembling Ultra Animatic...", expanded=True) as status_vid:
+                image_data_for_vid = []
+                for i_vid, scene_c in enumerate(st.session_state.story_treatment_scenes):
+                    img_p = st.session_state.generated_visual_paths[i_vid] if i_vid < len(st.session_state.generated_visual_paths) else None
+                    if img_p and os.path.exists(img_p):
+                        image_data_for_vid.append({
+                            'path':img_p, 'scene_num':scene_c.get('scene_number',i_vid+1), 
+                            'key_action':scene_c.get('key_plot_beat','') # Use key_plot_beat for overlay
+                        }); status_vid.write(f"Adding Scene {scene_c.get('scene_number', i_vid + 1)} to video.")
                 
-                if image_data_for_video:
-                    st.write("Calling video engine (with narration if available)...")
+                if image_data_for_vid:
+                    status_vid.write("Calling video engine...")
                     st.session_state.video_path = st.session_state.visual_engine.create_video_from_images(
-                        image_data_for_video, 
-                        overall_narration_path=st.session_state.overall_narration_path, # Pass narration
-                        output_filename="cinegen_narrated_masterpiece.mp4",
-                        duration_per_image=5, # Longer for narration
+                        image_data_for_vid, 
+                        overall_narration_path=st.session_state.overall_narration_audio_path,
+                        output_filename="cinegen_ultra_animatic.mp4",
+                        duration_per_image=st.session_state.visual_engine.video_overlay_font_size * 0.15, # Dynamic duration based on text overlay font size
                         fps=24 
                     )
                     if st.session_state.video_path and os.path.exists(st.session_state.video_path):
-                        status_video.update(label="Narrated masterpiece assembled! 🎉", state="complete", expanded=False); st.balloons()
-                    else: status_video.update(label="Video assembly failed. Check logs.", state="error", expanded=False)
-                else: status_video.update(label="No valid images to assemble video.", state="error", expanded=False)
-    elif st.session_state.story_scenes: st.info("Generate visuals for scenes before assembling the video.")
+                        status_vid.update(label="Ultra animatic assembled! 🎉", state="complete", expanded=False); st.balloons()
+                    else: status_vid.update(label="Video assembly failed. Check logs.", state="error", expanded=False)
+                else: status_vid.update(label="No valid images for video.", state="error", expanded=False)
+    elif st.session_state.story_treatment_scenes: st.info("Generate visuals before assembling video.")
 
-    # Video display and download - (same)
+    # Video display and download
     if st.session_state.video_path and os.path.exists(st.session_state.video_path):
-        st.header("🎬 Generated Narrated Masterpiece")
+        st.header("🎬 Generated Cinematic Animatic")
         try:
             with open(st.session_state.video_path, 'rb') as vf_obj: video_bytes = vf_obj.read()
             st.video(video_bytes, format="video/mp4") 
             with open(st.session_state.video_path, "rb") as fp_dl:
-                st.download_button(label="Download Narrated Masterpiece", data=fp_dl,
+                st.download_button(label="Download Ultra Animatic", data=fp_dl,
                     file_name=os.path.basename(st.session_state.video_path), mime="video/mp4",
-                    use_container_width=True, key="download_narrated_video_btn" )
+                    use_container_width=True, key="download_ultra_video_btn" )
         except Exception as e: st.error(f"Error displaying video: {e}")
 
 # --- Footer ---
 st.sidebar.markdown("---")
-st.sidebar.caption("CineGen AI Ultra | Powered by Gemini, DALL-E, ElevenLabs & Streamlit.")
+st.sidebar.caption("CineGen AI Ultra+ | Visionary Cinematic Pre-Production")