Update streamlit_app.py

streamlit_app.py

@@ -28,8 +28,9 @@ DATA_DIR.mkdir(exist_ok=True)
 
 DEFAULT_MODEL = "gemini-2.5-flash-lite"
 DEFAULT_PROMPT = (
-    "Analyze the video and provide an overview of the general actions and interactions observed. "
-    "Describe the emotional tone and patterns of movement without referencing specific or explicit content."
+    "Watch the video and provide a detailed behavioral report focusing on human actions, "
+    "interactions, posture, movement, and apparent intent. Keep language professional. "
+    "Include a list of observations for notable events."
 )
 
 MODEL_OPTIONS = [
@@ -192,31 +193,12 @@ def generate_report(video_path: Path, prompt: str, model_id: str, timeout: int =
     video_part = {"inline_data": {"mime_type": "video/mp4", "data": b64}}
     model = genai.GenerativeModel(model_name=model_id)
 
-    try:
-        resp = model.generate_content(
-            [prompt, video_part],
-            generation_config={"max_output_tokens": 1024},
-            request_options={"timeout": timeout},
-        )
-
-        # Check if the response contains valid candidates
-        if not resp.candidates:
-            feedback = getattr(resp, 'prompt_feedback', None)
-            if feedback:
-                st.warning(f"Feedback on the prompt: {feedback.block_reason}")
-                return "Analysis could not be generated due to content restrictions."
-
-            return "No valid analysis could be generated."
-        output_text = getattr(resp, "text", str(resp))
-        
-    except Exception as e:
-        st.error(f"An error occurred during report generation: {e}")
-        return "Failed to generate report."
-
-    # Define keywords to filter out
-    keywords = ["explicit", "graphic", "violence", "nudity"]  # Add any other terms if necessary
-    filtered_text = filter_output(output_text, keywords)
-    return filtered_text
+    resp = model.generate_content(
+        [prompt, video_part],
+        generation_config={"max_output_tokens": 1024},
+        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:
@@ -228,12 +210,6 @@ def _strip_prompt_echo(prompt: str, text: str, threshold: float = 0.68) -> str:
         return text[cut:].lstrip(" \n:-")
     return text
 
-def filter_output(text: str, keywords: list) -> str:
-    for keyword in keywords:
-        if keyword.lower() in text.lower():
-            text = text.replace(keyword, "[REDACTED]")
-    return text
-
 # ----------------------------------------------------------------------
 # UI helpers
 # ----------------------------------------------------------------------