Update app.py

app.py

@@ -101,6 +101,25 @@ def bot(
             "GOOGLE_API_KEY is not set. "
             "Please follow the instructions in the README to set it up.")
 
+    SAFETY_SETTINGS = [
+    {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "threshold": "BLOCK_NONE",
+    },
+    {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "threshold": "BLOCK_NONE",
+    },
+    {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "threshold": "BLOCK_NONE",
+    },
+    {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "threshold": "BLOCK_NONE",
+    },
+    ]
+    
     genai.configure(api_key=google_key)
     generation_config = genai.types.GenerationConfig(
         temperature=temperature,
@@ -118,14 +137,14 @@ def bot(
         response = model.generate_content(
             text_prompt + image_prompt,
             stream=True,
-            generation_config=generation_config)
+            generation_config=generation_config, safety_settings=SAFETY_SETTINGS)
     else:
         messages = preprocess_chat_history(chatbot)
         model = genai.GenerativeModel('gemini-pro')
         response = model.generate_content(
             messages,
             stream=True,
-            generation_config=generation_config)
+            generation_config=generation_config, safety_settings=SAFETY_SETTINGS)
 
     # streaming effect
     chatbot[-1][1] = ""