Update app.py

app.py

@@ -1,5 +1,6 @@
 from flask import Flask, render_template, request, jsonify
-import google.generativeai as genai
+from google import genai
+from google.genai import types
 import os
 from PIL import Image
 import tempfile
@@ -8,18 +9,27 @@ app = Flask(__name__)
 
 # Configuration de l'API Gemini
 token = os.environ.get("TOKEN")
-genai.configure(api_key=token)
-
-generation_config = {
-    "temperature": 1,
-    "max_output_tokens": 8192,
-}
-
-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"},
+genai_client = genai.Client(api_key=token)
+        
+
+
+SAFETY_SETTINGS = [
+    types.SafetySetting(
+        category=types.HarmCategory.HARM_CATEGORY_HATE_SPEECH,
+        threshold=types.HarmBlockThreshold.BLOCK_NONE,
+    ),
+    types.SafetySetting(
+        category=types.HarmCategory.HARM_CATEGORY_HARASSMENT,
+        threshold=types.HarmBlockThreshold.BLOCK_NONE,
+    ),
+    types.SafetySetting(
+        category=types.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
+        threshold=types.HarmBlockThreshold.BLOCK_NONE,
+    ),
+    types.SafetySetting(
+        category=types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
+        threshold=types.HarmBlockThreshold.BLOCK_NONE,
+    )
 ]
 
 prompt_tableau = """
@@ -162,19 +172,25 @@ Ainsi, ... (axe 2) est lié (e) à ... (sous-axe 1) et à ... (sous-axe 2).
 
 Somme toute, ... (titre du texte) organise son sens autour de … (axe 1) et de ... (axe 2). De ces deux centres d’intérêt découlent respectivement, d’une part, … (sous-axe 1 de l'axe 1) et ... (sous-axe 2 de l'axe 1) et, d’autre part, … (sous-axe 1 de l'axe 2) et … (sous-axe 2 de l'axe 2). À travers ce texte, ...(nom de l'auteur) nous ... (opinion personnelle). Une telle optique est perceptible dans la logique de... (nom de l'auteur nous permettant de faire un rapprochement thématique), dans son œuvre ...(titre de l’œuvre), dans lequel il aborde… (bref résumé de l'œuvre en question qui peut être facultatif).
 """
-
-model = genai.GenerativeModel(
-    model_name="gemini-2.0-flash",
-    generation_config=generation_config,
-    safety_settings=safety_settings
+generate_config = types.GenerateContentConfig(
+        system_instruction=SYSTEM_INSTRUCTION,
+        safety_settings=SAFETY_SETTINGS
 )
+model_id="gemini-2.0-flash"
+
+
 
 def generate_table(image, consignes=""):
     """Génère le tableau d'analyse à partir de l'image en intégrant éventuellement des consignes"""
     prompt = prompt_tableau
     if consignes:
         prompt += "\n" + consignes
-    response = model.generate_content([prompt, image])
+    #sresponse = model.generate_content([prompt, image])
+    response = client.models.generate_content(
+            model=model_id,
+            contents=[prompt,image],
+    config=generate_config
+    )
     return response.text
 
 def generate_dissertation(tableau):
@@ -182,7 +198,13 @@ def generate_dissertation(tableau):
     prompt = f"""
     
  {prompt_redaction}.Écris maintenant un rédaction pour ça en suivant l'exercice a trous ; {tableau}"""
-    response = model.generate_content(prompt)
+    #response = model.generate_content(prompt)
+
+    response = client.models.generate_content(
+            model=model_id,
+            contents=promtp,
+        config=generate_config)
+    
     return response.text
 
 @app.route('/')