Update app.py

app.py

@@ -2,42 +2,33 @@ import gradio as gr
 import google.generativeai as genai
 import os
 
-# Debug print (logs me dikhega)
-print("Starting Jarvis...")
-
-api_key = os.getenv("GEMINI_API_KEY")
-
-if not api_key:
-    raise ValueError("❌ API KEY NOT FOUND")
-
-genai.configure(api_key=api_key)
+genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 
 model = genai.GenerativeModel("gemini-1.5-flash")
 
-def chat(user_input, history):
+def chat(message, history):
     try:
-        print("User input:", user_input)
-
-        response = model.generate_content(user_input)
+        response = model.generate_content(message)
 
-        if not response or not response.text:
-            reply = "⚠️ No response from AI"
-        else:
+        reply = ""
+        if response and hasattr(response, "text"):
             reply = response.text
+        else:
+            reply = "⚠️ No response from AI"
 
-        history.append((user_input, reply))
+        history.append((message, reply))
         return history, history
 
     except Exception as e:
-        print("ERROR:", str(e))   # logs me exact error dikhega
-        history.append((user_input, f"❌ Error: {str(e)}"))
+        print("ERROR:", e)
+        history.append((message, f"❌ {str(e)}"))
         return history, history
 
 
-iface = gr.ChatInterface(
+demo = gr.ChatInterface(
     fn=chat,
     title="Jarvis AI 🤖",
     description="Internal AI Assistant"
 )
 
-iface.launch(server_name="0.0.0.0", server_port=7860)
+demo.launch(server_name="0.0.0.0", server_port=7860)