Create app.py

app.py

@@ -0,0 +1,32 @@
+import gradio as gr
+import google.generativeai as genai
+import os
+
+# Secure API key
+genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
+
+# Model
+model = genai.GenerativeModel("gemini-1.5-flash")
+
+# Chat memory
+chat_session = model.start_chat(history=[])
+
+# Chat function
+def chat(user_input, history):
+    try:
+        response = chat_session.send_message(user_input)
+        reply = response.text
+        history.append((user_input, reply))
+        return history, history
+    except Exception as e:
+        return str(e), history
+
+# UI (customized)
+iface = gr.ChatInterface(
+    fn=chat,
+    title="Jarvis AI 🤖",
+    description="Internal AI Assistant for Team",
+    theme="soft"
+)
+
+iface.launch()