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
+def load_api_key(key_name_streamlit, key_name_env, service_name): # Same as previous
     key = None; secrets_available = hasattr(st, 'secrets')
     try:
         if secrets_available and key_name_streamlit in st.secrets:
@@ -31,36 +31,23 @@ def load_api_key(key_name_streamlit, key_name_env, service_name): # Same
     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:
+if 'services_initialized' not in st.session_state: # Same as previous
     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")
-    # --- NEW: Load ElevenLabs Voice ID ---
     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:
-        # Pass default voice ID to VisualEngine, it will be overridden if ELEVENLABS_VOICE_ID_CONFIG is found
-        default_voice = "Rachel" # A common default
-        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 = "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)
         st.session_state.visual_engine.set_openai_api_key(st.session_state.OPENAI_API_KEY)
-        # Pass the API key AND the configured voice ID (if any) to set_elevenlabs_api_key
-        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 # This might be None
-        )
+        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).")
     except Exception as e:
@@ -68,8 +55,7 @@ if 'services_initialized' not in st.session_state:
         st.warning("VisualEngine critical setup issue. Some features will be disabled.")
     st.session_state.services_initialized = True; logger.info("Service initialization sequence complete.")
 
-# Initialize other session state variables (same)
-for key, default_val in [
+for key, default_val in [ # Same
     ('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', "")
@@ -95,30 +81,27 @@ 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 ---
-with st.sidebar: # Voice Customization UI slightly changed
-    st.title("🎬 CineGen AI Ultra+")
-    st.markdown("### Creative Seed")
-    # ... (user_idea, genre, mood, num_scenes, creative_guidance - same) ...
-    user_idea = st.text_area("Core Story Idea / Theme:", "A lone wanderer searches for a mythical oasis...", height=120, key="user_idea_main_v4")
-    genre = st.selectbox("Primary Genre:", ["Cyberpunk", "Sci-Fi", "Fantasy", "Noir", "Thriller", "Western", "Post-Apocalyptic"], index=6, key="genre_main_v4")
-    mood = st.selectbox("Overall Mood:", ["Hopeful yet Desperate", "Mysterious & Eerie", "Gritty & Tense"], index=0, key="mood_main_v4")
-    num_scenes = st.slider("Number of Key Scenes:", 1, 3, 1, key="num_scenes_main_v4")
+# --- UI Sidebar --- (Same as previous, with unique keys for this pass if needed)
+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")
+    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:", options=list(creative_guidance_options.keys()), key="creative_guidance_select_v4")
+    selected_creative_guidance_key = st.selectbox("AI Creative Director Style:", 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_v4", use_container_width=True):
-        # ... (Main generation flow - same logic for calling phases) ...
+    if st.button("🌌 Generate Cinematic Treatment", type="primary", key="generate_treatment_btn_v5", use_container_width=True):
         initialize_new_project()
         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:
+                try: # Main generation try-except block
                     status.write("Phase 1: Gemini crafting cinematic treatment..."); logger.info("Phase 1: 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 invalid scene list.")
+                    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")
@@ -129,12 +112,13 @@ with st.sidebar: # Voice Customization UI slightly changed
                         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"
-                    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()
-                    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}). "
-                    else: logger.info("All visuals OK.")
-                    status.update(label=f"{current_status_label_ph2}Generating narration script...", state=next_step_state if next_step_state=="error" else "running") # Propagate error state
+                    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()
+                    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.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)
@@ -145,57 +129,44 @@ with st.sidebar: # Voice Customization UI slightly changed
                     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)."
-                        logger.warning("Voiceover skipped/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)
-                except ValueError as ve: logger.error(f"ValueError: {ve}", exc_info=True); status.update(label=f"Input/Gemini response error: {ve}", state="error", expanded=True); 
+                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
-        char_name = st.text_input("Character Name", key="char_name_adv_ultra_v4"); char_desc = st.text_area("Visual Description", key="char_desc_adv_ultra_v4", height=100, placeholder="e.g., Jax: rugged male astronaut...")
-        if st.button("Save Character", key="add_char_adv_ultra_v4"):
+        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..."}
-        sel_preset = st.selectbox("Base Style Preset:", options=list(presets.keys()), key="style_preset_adv_ultra_v4")
-        custom_kw = st.text_area("Additional Custom Style Keywords:", key="custom_style_adv_ultra_v4", height=80, placeholder="e.g., 'Dutch angle'")
+        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
-        if st.button("Apply Global Styles", key="apply_styles_adv_ultra_v4"):
+        if st.button("Apply Global Styles", key="apply_styles_adv_ultra_v5"):
             final_s = presets[sel_preset]; 
             if custom_kw.strip(): final_s = f"{final_s}, {custom_kw.strip()}" if final_s else custom_kw.strip()
             st.session_state.global_style_additions = final_s.strip(); cur_style = final_s.strip()
             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):
