Update streamlit_app.py

streamlit_app.py

@@ -188,39 +188,42 @@ def _encode_video_b64(path: Path) -> str:
     return base64.b64encode(path.read_bytes()).decode()
 
 
-def generate_report(video_path: Path, prompt: str, model_id: str, timeout: int = 300) -> str:
+def generate_report(
+    video_path: Path,
+    prompt: str,
+    model_id: str,
+    timeout: int = 300,
+) -> str:
+    # Encode video as base‑64
     b64 = _encode_video_b64(video_path)
     video_part = {"inline_data": {"mime_type": "video/mp4", "data": b64}}
-    model = genai.GenerativeModel(model_name=model_id)
 
-    # Adjust safety settings here
+    # Define safety settings (these are passed to the model, not the config)
     safety_settings = [
-        {
-            "category": "HARM_CATEGORY_HARASSMENT",
-            "threshold": "BLOCK_NONE"
-        },
-        {
-            "category": "HARM_CATEGORY_HATE_SPEECH",
-            "threshold": "BLOCK_NONE"
-        },
-        {
-            "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-            "threshold": "BLOCK_NONE"
-        },
-        {
-            "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-            "threshold": "BLOCK_NONE"
-        }
+        {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
+        {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
+        {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"},
+        {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
     ]
 
-    # Modify the generation request to include safety settings
+    # Create the model with safety settings attached
+    model = genai.GenerativeModel(
+        model_name=model_id,
+        safety_settings=safety_settings,
+    )
+
+    # Generation‑specific config (no safety_settings here)
+    gen_config = {"max_output_tokens": 1024}
+
+    # Send the request
     resp = model.generate_content(
         [prompt, video_part],
-        generation_config={"max_output_tokens": 1024, "safety_settings": safety_settings},
+        generation_config=gen_config,
         request_options={"timeout": timeout},
     )
     return getattr(resp, "text", str(resp))
 
+
 def _strip_prompt_echo(prompt: str, text: str, threshold: float = 0.68) -> str:
     if not prompt or not text:
         return text