Added safety configs

Tabs/Gemini_Chabot_Nightly.py

@@ -1,9 +1,7 @@
 import os
-import time
-from typing import List, Tuple, Optional
 import google.generativeai as genai
+from google.generativeai.types import HarmCategory, HarmBlockThreshold
 import gradio as gr
-from PIL import Image
 from dotenv import load_dotenv
 
 load_dotenv()
@@ -48,24 +46,12 @@ def chatbot_stable(message, history):
     genai.configure(api_key=GEMINI_API_KEY_NIGHTLY)
     model = genai.GenerativeModel(
         model_name,
-        # 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"
-        #     },
-        # ],
+        safety_settings={
+            HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
+            HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
+            HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
+            HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
+        },
         generation_config={
             "temperature": 1,
             "top_p": 0.95,

app.py

@@ -10,6 +10,7 @@ import google.generativeai as genai
 import gradio as gr
 from PIL import Image
 from dotenv import load_dotenv
+from google.generativeai.types import HarmCategory, HarmBlockThreshold
 from Tabs.Gemini_Chabot_Nightly import TITLE_NIGHTLY, NOTICE_NIGHTLY, ERROR_NIGHTLY, gemini_chatbot_stable
 
 load_dotenv()
@@ -94,7 +95,13 @@ def bot(
     response = model.generate_content(
         inputs,
         stream=True,
-        generation_config=generation_config
+        generation_config=generation_config,
+        safety_settings={
+            HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
+            HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
+            HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
+            HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
+        },
     )
     response.resolve()