Update main.py

main.py

@@ -193,14 +193,15 @@ def process_with_gemini(model, text):
         ]
     }"""
     try:
-        resp = model.generate_content([prompt, text])
+        
+        resp = client.models.generate_content(model='gemini-2.0-flash-exp', contents=[prompt, text])
         time.sleep(6)  # match your Streamlit rate-limit workaround
         return resp.text
     except Exception as e:
         # retry once on 504
         if hasattr(e, "response") and getattr(e.response, "status_code", None) == 504:
             time.sleep(6)
-            resp = model.generate_content([prompt, text])
+            resp = client.models.generate_content(model='gemini-2.0-flash-exp', [prompt, text])
             return resp.text
         raise
 
@@ -262,9 +263,6 @@ def upload_statements():
     if not files:
         return jsonify({"error": "No files uploaded"}), 400
 
-    # configure exactly as in your Streamlit code
-    model = genai.GenerativeModel("gemini-2.0-flash-thinking-exp")
-    stored_count = 0
 
     for f in files:
         filename = f.filename or "statement.pdf"