Update app.py

app.py

@@ -1,32 +1,26 @@
 import gradio as gr
-import google.generativeai as genai
+from google import genai
 import os
 
-# Load API key
+# API KEY
 API_KEY = os.getenv("GOOGLE_API_KEY")
 
 if not API_KEY:
-    raise Exception("❌ API key not found in secrets")
+    raise Exception("❌ API key missing")
 
-# Configure Gemini
-genai.configure(api_key=API_KEY)
-
-# Use stable model
-model = genai.GenerativeModel("gemini-1.0-pro")
+# New client (IMPORTANT)
+client = genai.Client(api_key=API_KEY)
 
 def chat(message, history):
     try:
-        response = model.generate_content(message)
-
-        if response and hasattr(response, "text"):
-            return response.text
-        else:
-            return "⚠️ No response from AI"
+        response = client.models.generate_content(
+            model="gemini-1.5-flash",
+            contents=message
+        )
+        return response.text
 
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
-# UI
 iface = gr.ChatInterface(fn=chat, title="Jarvis AI 🤖")
-
-iface.launch()
+iface.launch()