Update app.py

app.py

@@ -4,6 +4,9 @@ import time
 import os
 from datetime import datetime
 from google import genai
+from google.genai import types
+
+
 import uuid
 
 app = Flask(__name__)
@@ -14,6 +17,26 @@ MODEL_ID = "gemini-2.5-flash-preview-05-20"  # Ou le modèle que vous utilisez
 UPLOAD_FOLDER = 'uploads'
 RESULTS_FOLDER = 'results'
 
+safety_settings = [
+    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",
+    ),
+]
+
+
 # Créer les dossiers s'ils n'existent pas
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 os.makedirs(RESULTS_FOLDER, exist_ok=True)
@@ -81,7 +104,10 @@ def generate_synthetic_data(file_path, task_id):
                     # Faire la requête
                     response = client.models.generate_content(
                         model=MODEL_ID,
-                        contents=[file_ref, prompt]
+                        contents=[file_ref, prompt],
+                                                config=types.GenerateContentConfig(
+        safety_settings=safety_settings,
+                                                )
                     )
                     
                     # Écrire la réponse dans le fichier