Update app.py

app.py

@@ -112,24 +112,21 @@ def _launch_demo(args, model, tokenizer):
             print("User uploaded an image.")
             query = {'image': chat_query[0]}  
         else:
-        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)
-
-    return _chatbot
-
-
+            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)
+        return _chatbot
 
     def regenerate(_chatbot, task_history):
         if not task_history:
@@ -149,9 +146,9 @@ def _launch_demo(args, model, tokenizer):
         task_text = text
         if len(text) >= 2 and text[-1] in PUNCTUATION and text[-2] not in PUNCTUATION:
             task_text = text[:-1]
-        history = history + [(_parse_text(text), None)]
-        task_history = task_history + [(task_text, None)]
-        return history, task_history, ""
+            history = history + [(_parse_text(text), None)]
+            task_history = task_history + [(task_text, None)]
+            return history, task_history, ""
 
     def add_file(history, task_history, file):
         if file is None: