Update app.py

app.py

@@ -1,6 +1,9 @@
 from flask import Flask, request, render_template, jsonify
 import PIL.Image
-import google.generativeai as genai
+from google import genai
+from google.genai import types
+
+
 import os
 from tempfile import NamedTemporaryFile
 import tempfile
@@ -13,27 +16,31 @@ 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",
+    ),
 ]
 
 GOOGLE_API_KEY = os.environ.get("TOKEN")
 
-genai.configure(api_key=GOOGLE_API_KEY)
+#genai.configure(api_key=GOOGLE_API_KEY)
+
+gen = GOOGLE_API_KEY 
+client = genai.Client(api_key=gen)
+
 
 @app.route('/', methods=['GET'])
 def svt():
@@ -145,8 +152,21 @@ def svt_submit():
                     temp_files.append(temp_file.name)
                     content.append(PIL.Image.open(temp_file.name))
 
-        model = genai.GenerativeModel(model_name="models/gemini-2.5-flash-preview-04-17", safety_settings=safety_settings)
-        response = model.generate_content(content, request_options={"timeout": 600})
+        #model = genai.GenerativeModel(model_name="models/gemini-2.5-flash-preview-04-17", safety_settings=safety_settings)
+        #response = model.generate_content(content, request_options={"timeout": 600})
+
+        response = client.models.generate_content(
+            model="gemini-2.5-flash-preview-04-17",
+            contents=[content],
+                        config=types.GenerateContentConfig(
+        safety_settings=safety_settings,
+
+    )
+        ),
+        )
+            
+        answer = response.candidates[0].content.parts[0].text
+        return answer.strip()
 
         return jsonify({"response": response.text})