Update app.py

app.py

@@ -17,6 +17,7 @@ from gradio_client import Client,handle_file
 from functools import wraps
 import os
 from google import genai
+from google.genai import types
 import textwrap
 import PIL.Image
 from werkzeug.utils import secure_filename
@@ -42,26 +43,28 @@ generation_config = {
 }
 
 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"
-  },
+    types.SafetySetting(
+        category="HARM_CATEGORY_HARASSMENT",
+        threshold="BLOCK_NONE",
+    ),
+    types.SafetySetting(
+        category="HARM_CATEGORY_HATE_SPEECH",
+        threshold="BLOCK_NONE",
+    ),
+    types.SafetySetting(
+        category="HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        threshold="BLOCK_NONE",
+    ),
+    types.SafetySetting(
+        category="HARM_CATEGORY_DANGEROUS_CONTENT",
+        threshold="BLOCK_NONE",
+    ),
 ]
 
 
 
+
+
 client = genai.Client(
     api_key=GOOGLE_API_KEY,
 )
@@ -406,7 +409,12 @@ La conclusion répond à la question critique posée en introduction, en s'appuy
 
             response = client.models.generate_content(
                 model=MODEL_ID,
+            
                 contents=prompt,
+                config=types.GenerateContentConfig(
+        safety_settings=safety_settings,
+    ),
+                
             )
 
             dissertation = response.text
@@ -708,7 +716,9 @@ La conclusion répond à la question critique posée en introduction, en s'appuy
 
             response = client.models.generate_content(
                 model=MODEL_ID,
-                contents=prompt, )
+                contents=prompt,config=types.GenerateContentConfig(
+        safety_settings=safety_settings,
+    ), )
 
             dissertation = response.text
             return jsonify({"response": dissertation}), 200