Update app.py

app.py

@@ -108,10 +108,11 @@ def process_input(text=None, file=None, task_history=None):
 
     if "<box>" in response and image_path:
         image_with_boxes_path = draw_boxes(image_path, response)
-        return [("bot", response, (image_with_boxes_path, "Qwen Image Caption"))], task_history
+        cleaned_response = clean_response(response)
+        return [("Qwen-VL_Image", (image_with_boxes_path, "Image with boxes")), ("Qwen-VL_Chat", cleaned_response)], task_history
     else:
         cleaned_response = clean_response(response)
-        return [("bot", cleaned_response)], task_history
+        return [("Qwen-VL_Chat", cleaned_response)], task_history
 
 with gr.Blocks() as demo:
     gr.Markdown("""