Update app.py

app.py

@@ -1,19 +1,18 @@
 import gradio as gr
 from PIL import Image
 from inference.main import MultiModalPhi2
+import os
+
+messages = []
 
-# Initialize the chatbot model
 multimodal_phi2 = MultiModalPhi2(
-    modelname_or_path="Navyabhat/Llava-Phi2",
+    modelname_or_path="RaviNaik/Llava-Phi2",
     temperature=0.2,
     max_new_tokens=1024,
     device="cpu",
 )
 
-# Initialize chatbot history
-messages = []
 
-# Function to add content to chatbot
 def add_content(chatbot, text, image, audio_upload, audio_mic) -> gr.Chatbot:
     textflag, imageflag, audioflag = False, False, False
     if text not in ["", None]:
@@ -31,10 +30,14 @@ def add_content(chatbot, text, image, audio_upload, audio_mic) -> gr.Chatbot:
             audioflag = True
     if not any([textflag, imageflag, audioflag]):
         # Raise an error if neither text nor file is provided
-        raise gr.Error("Enter a valid text, image, or audio")
+        raise gr.Error("Enter a valid text, image or audio")
     return chatbot
 
-# Function to run the chatbot
+
+def clear_data():
+    return {"text": None, "image": None, "audio_upload": None, "audio_mic": None, "chatbot": []}
+
+
 def run(history, text, image, audio_upload, audio_mic):
     if text in [None, ""]:
         text = None
@@ -56,14 +59,37 @@ def run(history, text, image, audio_upload, audio_mic):
 
     history.append((None, outputs.title()))
     return history, None, None, None, None
-    
-# UI setup
+
+
+def print_like_dislike(x: gr.LikeData):
+    print(x.index, x.value, x.liked)
+
+
+def add_text(history, text):
+    history = history + [(text, None)]
+    return history, gr.Textbox(value="", interactive=False)
+
+
+def add_file(history, file):
+    history = history + [((file.name,), None)]
+    return history
+
+
+def bot(history):
+    response = "**That's cool!**"
+    history[-1][1] = ""
+    for character in response:
+        history[-1][1] += character
+        time.sleep(0.05)
+        yield history
+
+
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot(
         [],
         elem_id="chatbot",
         bubble_full_width=False,
-        avatar_images=(None, None),  # You can specify avatar images if needed
+        avatar_images=(None, (os.path.join(os.path.dirname(__file__), "avatar.png"))),
     )
 
     with gr.Row():
@@ -75,18 +101,15 @@ with gr.Blocks() as demo:
         )
         btn = gr.UploadButton("📁", file_types=["image", "video", "audio"])
 
-    txt_msg = txt.submit(add_content, [chatbot, txt], [chatbot, txt], queue=False).then(
-        run, [chatbot, txt, None, None, None], [chatbot, txt, None, None, None], api_name="bot_response"
+    txt_msg = txt.submit(add_content, [chatbot, txt, None, None], [chatbot, txt, None, None], queue=False).then(
+        run, [chatbot, txt, None, None], [chatbot, txt, None, None], api_name="bot_response"
     )
     txt_msg.then(lambda: gr.Textbox(interactive=True), None, [txt], queue=False)
-    file_msg = btn.upload(add_content, [chatbot, None, None, btn, None], [chatbot, None, None, None], queue=False).then(
-        run, [chatbot, None, None, None, btn], [chatbot, None, None, None, None]
+    file_msg = btn.upload(add_content, [chatbot, None, None, btn], [chatbot, None, None, None], queue=False).then(
+        run, [chatbot, None, None, btn], [chatbot, None, None, None]
     )
 
     chatbot.like(print_like_dislike, None, None)
 
-
-
-# Launch the Gradio UI
 demo.queue()
 demo.launch()