Update streamlit_app.py

streamlit_app.py

@@ -26,11 +26,10 @@ import snscrape.modules.twitter as sntwitter
 DATA_DIR = Path("./data")
 DATA_DIR.mkdir(exist_ok=True)
 
-DEFAULT_MODEL = "gemini-2.0-flash-lite"
+DEFAULT_MODEL = "gemini-2.5-flash-lite"
 DEFAULT_PROMPT = (
-    "Analyze the video and summarize the overall themes and patterns observed, "
-    "focusing on abstract representations of human actions without explicit details. "
-    "Describe the emotional context, dynamics of movement, and interactions using metaphorical language."
+    "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."
 )
 
 MODEL_OPTIONS = [
@@ -193,22 +192,32 @@ 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)
 
-    resp = model.generate_content(
-        [prompt, video_part],
-        generation_config={"max_output_tokens": 1024},
-        request_options={"timeout": timeout},
-    )
+    try:
+        resp = model.generate_content(
+            [prompt, video_part],
+            generation_config={"max_output_tokens": 1024},
+            request_options={"timeout": timeout},
+        )
 
-    # Applying the filter right after generating the response
-    output_text = getattr(resp, "text", str(resp))
-    
-    # Define keywords that you want to filter out
+        # 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)  # Use the filtering function here
+    filtered_text = filter_output(output_text, keywords)
     return filtered_text
 
-
 def _strip_prompt_echo(prompt: str, text: str, threshold: float = 0.68) -> str:
     if not prompt or not text:
         return text
@@ -404,7 +413,6 @@ def main() -> None:
         with st.expander("Show error details"):
             st.code(st.session_state["last_error_detail"], language="text")
 
-
 # ----------------------------------------------------------------------
 # Entry point
 # ----------------------------------------------------------------------