Update app.py

app.py

@@ -95,26 +95,23 @@ def draw_boxes(image_path, response):
 def process_input(text=None, file=None, task_history=None):
     if task_history is None:
         task_history = []
-    image_buffer = None
+    image_path = None
     if file is not None:
-        image_buffer = save_image(file)
-        if image_buffer is None:
+        image_path = save_image(file)
+        if image_path is None:
             return [("bot", "Error: Uploaded file is not a recognizable image.")], task_history
 
-    response = chat_with_model(image=image_buffer, text_query=text, history=task_history)
+    response = chat_with_model(image_path=image_path, text_query=text, history=task_history)
     task_history.append((text, response))
 
-    if "<box>" in response:
-        if image_path:
-            image_with_boxes = draw_boxes(image_path, response)
-            image_with_boxes_path = image_path.replace(".jpg", "_boxed.jpg")
-            image_with_boxes.save(image_with_boxes_path)
-            return [("bot", response), "image", image_with_boxes_path], task_history
-        else:
-            return [("bot", response), "text", None], task_history
+    if "<box>" in response and image_path:
+        image_with_boxes = draw_boxes(image_path, response)
+        image_with_boxes_path = image_path.replace(".jpg", "_boxed.jpg")
+        image_with_boxes.save(image_with_boxes_path)
+        return [("bot", response), "image", image_with_boxes_path], task_history
     else:
-        clean_response = re.sub(r'<ref>(.*?)</ref>(?:<box>.*?</box>)*(?:<quad>.*?</quad>)*', r'\1', response).strip()
-        return [("bot", clean_response)], task_history
+        cleaned_response = clean_response(response)
+        return [("bot", cleaned_response)], task_history
 
 with gr.Blocks() as demo:
     gr.Markdown("""