-        # Allow user to input a specific Voice ID if they know one, otherwise use the configured one.
-        default_voice_from_secret = st.session_state.get("ELEVENLABS_VOICE_ID_CONFIG", "Rachel") # Default if secret not set
-        
-        user_voice_id_override = st.text_input(
-            "ElevenLabs Voice ID (optional - override secret/default):", 
-            value=default_voice_from_secret, # Pre-fill with secret or default
-            key="el_voice_id_override_v4",
-            help="Enter a specific Voice ID from your ElevenLabs account. If empty, uses the one from secrets or a default."
-        )
-        
+    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.")
         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_v4", index=0) # Default to cinematic
-        
-        if st.button("Set Voice & Narration Style", key="set_voice_btn_ultra_v4"):
+        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
+            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 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} and Script Style to: {sel_prompt_v_style_key}")
-
+            st.success(f"Narrator Voice ID: {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 --- (Scene Display & Edit Popovers - mostly same structure, use_column_width removed from st.image)
+# --- Main Content Area ---
 st.header("🎬 Cinematic Storyboard & Treatment")
 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}_")
@@ -204,10 +175,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]))}_v4" 
+        key_base = f"s{scene_n}_{''.join(filter(str.isalnum, scene_t[:10]))}_v5" # Updated key for uniqueness
         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 logic)
+        with col_d: # Scene Details (same display)
             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
@@ -217,7 +188,7 @@ else:
                 if pexels_q: st.caption(f"Pexels Fallback Query: `{pexels_q}`")
         with col_v: # Image Display and Edit Popovers
             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}") # Removed use_column_width
+            if cur_img_p and os.path.exists(cur_img_p): st.image(cur_img_p, caption=f"Scene {scene_n}: {scene_t}") 
             else:
                 if st.session_state.story_treatment_scenes: st.caption("Visual pending/failed.")
             
@@ -231,9 +202,17 @@ 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")
-                                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: v_num = 2 
-                                else: v_num = 1 
+                                
+                                # 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
+                                
                                 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()
@@ -253,9 +232,17 @@ 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: 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: v_num=2
+                                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
+                                
                                 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()
@@ -265,13 +252,13 @@ else:
     
     # 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_v4", type="primary", use_container_width=True):
+        if st.button("🎬 Assemble Narrated Cinematic Animatic", key="assemble_ultra_video_btn_v5", type="primary", use_container_width=True): # Key updated
             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):
                     img_p_v = st.session_state.generated_visual_paths[i_v] if i_v < len(st.session_state.generated_visual_paths) else None
                     if img_p_v and os.path.exists(img_p_v):
-                        img_data_vid.append({'path':img_p_v, 'scene_num':sc_c.get('scene_number',i_v+1), 'key_action':sc_c.get('key_plot_beat','')}); status_vid.write(f"Adding Scene {sc_c.get('scene_number', i_v + 1)} to video.")
+                        img_data_vid.append({'path':img_p_v, 'scene_num':sc_c.get('scene_number',i_v+1), 'key_action':sc_c.get('key_plot_beat','')}); status_vid.write(f"Adding Scene {sc_c.get('scene_number', i_v + 1)}.")
                 if img_data_vid:
                     status_vid.write("Calling video engine..."); st.session_state.video_path = st.session_state.visual_engine.create_video_from_images(
                         img_data_vid, overall_narration_path=st.session_state.overall_narration_audio_path,
@@ -287,7 +274,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_v4" )
+                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
         except Exception as e: st.error(f"Error displaying video: {e}"); logger.error(f"Error displaying video: {e}", exc_info=True)
 
 # --- Footer ---