Update utils.py

utils.py

@@ -1,3 +1,28 @@
-from google.generativeai import SafetySettings
+"""https://ai.google.dev/api/rest/v1beta/HarmCategory"""
 
-SAFETY_SETTINGS = SafetySettings()
+SAFETY_SETTINGS = [
+    {
+        "category": "HARM_CATEGORY_SEXUAL",
+        "threshold": "ALLOW",
+    },
+    {
+        "category": "HARM_CATEGORY_DANGEROUS",
+        "threshold": "ALLOW",
+    },
+    {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "threshold": "ALLOW",
+    },
+    {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "threshold": "ALLOW",
+    },
+    {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "threshold": "ALLOW",
+    },
+    {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "threshold": "ALLOW",
+    },
+]