Update app.py

app.py

@@ -14,7 +14,7 @@ import logging
 import os
 
 # Configure logging
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 
 # Load model and preprocessors for Image-Text Matching (LAVIS)
 device = torch.device("cuda") if torch.cuda.is_available() else "cpu"
@@ -81,8 +81,14 @@ def is_image_file(file):
     return file_extension.lower() in allowed_extensions
 
 
+# Main function to perform image captioning and image-text matching
 # Main function to perform image captioning and image-text matching
 def process_images_and_statements(file):
+    logging.debug("Entered process_images_and_statements function")
+    logging.debug(f"File object: {file}")
+    logging.debug(f"File name: {file.name}")
+    logging.debug(f"File size: {file.tell()}")
+
     # Check if the uploaded file is an image
     if not is_image_file(file):
         return "Invalid file type. Please upload an image file (e.g., .jpg, .png, .jpeg)."
@@ -91,7 +97,13 @@ def process_images_and_statements(file):
     filename = file.name
 
     # Load the image data from the file (convert file object to bytes using file.read())
-    image = Image.open(io.BytesIO(file.read()))
+    try:
+        logging.debug("Attempting to open image")
+        image = Image.open(io.BytesIO(file.read()))
+        logging.debug("Image opened successfully")
+    except Exception as e:
+        logging.exception("Error occurred while opening image")
+        return str(e)  # Return error message to the user
 
     # Generate image caption for the uploaded image using git-large-r-textcaps
     caption = generate_caption(git_processor_large_textcaps, git_model_large_textcaps, image)