Final update 🤞

app.py

@@ -49,6 +49,15 @@ def update_images(image_component, directory="/home/user/app/"):
 
 def create_chat_widget():
     with gr.Blocks() as chatblock:
+        # Adding a row for the New Chat button at the top
+        with gr.Row():
+            new_chat_button = gr.Button("New Chat")
+            new_chat_button.click(
+                lambda: ([], ""),  
+                inputs=[], 
+                outputs=[chatbot] )
+
+        # Main chat interface
         with gr.Row():
             chatbot = gr.Chatbot(
                 [],
@@ -67,6 +76,7 @@ def create_chat_widget():
             txt.submit(update_bot, inputs=[txt, chatbot], outputs=[chatbot, txt])
             send_button.click(update_bot, inputs=[txt, chatbot], outputs=[chatbot, txt])
 
+        # Image display row
         with gr.Row():
             image_display = gr.Image()
             update_image_button = gr.Button("Update Image")
@@ -74,14 +84,15 @@ def create_chat_widget():
 
         return chatblock
 
+
 def update_bot(text, chatbot):
-    if text:  # Append user input to the chat first
-        chatbot.append(("User", text))
+    if text:  # Append user input to the chat with "User:" prefix
+        chatbot.append(("User", f"User: {text}"))
     response_json = interpreter.chat(text, stream=True, display=False)
     formatted_response, images = json_to_markdown(response_json)
-    # Append text response if it exists
+    # Append text response with "Assistant:" prefix if it exists
     if formatted_response.strip():
-        chatbot.append(("Assistant", formatted_response))
+        chatbot.append(("Assistant", f"Assistant: {formatted_response}"))
     # Append images if any
     for img_path in images:
         if os.path.exists(img_path) and img_path.endswith('.png'):