Update app.py

app.py

@@ -106,9 +106,10 @@ def process_input(text=None, file=None, task_history=None):
 
     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
+        buffered = BytesIO()
+        image_with_boxes.save(buffered, format="PNG")
+        img_str = buffered.getvalue()
+        return [("bot", response, "data:image/png;base64," + base64.b64encode(img_str).decode())], task_history
     else:
         cleaned_response = clean_response(response)
         return [("bot", cleaned_response)], task_history