Update app.py

app.py

@@ -2,37 +2,31 @@ import gradio as gr
 import google.generativeai as genai
 import os
 
-# API KEY from HuggingFace Secrets
+# Load API key
 API_KEY = os.getenv("GOOGLE_API_KEY")
 
-# Safe handling (crash avoid karega)
 if not API_KEY:
-    print("⚠️ API key missing")
-else:
-    genai.configure(api_key=API_KEY)
+    raise Exception("❌ API key not found in secrets")
 
-# Model (safe version)
+# Configure Gemini
+genai.configure(api_key=API_KEY)
+
+# Use stable model
 model = genai.GenerativeModel("gemini-1.0-pro")
 
 def chat(message, history):
     try:
-        if not API_KEY:
-            return "❌ API key missing. Please set GOOGLE_API_KEY in settings."
-
         response = model.generate_content(message)
 
         if response and hasattr(response, "text"):
             return response.text
         else:
-            return "No response from model"
+            return "⚠️ No response from AI"
 
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"❌ Error: {str(e)}"
 
-# Gradio UI
-demo = gr.ChatInterface(
-    fn=chat,
-    type="messages"
-)
+# UI
+iface = gr.ChatInterface(fn=chat, title="Jarvis AI 🤖")
 
-demo.launch()
+iface.launch()