Update app.py

app.py

@@ -18,7 +18,7 @@ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(level
 logger = logging.getLogger(__name__)
 
 # --- Global State Variables & API Key Setup ---
-def load_api_key(key_name_streamlit, key_name_env, service_name): # Same as previous
+def load_api_key(key_name_streamlit, key_name_env, service_name):
     key = None; secrets_available = hasattr(st, 'secrets')
     try:
         if secrets_available and key_name_streamlit in st.secrets:
@@ -31,22 +31,31 @@ def load_api_key(key_name_streamlit, key_name_env, service_name): # Same as prev
     if not key: logger.warning(f"{service_name} API Key NOT FOUND. Related features may be disabled or use fallbacks.")
     return key
 
-if 'services_initialized' not in st.session_state: # Same as previous
+if 'services_initialized' not in st.session_state:
     logger.info("Initializing services and API keys for the first time this session...")
     st.session_state.GEMINI_API_KEY = load_api_key("GEMINI_API_KEY", "GEMINI_API_KEY", "Gemini")
     st.session_state.OPENAI_API_KEY = load_api_key("OPENAI_API_KEY", "OPENAI_API_KEY", "OpenAI/DALL-E")
     st.session_state.ELEVENLABS_API_KEY = load_api_key("ELEVENLABS_API_KEY", "ELEVENLABS_API_KEY", "ElevenLabs")
     st.session_state.PEXELS_API_KEY = load_api_key("PEXELS_API_KEY", "PEXELS_API_KEY", "Pexels")
     st.session_state.ELEVENLABS_VOICE_ID_CONFIG = load_api_key("ELEVENLABS_VOICE_ID", "ELEVENLABS_VOICE_ID", "ElevenLabs Voice ID")
-    if not st.session_state.GEMINI_API_KEY: st.error("CRITICAL: Gemini API Key is essential and missing!"); logger.critical("Gemini API Key missing. Halting."); st.stop()
+    
+    if not st.session_state.GEMINI_API_KEY: 
+        st.error("CRITICAL: Gemini API Key is essential and missing!"); logger.critical("Gemini API Key missing. Halting."); st.stop()
+    
     try:
         st.session_state.gemini_handler = GeminiHandler(api_key=st.session_state.GEMINI_API_KEY)
         logger.info("GeminiHandler initialized successfully.")
     except Exception as e: st.error(f"Failed to init GeminiHandler: {e}"); logger.critical(f"GeminiHandler init failed: {e}", exc_info=True); st.stop()
+    
     try:
-        default_voice = "Rachel" 
-        st.session_state.visual_engine = VisualEngine(output_dir="temp_cinegen_media", default_elevenlabs_voice_id=st.session_state.ELEVENLABS_VOICE_ID_CONFIG or default_voice)
+        default_voice_id = "Rachel" # A common fallback if no secret is set
+        configured_voice_id = st.session_state.ELEVENLABS_VOICE_ID_CONFIG or default_voice_id
+        st.session_state.visual_engine = VisualEngine(
+            output_dir="temp_cinegen_media",
+            default_elevenlabs_voice_id=configured_voice_id # Pass it to __init__
+        )
         st.session_state.visual_engine.set_openai_api_key(st.session_state.OPENAI_API_KEY)
+        # set_elevenlabs_api_key will use the voice_id passed to __init__ or overridden by its own param
         st.session_state.visual_engine.set_elevenlabs_api_key(st.session_state.ELEVENLABS_API_KEY, voice_id_from_secret=st.session_state.ELEVENLABS_VOICE_ID_CONFIG)
         st.session_state.visual_engine.set_pexels_api_key(st.session_state.PEXELS_API_KEY)
         logger.info("VisualEngine initialized and API keys set (or attempted).")
@@ -55,14 +64,14 @@ if 'services_initialized' not in st.session_state: # Same as previous
         st.warning("VisualEngine critical setup issue. Some features will be disabled.")
     st.session_state.services_initialized = True; logger.info("Service initialization sequence complete.")
 
-for key, default_val in [ # Same
+# Initialize other session state variables
+for key, default_val in [
     ('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
 
-# --- Helper Functions --- (initialize_new_project, generate_visual_for_scene_core - same)
 def initialize_new_project(): # Same
     st.session_state.story_treatment_scenes, st.session_state.scene_dalle_prompts, st.session_state.generated_visual_paths = [], [], []
     st.session_state.video_path, st.session_state.overall_narration_audio_path, st.session_state.narration_script_display = None, None, ""
@@ -81,15 +90,16 @@ def generate_visual_for_scene_core(scene_index, scene_data, version=1): # Same
     else:
         st.session_state.generated_visual_paths[scene_index] = None; logger.warning(f"Visual generation FAILED for Scene {scene_data.get('scene_number', scene_index+1)}. img_path was: {img_path}"); return False
 
-# --- UI Sidebar --- (Same as previous, with unique keys for this pass if needed)
+# --- UI Sidebar ---
 with st.sidebar:
-    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...", height=120, key="user_idea_main_v5") # Key updated
-    genre = st.selectbox("Primary Genre:", ["Cyberpunk", "Sci-Fi", "Post-Apocalyptic"], index=2, key="genre_main_v5")
-    mood = st.selectbox("Overall Mood:", ["Hopeful yet Desperate", "Mysterious & Eerie"], index=0, key="mood_main_v5")
+    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_v5")
+    genre = st.selectbox("Primary Genre:", ["Cyberpunk", "Sci-Fi", "Fantasy", "Noir", "Thriller", "Western", "Post-Apocalyptic", "Historical Drama", "Surreal"], index=6, key="genre_main_v5")
+    mood = st.selectbox("Overall Mood:", ["Hopeful yet Desperate", "Mysterious & Eerie", "Gritty & Tense", "Epic & Awe-Inspiring", "Melancholy & Reflective", "Whimsical & Lighthearted"], index=0, key="mood_main_v5")
     num_scenes = st.slider("Number of Key Scenes:", 1, 3, 1, key="num_scenes_main_v5")
     creative_guidance_options = {"Standard Director": "standard", "Artistic Visionary": "more_artistic", "Experimental Storyteller": "experimental_narrative"}
-    selected_creative_guidance_key = st.selectbox("AI Creative Director Style:", list(creative_guidance_options.keys()), key="creative_guidance_select_v5")
+    selected_creative_guidance_key = st.selectbox("AI Creative Director Style:", options=list(creative_guidance_options.keys()), key="creative_guidance_select_v5")
     actual_creative_guidance = creative_guidance_options[selected_creative_guidance_key]
 
     if st.button("🌌 Generate Cinematic Treatment", type="primary", key="generate_treatment_btn_v5", use_container_width=True):
@@ -97,53 +107,62 @@ with st.sidebar:
         if not user_idea.strip(): st.warning("Please provide a story idea.")
         else:
             with st.status("AI Director is envisioning your masterpiece...", expanded=True) as status:
-                try: # Main generation try-except block
-                    status.write("Phase 1: Gemini crafting cinematic treatment..."); logger.info("Phase 1: Treatment Gen.")
+                try:
+                    status.write("Phase 1: Gemini crafting cinematic treatment... 📜"); logger.info("Phase 1: Cinematic Treatment Gen.")
                     treatment_prompt = create_cinematic_treatment_prompt(user_idea, genre, mood, num_scenes, actual_creative_guidance)
                     treatment_result_json = st.session_state.gemini_handler.generate_story_breakdown(treatment_prompt)
                     if not isinstance(treatment_result_json, list) or not treatment_result_json: raise ValueError("Gemini returned invalid scene list.")
                     st.session_state.story_treatment_scenes = treatment_result_json; 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
                     logger.info(f"Phase 1 complete. {num_gen_scenes} scenes."); status.update(label="Treatment complete! ✅ Generating visuals...", state="running")
-                    status.write("Phase 2: Creating visuals (DALL-E/Pexels)..."); logger.info("Phase 2: Visual Gen.")
+
+                    status.write("Phase 2: Creating visuals (DALL-E/Pexels)... 🖼️"); logger.info("Phase 2: Visual Gen.")
                     visual_successes = 0
                     for i, sc_data in enumerate(st.session_state.story_treatment_scenes):
                         sc_num_log = sc_data.get('scene_number', i+1)
                         status.write(f"  Visual for Scene {sc_num_log}..."); logger.info(f"  Processing visual for Scene {sc_num_log}.")
                         if generate_visual_for_scene_core(i, sc_data, version=1): visual_successes += 1
-                    current_status_label_ph2 = "Visuals ready! "; next_step_state = "running"
+                    
+                    current_status_label_ph2 = "Visuals ready! "
+                    next_step_state = "running" # Assume success unless visuals totally fail
                     if visual_successes == 0 and num_gen_scenes > 0: 
                         logger.error("Visual gen failed all scenes."); current_status_label_ph2 = "Visual gen FAILED for all scenes."; next_step_state="error"; 
-                        status.update(label=current_status_label_ph2, state=next_step_state, expanded=True); st.stop()
+                        status.update(label=current_status_label_ph2, state=next_step_state, expanded=True); st.stop() # Stop if no visuals
                     elif visual_successes < num_gen_scenes: 
                         logger.warning(f"Visuals partial ({visual_successes}/{num_gen_scenes})."); current_status_label_ph2 = f"Visuals partially generated ({visual_successes}/{num_gen_scenes}). "
-                    status.update(label=f"{current_status_label_ph2}Generating narration script...", state=next_step_state if next_step_state=="error" else "running")
-                    if next_step_state == "error": st.stop()
+                    status.update(label=f"{current_status_label_ph2}Generating narration script...", state=next_step_state) # Propagate error state if needed
+                    if next_step_state == "error": st.stop() # Should have already stopped
+
                     status.write("Phase 3: Generating narration script..."); logger.info("Phase 3: Narration Script Gen.")
-                    voice_style = st.session_state.get("selected_voice_style_for_generation", "cinematic_trailer")
-                    narr_prompt = create_narration_script_prompt_enhanced(st.session_state.story_treatment_scenes, mood, genre, voice_style)
+                    voice_style_for_prompt = st.session_state.get("selected_voice_style_for_generation", "cinematic_trailer")
+                    narr_prompt = create_narration_script_prompt_enhanced(st.session_state.story_treatment_scenes, mood, genre, voice_style_for_prompt)
                     st.session_state.narration_script_display = st.session_state.gemini_handler.generate_image_prompt(narr_prompt)
                     logger.info("Narration script generated."); status.update(label="Narration script ready! Synthesizing voice...", state="running")
-                    status.write("Phase 4: Synthesizing voice (ElevenLabs)..."); logger.info("Phase 4: Voice Synthesis.")
+                    
+                    status.write("Phase 4: Synthesizing voice (ElevenLabs)... 🔊"); logger.info("Phase 4: Voice Synthesis.")
                     st.session_state.overall_narration_audio_path = st.session_state.visual_engine.generate_narration_audio(st.session_state.narration_script_display)
-                    final_label = "All components ready! Storyboard below. 🚀"; final_state_val = "complete"
+                    
+                    final_label = "All components ready! Storyboard below. 🚀"
+                    final_state_val = "complete" 
                     if not st.session_state.overall_narration_audio_path:
-                        final_label = f"{current_status_label_ph2}Storyboard ready (Voiceover skipped/failed)."
+                        final_label = f"{current_status_label_ph2}Storyboard ready (Voiceover skipped or failed)."
                         logger.warning("Voiceover was skipped or failed.")
                     else: logger.info("Voiceover generated successfully.")
-                    status.update(label=final_label, state=final_state_val, expanded=False)
+                    status.update(label=final_label, state=final_state_val, expanded=False) # Always complete if this far
+
                 except ValueError as ve: logger.error(f"ValueError: {ve}", exc_info=True); status.update(label=f"Input or Gemini response error: {ve}", state="error", expanded=True); 
                 except Exception as e: logger.error(f"Unhandled Exception: {e}", exc_info=True); status.update(label=f"An unexpected error occurred: {e}", state="error", expanded=True);
     
     st.markdown("---"); st.markdown("### Fine-Tuning Options")
-    with st.expander("Define Characters", expanded=False): # Same
+    with st.expander("Define Characters", expanded=False):
         char_name = st.text_input("Character Name", key="char_name_adv_ultra_v5"); char_desc = st.text_area("Visual Description", key="char_desc_adv_ultra_v5", height=100, placeholder="e.g., Jax: rugged male astronaut...")
         if st.button("Save Character", key="add_char_adv_ultra_v5"):
             if char_name and char_desc: st.session_state.character_definitions[char_name.strip().lower()] = char_desc.strip(); st.success(f"Char '{char_name.strip()}' saved.")
             else: st.warning("Name and description needed.")
         if st.session_state.character_definitions: st.caption("Current Characters:"); [st.markdown(f"**{k.title()}:** _{v}_") for k,v in st.session_state.character_definitions.items()]
-    with st.expander("Global Style Overrides", expanded=False): # Same
-        presets = { "Default (Director's Choice)": "", "Hyper-Realistic Gritty Noir": "hyper-realistic gritty neo-noir...", "Surreal Dreamscape Fantasy": "surreal dreamscape...", "Vintage Analog Sci-Fi": "70s analog sci-fi..."}
+    
+    with st.expander("Global Style Overrides", expanded=False):
+        presets = { "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."}
         sel_preset = st.selectbox("Base Style Preset:", options=list(presets.keys()), key="style_preset_adv_ultra_v5")
         custom_kw = st.text_area("Additional Custom Style Keywords:", key="custom_style_adv_ultra_v5", height=80, placeholder="e.g., 'Dutch angle'")
         cur_style = st.session_state.global_style_additions
@@ -154,16 +173,31 @@ with st.sidebar:
             if cur_style: st.success("Global styles applied!")
             else: st.info("Global style additions cleared.")
         if cur_style: st.caption(f"Active global styles: \"{cur_style}\"")
-    with st.expander("Voice & Narration Style", expanded=False): # Same
-        default_voice_from_secret = st.session_state.get("ELEVENLABS_VOICE_ID_CONFIG", "Rachel")
-        user_voice_id_override = st.text_input("ElevenLabs Voice ID (optional):", value=default_voice_from_secret, key="el_voice_id_override_v5",help="Overrides secret/default.")
+
+    with st.expander("Voice & Narration Style", expanded=False):
+        # User can override the voice ID from secrets if they want to experiment
+        default_voice_from_engine = "Rachel" # Fallback if engine not init or no secret
+        if hasattr(st.session_state, 'visual_engine') and st.session_state.visual_engine:
+            default_voice_from_engine = st.session_state.visual_engine.elevenlabs_voice_id
+
+        user_voice_id_override = st.text_input(
+            "ElevenLabs Voice ID (optional override):", 
+            value=default_voice_from_engine, 
+            key="el_voice_id_override_v5",
+            help=f"Defaulting to '{default_voice_from_engine}' from secrets/config. Enter a specific Voice ID from your ElevenLabs account to override."
+        )
         prompt_v_styles = {"Cinematic Trailer": "cinematic_trailer", "Neutral Documentary": "documentary_neutral", "Character Introspection": "introspective_character"}
         sel_prompt_v_style_key = st.selectbox("Narration Script Style:", list(prompt_v_styles.keys()), key="narr_style_sel_v5", index=0)
+        
         if st.button("Set Narrator Voice & Style", key="set_voice_btn_ultra_v5"):
-            final_voice_id_to_use = user_voice_id_override.strip() or default_voice_from_secret
-            if hasattr(st.session_state, 'visual_engine'): st.session_state.visual_engine.elevenlabs_voice_id = final_voice_id_to_use
+            final_voice_id_to_use = user_voice_id_override.strip()
+            if not final_voice_id_to_use: # If user cleared the input, revert to the one from secrets/default
+                final_voice_id_to_use = st.session_state.get("ELEVENLABS_VOICE_ID_CONFIG", "Rachel")
+            
+            if hasattr(st.session_state, 'visual_engine'): 
+                st.session_state.visual_engine.elevenlabs_voice_id = final_voice_id_to_use
             st.session_state.selected_voice_style_for_generation = prompt_v_styles[sel_prompt_v_style_key]
-            st.success(f"Narrator Voice ID: {final_voice_id_to_use}. Script Style: {sel_prompt_v_style_key}")
+            st.success(f"Narrator Voice ID set to: {final_voice_id_to_use}. Script Style: {sel_prompt_v_style_key}")
             logger.info(f"User updated ElevenLabs Voice ID to: {final_voice_id_to_use}, Script Style: {sel_prompt_v_style_key}")
 
 # --- Main Content Area ---
@@ -175,10 +209,10 @@ if not st.session_state.story_treatment_scenes: st.info("Use the sidebar to gene
 else:
     for i_main, scene_content_display in enumerate(st.session_state.story_treatment_scenes):
         scene_n = scene_content_display.get('scene_number', i_main + 1); scene_t = scene_content_display.get('scene_title', 'Untitled')
-        key_base = f"s{scene_n}_{''.join(filter(str.isalnum, scene_t[:10]))}_v5" # Updated key for uniqueness
+        key_base = f"s{scene_n}_{''.join(filter(str.isalnum, scene_t[:10]))}_v5" 
         if "director_note" in scene_content_display and scene_content_display['director_note']: st.info(f"🎬 Director Note S{scene_n}: {scene_content_display['director_note']}")
         st.subheader(f"SCENE {scene_n}: {scene_t.upper()}"); col_d, col_v = st.columns([0.45, 0.55])
-        with col_d: # Scene Details (same display)
+        with col_d:
             with st.expander("📝 Scene Treatment", expanded=True):
                 st.markdown(f"**Beat:** {scene_content_display.get('emotional_beat', 'N/A')}"); st.markdown(f"**Setting:** {scene_content_display.get('setting_description', 'N/A')}"); st.markdown(f"**Chars:** {', '.join(scene_content_display.get('characters_involved', ['N/A']))}"); st.markdown(f"**Focus Moment:** _{scene_content_display.get('character_focus_moment', 'N/A')}_"); st.markdown(f"**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"**Dir. Visual Style:** _{scene_content_display.get('PROACTIVE_visual_style_감독', 'N/A')}_"); st.markdown(f"**Dir. Camera:** _{scene_content_display.get('PROACTIVE_camera_work_감독', 'N/A')}_"); st.markdown(f"**Dir. Sound:** _{scene_content_display.get('PROACTIVE_sound_design_감독', 'N/A')}_")
                 cur_d_prompt = st.session_state.scene_dalle_prompts[i_main] if i_main < len(st.session_state.scene_dalle_prompts) else None
@@ -186,7 +220,7 @@ else:
                      with st.popover("👁️ DALL-E Prompt"): st.markdown(f"**DALL-E Prompt:**"); st.code(cur_d_prompt, language='text')
                 pexels_q = scene_content_display.get('pexels_search_query_감독', None)
                 if pexels_q: st.caption(f"Pexels Fallback Query: `{pexels_q}`")
-        with col_v: # Image Display and Edit Popovers
+        with col_v:
             cur_img_p = st.session_state.generated_visual_paths[i_main] if i_main < len(st.session_state.generated_visual_paths) else None
             if cur_img_p and os.path.exists(cur_img_p): st.image(cur_img_p, caption=f"Scene {scene_n}: {scene_t}") 
             else:
@@ -202,17 +236,11 @@ else:
                                 updated_sc_data = st.session_state.gemini_handler.regenerate_scene_script_details(prompt_text)
                                 st.session_state.story_treatment_scenes[i_main] = updated_sc_data 
                                 s_treat_regen.update(label="Treatment updated! Regenerating visual...", state="running")
-                                
-                                # CORRECTED VERSIONING LOGIC
-                                v_num = 1 
-                                if cur_img_p and os.path.exists(cur_img_p):
-                                    try: 
-                                        base,_=os.path.splitext(os.path.basename(cur_img_p))
-                                        if '_v' in base: v_num = int(base.split('_v')[-1])+1
-                                        else: v_num = 2
-                                    except (ValueError, IndexError, TypeError): v_num = 2
-                                else: v_num = 1
-                                
+                                v_num = 1
+                                if cur_img_p and os.path.exists(cur_img_p): 
+                                    try: b,_=os.path.splitext(os.path.basename(cur_img_p)); v_num = int(b.split('_v')[-1])+1 if '_v' in b else 2
+                                    except (ValueError, IndexError, TypeError): v_num = 2 
+                                else: v_num = 1 
                                 if generate_visual_for_scene_core(i_main, updated_sc_data, version=v_num): s_treat_regen.update(label="Treatment & Visual Updated! 🎉", state="complete", expanded=False)
                                 else: s_treat_regen.update(label="Treatment updated, visual failed.", state="complete", expanded=False) 
                                 st.rerun()
@@ -232,17 +260,11 @@ else:
                                 refined_d_prompt = st.session_state.gemini_handler.generate_image_prompt(ref_req_prompt)
                                 st.session_state.scene_dalle_prompts[i_main] = refined_d_prompt 
                                 s_visual_regen.update(label="DALL-E prompt refined! Regenerating visual...", state="running")
-                                
-                                # CORRECTED VERSIONING LOGIC
                                 v_num = 1
                                 if cur_img_p and os.path.exists(cur_img_p): 
-                                    try: 
-                                        base,_=os.path.splitext(os.path.basename(cur_img_p))
-                                        if '_v' in base: v_num = int(base.split('_v')[-1])+1
-                                        else: v_num = 2
+                                    try: b,_=os.path.splitext(os.path.basename(cur_img_p)); v_num = int(b.split('_v')[-1])+1 if '_v' in b else 2
                                     except (ValueError, IndexError, TypeError): v_num=2
                                 else: v_num = 1
-                                
                                 if generate_visual_for_scene_core(i_main, scene_content_display, version=v_num): s_visual_regen.update(label="Visual Updated! 🎉", state="complete", expanded=False)
                                 else: s_visual_regen.update(label="Prompt refined, visual failed.", state="complete", expanded=False)
                                 st.rerun()
@@ -250,9 +272,8 @@ else:
                     else: st.warning("Please provide feedback.")
         st.markdown("---")
     
-    # Video Generation Button & Display (same logic)
     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_v5", type="primary", use_container_width=True): # Key updated
+        if st.button("🎬 Assemble Narrated Cinematic Animatic", key="assemble_ultra_video_btn_v5", type="primary", use_container_width=True):
             with st.status("Assembling Ultra Animatic...", expanded=True) as status_vid:
                 img_data_vid = []
                 for i_v, sc_c in enumerate(st.session_state.story_treatment_scenes):
@@ -274,7 +295,7 @@ else:
             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 Ultra Animatic", data=fp_dl, file_name=os.path.basename(st.session_state.video_path), mime="video/mp4", use_container_width=True, key="download_ultra_video_btn_v5" ) # Key updated
+                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_ultra_video_btn_v5" )
         except Exception as e: st.error(f"Error displaying video: {e}"); logger.error(f"Error displaying video: {e}", exc_info=True)
 
 # --- Footer ---