Update app.py

app.py

@@ -22,14 +22,27 @@ task_history = []
 BOX_TAG_PATTERN = r"<box>([\s\S]*?)</box>"
 PUNCTUATION = "！？。＂＃＄％＆＇（）＊＋，－／：；＜＝＞＠［＼］＾＿｀｛｜｝～｟｠｢｣､、〃》「」『』&#8203;``【oaicite:0】``&#8203;〔〕〖〗〘〙〚〛〜〝〞〟〰〾〿–—‘’‛“”„‟…‧﹏."
 
-def save_image(image_file, upload_dir: str) -> str:
-    Path(upload_dir).mkdir(parents=True, exist_ok=True)
-    filename = secrets.token_hex(10) + Path(image_file.name).suffix
-    file_path = Path(upload_dir) / filename
-    with open(file_path, "wb") as f_output:
-        f_output.write(image_file.read())
-    return str(file_path)
+def save_image(image_file) -> str:
+    upload_dir = "uploaded_images"
+    print("Creating upload directory if it doesn't exist.")
+    os.makedirs(upload_dir, exist_ok=True)
 
+    try:
+        print("Attempting to open and convert the image.")
+        image = Image.open(image_file).convert("RGB")
+        file_name = secrets.token_hex(10) + ".png"
+        file_path = os.path.join(upload_dir, file_name)
+        print(f"Generated file path: {file_path}")
+        print("Saving the image.")
+        image.save(file_path, format="PNG")
+        print("Image saved successfully.")
+        return file_path
+    except UnidentifiedImageError:
+        print("Error: The file is not a recognizable image.")
+        return None
+    except Exception as e:
+        print(f"An unexpected error occurred: {e}")
+        return None
 
 def clean_response(response: str) -> str:
     response = re.sub(r'<ref>(.*?)</ref>(?:<box>.*?</box>)*(?:<quad>.*?</quad>)*', r'\1', response).strip()
@@ -78,14 +91,13 @@ def draw_boxes(image_path, response):
 def process_input(text=None, file=None, task_history=None):
     if task_history is None:
         task_history = []
-    image_path = None
+    image_buffer = None
     if file is not None:
-        image_path = save_image(file, "uploaded_images")
-    response = chat_with_model(image_path=image_path, text_query=text, history=task_history)
-    if "Error:" in response:
-        return [("bot", response)], task_history
-        
-    response = chat_with_model(image_path=image_path, text_query=text, history=task_history)
+        image_buffer = save_image(file)
+        if image_buffer 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)
     task_history.append((text, response))
 
     if "<box>" in response: