Update sozo_gen.py

sozo_gen.py

@@ -27,7 +27,7 @@ import requests
 # In sozo_gen.py, near the other google imports
 from google.genai import types as genai_types
 import math # Add this import at the top of your sozo_gen.py file
-
+import shutil
 # --- Configuration ---
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - [%(funcName)s] - %(message)s')
 FPS, WIDTH, HEIGHT = 24, 1280, 720
@@ -612,6 +612,7 @@ def generate_video_from_project(df: pd.DataFrame, raw_md: str, data_context: Dic
         video_dur = audio_dur + 1.5 
         
         try:
+            # --- Primary Visual Generation ---
             chart_descs = extract_chart_tags(sc)
             pexels_descs = extract_pexels_tags(sc)
             is_conclusion_scene = any(k in narrative.lower() for k in ["conclusion", "summary", "in closing", "final thoughts"])
@@ -627,13 +628,28 @@ def generate_video_from_project(df: pd.DataFrame, raw_md: str, data_context: Dic
                     conclusion_video_path = video_path
             elif chart_descs:
                 logging.info(f"Scene {i+1}: Primary attempt with animated chart.")
+                if not chart_descs: raise ValueError("AI script failed to provide a chart tag for this scene.")
                 safe_chart(chart_descs[0], df, video_dur, mp4, data_context)
                 video_parts.append(str(mp4))
             else:
-                raise ValueError("No visual tag found in scene.")
+                raise ValueError("No visual tag found in scene script.")
         except Exception as e:
-            logging.warning(f"Scene {i+1}: Primary visual failed ({e}). Marking for fallback.")
-            video_parts.append("FALLBACK_NEEDED")
+            logging.warning(f"Scene {i+1}: Primary visual failed ({e}). Triggering Fallback Tier 1.")
+            # --- Fallback Tier 1: Context-Aware Pexels Replacement ---
+            try:
+                fallback_keywords = extract_keywords_for_query(narrative, llm)
+                final_fallback_query = f"{fallback_keywords} {domain}"
+                logging.info(f"Fallback Tier 1: Searching Pexels with query: '{final_fallback_query}'")
+                
+                video_path = search_and_download_pexels_video(final_fallback_query, video_dur, mp4)
+                if not video_path: raise ValueError("Fallback Pexels search returned no results.")
+                
+                video_parts.append(video_path)
+                logging.info(f"Scene {i+1}: Successfully recovered with a relevant Pexels video.")
+            except Exception as fallback_e:
+                # --- Fallback Tier 2: Looping Conclusion Failsafe ---
+                logging.error(f"Scene {i+1}: Fallback Tier 1 also failed ({fallback_e}). Marking for final failsafe.")
+                video_parts.append("FALLBACK_NEEDED")
 
         temps.append(mp4)
 
@@ -647,10 +663,13 @@ def generate_video_from_project(df: pd.DataFrame, raw_md: str, data_context: Dic
     for part in video_parts:
         if part == "FALLBACK_NEEDED":
             if conclusion_video_path:
-                logging.info("Applying conclusion video as fallback for a failed scene.")
-                final_video_parts.append(conclusion_video_path)
+                fallback_copy_path = Path(tempfile.gettempdir()) / f"fallback_{uuid.uuid4().hex}.mp4"
+                shutil.copy(conclusion_video_path, fallback_copy_path)
+                temps.append(fallback_copy_path)
+                final_video_parts.append(str(fallback_copy_path))
+                logging.info(f"Applying unique copy of conclusion video as fallback for a failed scene.")
             else:
-                logging.error("Cannot apply fallback; no conclusion video available.")
+                logging.error("Cannot apply fallback; no conclusion video available. A scene will be missing.")
         else:
             final_video_parts.append(part)