Update app.py

app.py

@@ -108,31 +108,27 @@ def _launch_demo(args, model, tokenizer):
         if not _chatbot:
             return _chatbot 
         chat_query = _chatbot[-1][0]
-        print("User: " + _parse_text(chat_query))
-
-        history = process_history_for_model(task_history)
-
-        if isinstance(chat_query, tuple):  # Image input
-            message = {'image': chat_query[0]}
-        else:  # Text input
-            message = {'text': chat_query}
-
-        response, updated_history = model.chat(tokenizer, query=message, history=history)
-
-        if "<box>" in response:
-            image = tokenizer.draw_bbox_on_latest_picture(response, updated_history)
-            if image is not None:
-                image_path = save_image(image)  
-                _chatbot[-1] = (_parse_text(chat_query), (image_path,))
-                response = process_response(response)
-            else:
-                _chatbot[-1] = (_parse_text(chat_query), response)
+        if isinstance(chat_query, tuple):
+            print("User uploaded an image.")
+            query = {'image': chat_query[0]}  
         else:
-            _chatbot[-1] = (_parse_text(chat_query), response)
+        print("User: " + _parse_text(chat_query))
+        query = {'text': chat_query}
+    history_cp = copy.deepcopy(task_history)
+    full_response = ""
+    history = process_history_for_model(history_cp)
+    response, history = model.chat(tokenizer, query=query, history=history)
+    image = tokenizer.draw_bbox_on_latest_picture(response, history)
+    if image is not None:
+        image_path = save_image(image)  
+        _chatbot[-1] = (chat_query, (image_path,))
+        response = process_response(response)  
+    else:
+        _chatbot[-1] = (chat_query, response)
+    task_history = update_task_history(task_history, history, response)
 
-        task_history = update_task_history(task_history, updated_history, response)
+    return _chatbot
 
-        return _chatbot
 
 
     def regenerate(_chatbot, task_history):