Update streamlit_app.py

streamlit_app.py

@@ -193,9 +193,30 @@ 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)
 
+    # Adjust safety settings here
+    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"
+        }
+    ]
+
+    # Modify the generation request to include safety settings
     resp = model.generate_content(
         [prompt, video_part],
-        generation_config={"max_output_tokens": 1024},
+        generation_config={"max_output_tokens": 1024, "safety_settings": safety_settings},
         request_options={"timeout": timeout},
     )
     return getattr(resp, "text", str(resp))
@@ -345,30 +366,6 @@ if st.button("Run Analysis"):
         st.error("No video loaded – load a video first.")
     elif not st.session_state.get("api_key"):
         st.error("Google API key missing – enter it in the sidebar.")
-    else:
-        # Configure Gemini model with safety settings
-        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"
-            }
-        ]
-        
-        # Configure Gemini with API key and safety settings
-        genai.configure(api_key=st.session_state["api_key"], 
-                        safety_settings=safety_settings)  # Set safety to lowest (none)
 
         st.session_state["busy"] = True
         st.session_state["analysis_out"] = ""