Update app.py

app.py

@@ -1,12 +1,18 @@
+import gradio as gr
+import os
+import google.generativeai as genai
+
+genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
+
+model = genai.GenerativeModel("gemini-1.5-flash")
+
 def chat(message, history):
     try:
         response = model.generate_content(message)
 
-        # Safe extraction
+        reply = ""
         if hasattr(response, "text") and response.text:
             reply = response.text
-        elif hasattr(response, "candidates"):
-            reply = response.candidates[0].content.parts[0].text
         else:
             reply = "No response from AI"
 
@@ -16,4 +22,13 @@ def chat(message, history):
     except Exception as e:
         print("ERROR:", e)
         history.append((message, f"Error: {str(e)}"))
-        return history, history
+        return history, history
+
+
+demo = gr.ChatInterface(
+    fn=chat,
+    title="Jarvis AI 🤖",
+    description="Internal AI Assistant"
+)
+
+demo.launch()