Update app.py

app.py

@@ -9,6 +9,8 @@ import torch
 import os
 import io
 from io import BytesIO
+import base64
+import PIL
 from PIL import Image, ImageDraw, UnidentifiedImageError
 
 base_url = "https://huggingface.co/spaces/Tonic1/Official-Qwen-VL-Chat"
@@ -97,9 +99,10 @@ def draw_boxes(image_path, response):
 def draw_boxes_with_tokenizer(response, history):
     image = tokenizer.draw_bbox_on_latest_picture(response, history)
     if image is not None:
-        with tempfile.NamedTemporaryFile(delete=False, suffix='.png', dir=os.path.abspath("uploaded_images")) as temp_file:
-            image.save(temp_file, format="PNG")
-            return temp_file.name
+        buffered = io.BytesIO()
+        image.save(buffered, format="PNG")
+        img_str = base64.b64encode(buffered.getvalue()).decode()
+        return "data:image/png;base64," + img_str
     else:
         print("No box found or drawing failed.")
         return None
@@ -117,10 +120,10 @@ def process_input(text=None, file=None, task_history=None):
     task_history.append((text, response))
 
     if "<box>" in response and image_path:
-        image_with_boxes_path = draw_boxes_with_tokenizer(response, task_history)
-        if image_with_boxes_path:
+        image_with_boxes_base64 = draw_boxes_with_tokenizer(response, task_history)
+        if image_with_boxes_base64:
             cleaned_response = clean_response(response)
-            return [("Qwen-VL_Image", (image_with_boxes_path, "Image with boxes")), ("Qwen-VL_Chat", cleaned_response)], task_history
+            return [("Qwen-VL_Image", image_with_boxes_base64), ("Qwen-VL_Chat", cleaned_response)], task_history
         else:
             return [("bot", "Unable to draw boxes on the image.")], task_history
     else: