init space

app.py

@@ -0,0 +1,58 @@
+import os
+import gradio as gr
+import google.generativeai as genai
+
+
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+
+genai.configure(api_key=GEMINI_API_KEY)
+model = genai.GenerativeModel('gemini-pro')
+
+header = r"""
+<center>
+    <h1>Gemini Chat</h1>
+    <h2>
+        This demo is for code reference, instead of use it, clone it and set your own Api key
+    </h2>
+    Get an Api Key: <a href='https://aistudio.google.com/app/apikey'>Google Ai Studio</a>
+</center>
+"""
+
+footer = r"""
+<center>
+    <b>
+        Demo for <a href='https://gemini.google.com/app'>Google Gemini</a>
+    </b>
+</center>
+"""
+
+with gr.Blocks(title="Gemini Chat") as app:
+
+    gr.HTML(header)
+
+    with gr.Row(equal_height=True):
+        with gr.Column():
+            chatbot = gr.Chatbot()
+            msg = gr.Textbox()
+            clear = gr.Button("Clear")
+
+            def user(user_message, history):
+                return "", history + [[user_message, None]]
+
+            def bot(history):
+                response = model.generate_content(history[-1][0], stream=True)
+                history[-1][1] = ""
+                for chunk in response:
+                    history[-1][1] += chunk.text
+                    yield history
+
+            msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+                bot, chatbot, chatbot
+            )
+            clear.click(lambda: None, None, chatbot, queue=False)
+    
+    with gr.Row():
+        gr.HTML(footer)
+    
+app.queue()
+app.launch(share=True, debug=True, show_error=True)

requirements.txt

@@ -0,0 +1,2 @@
+google-generativeai
+