Update app.py

app.py

@@ -1,4 +1,33 @@
+#import os
+#exec(os.environ.get('CODE'))
+
+
+import gradio as gr
 import os
+import google.generativeai as genai
+
+def gemini_chat(message, history):
+    response = chat.send_message(message)
+    return response.text
+
+API_KEY = os.getenv('GOOGLE_API_KEY')
+genai.configure(api_key=API_KEY)
+
+model = genai.GenerativeModel('gemini-pro')
+chat = model.start_chat()
 
+iface = gr.ChatInterface(
+    fn=gemini_chat, 
+    title='Gemini Chat',
+    chatbot=gr.Chatbot(height=1000),
+    textbox=gr.Textbox(
+        placeholder="Message Gemini",
+        scale=7
+    ),
+    retry_btn=None,
+    undo_btn=None,
+    clear_btn=None
+)
 
-exec(os.environ.get('CODE'))
+if __name__ == '__main__':
+    iface.launch()