Update main.py

main.py

@@ -22,15 +22,15 @@ import matplotlib
 matplotlib.use("Agg")
 import matplotlib.pyplot as plt
 import matplotlib.cm as cm
-import google.generativeai as genai
+from google import genai
+from google.genai import types as genai_types
 from typing import Optional
 
 # ── Gemini setup ──────────────────────────────────────────────────────────────
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY", "")
-if GEMINI_API_KEY:
-    genai.configure(api_key=GEMINI_API_KEY)
+_gemini_client = genai.Client(api_key=GEMINI_API_KEY) if GEMINI_API_KEY else None
 
-GEMINI_MODEL = "gemini-1.5-flash"
+GEMINI_MODEL = "gemini-2.0-flash"
 
 # نوع الصورة لكل موديل — بيساعد Gemini يفهم السياق
 MODEL_IMAGE_TYPE = {
@@ -58,7 +58,7 @@ def generate_medical_report(
     يبعت بيانات المريض والتشخيص لـ Gemini ويرجع تقرير طبي مخصص بالعربي.
     لو Gemini مش متاح يرجع None وهنستخدم الـ recommendations الثابتة.
     """
-    if not GEMINI_API_KEY:
+    if not _gemini_client:
         return None
 
     # تحديد مستوى الـ confidence بلغة مفهومة
@@ -111,11 +111,11 @@ def generate_medical_report(
 اكتب التقرير مباشرة بدون عناوين أو نقاط أو مقدمات."""
 
     try:
-        gemini = genai.GenerativeModel(GEMINI_MODEL)
-        response = gemini.generate_content(
-            prompt,
-            generation_config=genai.types.GenerationConfig(
-                temperature=0.4,   # منخفض عشان يكون consistent ومضبوط
+        response = _gemini_client.models.generate_content(
+            model=GEMINI_MODEL,
+            contents=prompt,
+            config=genai_types.GenerateContentConfig(
+                temperature=0.4,
                 max_output_tokens=350,
             ),
         )