Update app.py

app.py

@@ -1,9 +1,10 @@
-import gradio as gr
 import google.generativeai as genai
+import gradio as gr
+import os
+import random
 
-genai.configure(api_key="KEY")
+genai.configure(api_key=os.getenv("KEY"))
 
-# Set up the model
 generation_config = {
   "temperature": 0.9,
   "top_p": 1,
@@ -11,19 +12,30 @@ generation_config = {
   "max_output_tokens": 2048,
 }
 
-model = genai.GenerativeModel(model_name="gemini-pro",
-                              generation_config=generation_config)
+emoji = ['1.png','2,png','3.png','4.png','5.png']
+
+a = ramdom.choice(emoji)
+
+model = genai.GenerativeModel(model_name="gemini-pro",generation_config=generation_config)
 
 def chat(prompt):
     convo = model.start_chat(history=[])
     convo.send_message(prompt)
     return convo.last.text
 
-# Create the Gradio interface
-chat_interface = gr.Interface(
-    fn=chat,
-    inputs=gr.Textbox(lines=2, placeholder="Enter your prompt..."),
-    outputs=gr.Textbox(lines=2, placeholder="Bot response..."),
-)
+'''
+def generate(prompt, history):
+  response = genai.chat(
+    messages=prompt)
+    
+  return response.last'''
+
 
-chat_interface.launch()
+llava=gr.ChatInterface(
+    fn=generate,
+    chatbot=gr.Chatbot(show_label=False, avatar_images=(a, 'llava-logo.svg'), show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
+    title="LLaVa-2",
+    description="This Is Official Demo Of ```LLaVa-2```. ```History/context``` memory does not work in this demo",
+    concurrency_limit=20,#layout="vertical",#bubble_full_width=False
+)
+llava.launch(share=True,show_api=False)