Update app.py

app.py

@@ -122,6 +122,9 @@ def _launch_demo(args):
     )
 
     def predict(_chatbot, task_history):
+        
+        print("chatbot", _chatbot)
+        print("history",task_history)
         chat_query = _chatbot[-1][0]
         query = task_history[-1][0]
         if len(chat_query) == 0:
@@ -171,23 +174,13 @@ def _launch_demo(args):
             _chatbot[-1] = (_parse_text(chat_query), _remove_image_special(''.join(response_text)))
             yield _chatbot
 
-        # if len(response) > 1:
-        #     result_image = response[-1]['result_image']
-        #     resp = requests.get(result_image)
-        #     os.makedirs(uploaded_file_dir, exist_ok=True)
-        #     name = f"tmp{secrets.token_hex(20)}.jpg"
-        #     filename = os.path.join(uploaded_file_dir, name)
-        #     with open(filename, 'wb') as f:
-        #         f.write(resp.content)
-        #     response = ''.join(r['box'] if 'box' in r else r['text'] for r in response[:-1])
-        #     _chatbot.append((None, (filename,)))
-        # else:
-        #     _chatbot[-1] = (_parse_text(chat_query), response)
-        # full_response = _parse_text(response)
-
-        # task_history[-1] = (query, full_response)
-        # print("Qwen2.5-VL-Chat: " + _parse_text(full_response))
-        # yield _chatbot
+
+        _chatbot[-1] = (_parse_text(chat_query), _remove_image_special(''.join(response_text)))
+        full_response = _parse_text(''.join(response_text))
+
+        task_history[-1] = (query, full_response)
+        print("Qwen2.5-VL-Chat: " + _parse_text(full_response))
+        yield _chatbot
 
 
     def regenerate(_chatbot, task_history):
@@ -264,4 +257,4 @@ def main():
 
 
 if __name__ == '__main__':
-    main()
+    main()