Update app.py

app.py

@@ -1,5 +1,9 @@
 import gradio as gr
-import google.generativeai as genai
+import os
+import faiss
+from sentence_transformers import SentenceTransformer
+from google import genai
+from google.genai import types
 
 # --- Load knowledge base from repo at startup ---
 with open("knowledge.txt", "r", encoding="utf-8") as f:
@@ -27,14 +31,17 @@ def chat(user_input):
     if user_input.lower() in knowledge_base.lower():
         return "📖 Based on knowledge base: " + user_input
 
-    # Fallback to Gemini model
-    genai.api_key = global_api_key
+    # Fallback to Gemini model using your provided code style
     try:
-        response = genai.models.generate(
-            model="gemini-pro",
-            messages=[{"author": "user", "content": user_input}]
+        client = genai.Client(api_key=global_api_key)
+        response = client.models.generate_content(
+            model="gemini-2.5-flash",
+            contents=user_input,
+            config=types.GenerateContentConfig(
+                thinking_config=types.ThinkingConfig(thinking_budget=0)
+            ),
         )
-        return response.candidates[0].content
+        return response.text
     except Exception as e:
         return f"❌ Error calling Gemini model: {e}"