Update sozo_gen.py

sozo_gen.py

@@ -966,10 +966,14 @@ 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)
-    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}"
+    
+    # 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}"
+    
     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)