Update streamlit_app.py

streamlit_app.py

@@ -194,11 +194,15 @@ def generate_report(
     model_id: str,
     timeout: int = 300,
 ) -> str:
-    # Encode video as base‑64
+    # --------------------------------------------------------------
+    # 1️⃣ Encode the video as base‑64
+    # --------------------------------------------------------------
     b64 = _encode_video_b64(video_path)
     video_part = {"inline_data": {"mime_type": "video/mp4", "data": b64}}
 
-    # Define safety settings (these are passed to the model, not the config)
+    # --------------------------------------------------------------
+    # 2️⃣ Safety settings – keep BLOCK_NONE (as you requested)
+    # --------------------------------------------------------------
     safety_settings = [
         {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
         {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
@@ -206,22 +210,58 @@ def generate_report(
         {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
     ]
 
-    # Create the model with safety settings attached
+    # --------------------------------------------------------------
+    # 3️⃣ Build the model with those safety settings
+    # --------------------------------------------------------------
     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
+    # --------------------------------------------------------------
+    # 4️⃣ Send the request (generation_config now contains only token limits)
+    # --------------------------------------------------------------
     resp = model.generate_content(
         [prompt, video_part],
-        generation_config=gen_config,
+        generation_config={"max_output_tokens": 1024},
         request_options={"timeout": timeout},
     )
-    return getattr(resp, "text", str(resp))
+
+    # --------------------------------------------------------------
+    # 5️⃣ Pull the safety feedback (always safe to access)
+    # --------------------------------------------------------------
+    feedback = resp.prompt_feedback
+    block_reason = getattr(feedback, "block_reason", None)
+    safety_ratings = getattr(feedback, "safety_ratings", [])
+
+    # --------------------------------------------------------------
+    # 6️⃣ Assemble the output we will return
+    # --------------------------------------------------------------
+    parts = []
+
+    # 6a – Normal text (if any candidate was returned)
+    if resp.candidates:
+        parts.append(resp.text)                     # quick accessor works here
+    else:
+        parts.append("[No candidate output – request blocked]")
+
+    # 6b – Full safety feedback
+    if block_reason:
+        parts.append(f"\n**Block reason:** {block_reason}")
+
+    if safety_ratings:
+        rating_lines = []
+        for rating in safety_ratings:
+            cat = rating.category.name
+            sev = rating.probability.name
+            rating_lines.append(f"- {cat}: {sev}")
+        parts.append("\n**Safety ratings**:\n" + "\n".join(rating_lines))
+
+    # 6c – Any additional message the API may include
+    if getattr(resp, "message", None):
+        parts.append(f"\n**Message:** {resp.message}")
+
+    return "\n".join(parts)
 
 
 def _strip_prompt_echo(prompt: str, text: str, threshold: float = 0.68) -> str: