Update app.py

app.py

@@ -50,11 +50,16 @@ def detect_milestone_from_image(image_path):
 
 def process_image(images, project_name):
     try:
-        if not images or len(images) == 0:
+        # Ensure images is a list and not empty
+        if not images or len(images) == 0 or not isinstance(images, list):
             return "Error: Please upload at least one image to proceed.", "Pending", "", "", 0
 
         milestones = []
         for image in images:
+            # Ensure each image is a valid file path
+            if not isinstance(image, str) or not os.path.isfile(image):
+                return "Error: Invalid image file.", "Failure", "", "", 0
+
             img = Image.open(image)
 
             image_size_mb = os.path.getsize(image) / (1024 * 1024)
@@ -120,16 +125,15 @@ def process_image(images, project_name):
     except Exception as e:
         return f"Error: {str(e)}", "Failure", "", "", "0%"
 
-# Gradio UI with modified input to show image preview
+# Gradio UI with gallery input
 with gr.Blocks(css="") as demo:
     gr.Markdown("<h1 class='title'>Construction Progress Analyzer</h1>")
     with gr.Row():
-        # Changed from gr.Files to gr.Gallery for visual image preview
         image_input = gr.Gallery(
             label="Upload Construction Site Photos (JPG/PNG, ≤ 20MB)",
             type="filepath",
             interactive=True,
-            show_download_button=False,  # Avoid showing filenames in UI
+            show_download_button=False,
             show_share_button=False
         )
         project_name_input = gr.Textbox(label="Project Name (Required)", placeholder="e.g. Project_12345")