Update streamlit_app.py

streamlit_app.py

@@ -14,6 +14,14 @@ import google.generativeai as genai
 from pathlib import Path
 import tempfile
 
+from google.genai.types import (
+    GenerateContentConfig,
+    HarmCategory,
+    HarmBlockThreshold,
+    HttpOptions,
+    SafetySetting,
+)
+
 from dotenv import load_dotenv
 load_dotenv()
 
@@ -63,9 +71,28 @@ API_KEY = expander.text_input("Google API Key", "AIzaSyB2uDj6IeuNszSuk80feW-eBgp
 if API_KEY:
     genai.configure(api_key=API_KEY)
     
+#Safety Settings
+safety_settings = [
+    SafetySetting(
+        category=HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
+        threshold=HarmBlockThreshold.OFF,
+    ),
+    SafetySetting(
+        category=HarmCategory.HARM_CATEGORY_HARASSMENT,
+        threshold=HarmBlockThreshold.OFF,
+    ),
+    SafetySetting(
+        category=HarmCategory.HARM_CATEGORY_HATE_SPEECH,
+        threshold=HarmBlockThreshold.OFF,
+    ),
+    SafetySetting(
+        category=HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
+        threshold=HarmBlockThreshold.OFF,
+    ),
+]
+
 #Gemini Model
 model_id = expander.text_input("Gemini Model", "gemini-2.0-flash-exp")
-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" }, ]
 
 # Analysis prompt
 analysis_prompt = expander.text_area("Enter analysis")