Update app.py

app.py

@@ -34,15 +34,19 @@ def clean_response(response: str) -> str:
     return response
 
 def chat_with_model(image_path=None, text_query=None, history=None):
-    try:
-        with Image.open(image_path) as img:
-            print(f"Image {image_path} opened successfully.")
-    default_image_path = None  
-    default_text = ""
-    image_input = image_path if image_path else default_image_path
-    text_input = text_query if text_query else default_text
+    default_image_path = 'path/to/default/image.jpg'
+    if image_path and 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.")
+            image_path = default_image_path
+    else:
+        image_path = default_image_path
+    text_input = text_query if text_query else ""
     query_elements = [
-        {'image': image_input},
+        {'image': image_path},
         {'text': text_input}
     ]
     try:
@@ -52,9 +56,10 @@ def chat_with_model(image_path=None, text_query=None, history=None):
         response = tokenizer.decode(output[0], skip_special_tokens=True)
         cleaned_response = clean_response(response)
         return cleaned_response
-    except UnidentifiedImageError:
-        return "Error: Unable to process the image. Please try a different image."
-        
+    except Exception as e:
+        print(f"An error occurred: {e}")
+        return "An error occurred while processing your request."
+
 def draw_boxes(image_path, response):
     image = Image.open(image_path)
     draw = ImageDraw.Draw(image)