Update app.py

app.py

@@ -36,23 +36,20 @@ def clean_response(response: str) -> str:
     return response
 
 def chat_with_model(image_path=None, text_query=None, history=None):
-    default_image_path = 'https://huggingface.co/spaces/Tonic1/Official-Qwen-VL-Chat/blob/main/2saj7l.png'
     if image_path:
-        full_image_path = os.path.join(base_url, image_path) if base_url else image_path
-        if os.path.isfile(full_image_path):
+        if os.path.isfile(image_path):
             try:
-                with Image.open(full_image_path) as img:
-                    print(f"Image {full_image_path} opened successfully.")
+                with Image.open(image_path) as img:
+                    print(f"Image {image_path} opened successfully.")
             except UnidentifiedImageError:
-                print(f"Error: The file at {full_image_path} is not a recognizable image.")
-                full_image_path = default_image_path
+                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: {full_image_path}")
-            full_image_path = default_image_path
+            print(f"File not found: {image_path}")
+            return "Error: Uploaded file not found."
     else:
-        full_image_path = default_image_path
-        print("No image path provided, using default image.")
-        image_path = default_image_path
+        print("No image path provided, using text-only mode.")
+
     text_input = text_query if text_query else ""
     query_elements = [
         {'image': image_path},
@@ -78,6 +75,7 @@ def draw_boxes(image_path, response):
         draw.rectangle([x1, y1, x2, y2], outline="red", width=3)
     return image
 
+def process_input(text=None, file=None, task_history=None):
 def process_input(text=None, file=None, task_history=None):
     if task_history is None:
         task_history = []
@@ -85,6 +83,10 @@ def process_input(text=None, file=None, task_history=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)
     task_history.append((text, response))
 
     if "<box>" in response:
@@ -96,7 +98,6 @@ def process_input(text=None, file=None, task_history=None):
         else:
             return [("bot", response), "text", None], task_history
     else:
-        # Clean the response if it contains any box-like annotations
         clean_response = re.sub(r'<ref>(.*?)</ref>(?:<box>.*?</box>)*(?:<quad>.*?</quad>)*', r'\1', response).strip()
         return [("bot", clean_response)], task_history