Update sozo_gen.py

sozo_gen.py

@@ -966,14 +966,10 @@ def generate_single_chart(df: pd.DataFrame, description: str, uid: str, project_
 def generate_video_from_project(df: pd.DataFrame, raw_md: str, 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.0-flash", google_api_key=API_KEY, temperature=0.2)
-    
-    # Modified prompt to output script directly without prefacing
-    story_prompt = f"Generate a {VIDEO_SCENES}-scene video script. Each scene must be separated by '[SCENE_BREAK]' and contain narration and one chart tag. Report: {raw_md}"
-    
+    story_prompt = f"Based on the following report, create a script for a {VIDEO_SCENES}-scene video. Each scene must be separated by '[SCENE_BREAK]' and contain narration and one chart tag. Report: {raw_md}. only output the script no quips"
     script = llm.invoke(story_prompt).content
     scenes = [s.strip() for s in script.split("[SCENE_BREAK]") if s.strip()]
     video_parts, audio_parts, temps = [], [], []
-    
     for sc in scenes:
         descs, narrative = extract_chart_tags(sc), clean_narration(sc)
         audio_bytes = deepgram_tts(narrative, voice_model)
@@ -991,10 +987,10 @@ def generate_video_from_project(df: pd.DataFrame, raw_md: str, uid: str, project
             img_cv = cv2.cvtColor(np.array(img.resize((WIDTH, HEIGHT))), cv2.COLOR_RGB2BGR)
             animate_image_fade(img_cv, dur, mp4)
         video_parts.append(str(mp4)); temps.append(mp4)
-    
+
     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=".mp3", delete=False) as temp_aud, \
+            tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as final_vid:
         
         silent_vid_path = Path(temp_vid.name)
         audio_mix_path = Path(temp_aud.name)
@@ -1005,8 +1001,8 @@ def generate_video_from_project(df: pd.DataFrame, raw_md: str, uid: str, project
         
         subprocess.run(
             ["ffmpeg", "-y", "-i", str(silent_vid_path), "-i", str(audio_mix_path),
-             "-c:v", "libx264", "-pix_fmt", "yuv420p", "-c:a", "aac",
-             "-map", "0:v:0", "-map", "1:a:0", "-shortest", str(final_vid_path)],
+            "-c:v", "libx264", "-pix_fmt", "yuv420p", "-c:a", "aac",
+            "-map", "0:v:0", "-map", "1:a:0", "-shortest", str(final_vid_path)],
             check=True, capture_output=True,
         )