Update sozo_gen.py

sozo_gen.py

@@ -567,6 +567,7 @@ def generate_single_chart(df: pd.DataFrame, description: str, uid: str, project_
                 os.unlink(img_path)
     return None
 
+
 def generate_video_from_project(df: pd.DataFrame, raw_md: str, data_context: Dict, uid: str, project_id: str, voice_model: str, bucket):
     logging.info(f"Generating video for project {project_id} with voice {voice_model}")
     llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash", google_api_key=API_KEY, temperature=0.2)
@@ -655,12 +656,14 @@ def generate_video_from_project(df: pd.DataFrame, raw_md: str, data_context: Dic
 
     with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as temp_vid, \
             tempfile.NamedTemporaryFile(suffix=".mp3", delete=False) as temp_aud, \
-            tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as final_vid:
+            tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as final_vid, \
+            tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as branded_vid:
         
         silent_vid_path = Path(temp_vid.name)
         audio_mix_path = Path(temp_aud.name)
         final_vid_path = Path(final_vid.name)
-        
+        branded_vid_path = Path(branded_vid.name)
+
         concat_media(final_video_parts, silent_vid_path)
         concat_media(audio_parts, audio_mix_path)
         
@@ -673,18 +676,44 @@ def generate_video_from_project(df: pd.DataFrame, raw_md: str, data_context: Dic
         ]
         subprocess.run(cmd, check=True, capture_output=True)
         
+        upload_path = final_vid_path
+        logo_path = Path("sozob.png")
+
+        if logo_path.exists():
+            logging.info("Logo 'sozob.png' found. Adding full-screen end-card.")
+            duration_for_filter = total_audio_duration
+            
+            filter_complex = f"[1:v]scale={WIDTH}:{HEIGHT}[logo];[0:v][logo]overlay=0:0:enable='gte(t,{duration_for_filter - 2})'"
+
+            logo_cmd = [
+                "ffmpeg", "-y",
+                "-i", str(final_vid_path),
+                "-i", str(logo_path),
+                "-filter_complex", filter_complex,
+                "-map", "0:a",
+                "-c:a", "copy",
+                "-c:v", "libx264", "-pix_fmt", "yuv420p",
+                str(branded_vid_path)
+            ]
+            try:
+                subprocess.run(logo_cmd, check=True, capture_output=True)
+                upload_path = branded_vid_path
+            except subprocess.CalledProcessError as e:
+                logging.error(f"Failed to add logo end-card. Uploading unbranded video. Error: {e.stderr.decode()}")
+        else:
+            logging.warning("Logo 'sozob.png' not found in root directory. Skipping end-card.")
+
         blob_name = f"sozo_projects/{uid}/{project_id}/video.mp4"
         blob = bucket.blob(blob_name)
-        blob.upload_from_filename(str(final_vid_path))
+        blob.upload_from_filename(str(upload_path))
         logging.info(f"Uploaded video to {blob.public_url}")
         
-        for p in temps + [silent_vid_path, audio_mix_path, final_vid_path]:
+        for p in temps + [silent_vid_path, audio_mix_path, final_vid_path, branded_vid_path]:
             if os.path.exists(p): os.unlink(p)
             
         return blob.public_url
     return None
 
-
 # In sozo_gen.py, add these new functions at the end of the file
 
 def generate_image_with_gemini(prompt: str) -> Image.Image: