Update app.py

app.py

@@ -34,13 +34,6 @@ def save_image(image_file) -> str:
     print(f"Creating upload directory at {upload_dir} if it doesn't exist.")
     os.makedirs(upload_dir, exist_ok=True)
     try:
-        temp_file_path = image_file.name
-        print(f"Attempting to open the image file: {temp_file_path}")
-        if os.path.exists(temp_file_path) and os.path.getsize(temp_file_path) > 0:
-            print(f"File exists and size is {os.path.getsize(temp_file_path)} bytes")
-        else:
-            print("File does not exist or is empty.")
-            return None
         image = Image.open(image_file).convert("RGB")
         file_name = secrets.token_hex(10) + ".png"
         file_path = os.path.join(upload_dir, file_name)
@@ -62,16 +55,15 @@ def clean_response(response: str) -> str:
 
 def chat_with_model(image_path=None, text_query=None, history=None):
     if image_path:
-        if os.path.isfile(image_path):
-            try:
-                with Image.open(image_path) as img:
-                    print(f"Image {image_path} opened successfully.")
-            except UnidentifiedImageError:
-                print(f"Error: The file at {image_path} is not a recognizable image.")
-                return "Error: Uploaded file is not a recognizable image."
-        else:
-            print(f"File not found: {image_path}")
-            return "Error: Uploaded file not found."
+        try:
+            with Image.open(image_path) as img:
+                print(f"Image {image_path} opened successfully.")
+        except UnidentifiedImageError:
+            print(f"Error: The file at {image_path} is not a recognizable image.")
+            return "Error: Uploaded file is not a recognizable image."
+        except Exception as e:
+            print(f"An error occurred while processing the image: {e}")
+            return "An error occurred while processing your request."
     else:
         print("No image path provided, using text-only mode.")
 
@@ -124,7 +116,6 @@ def process_input(text=None, file=None, task_history=None):
         clean_response = re.sub(r'<ref>(.*?)</ref>(?:<box>.*?</box>)*(?:<quad>.*?</quad>)*', r'\1', response).strip()
         return [("bot", clean_response)], task_history
 
-# Define Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("""
 # 🙋🏻‍♂️欢迎来到🌟Tonic 的🦆Qwen-VL-Chat🤩Bot！🚀
@@ -141,7 +132,7 @@ Join us:  TeamTonic  is always making cool demos! Join our active builder's comm
         with gr.Column(scale=1):
             with gr.Row():
                 query = gr.Textbox(lines=2, label='Input', placeholder="Type your message here...")
-                file_upload = gr.File(label="Upload Image")
+                file_upload = gr.Image(type="filepath", label="Upload Image")
                 submit_btn = gr.Button("Submit")
     
     task_history = gr.State([])