Update app.py

app.py

@@ -40,7 +40,7 @@ local_timezone = pytz.timezone("Asia/Kolkata")
 # Image processing and Salesforce upload
 def process_image(images, project_name):
     try:
-        if not images or not isinstance(images, list):
+        if not images or len(images) == 0:
             return "Error: Please upload at least one image to proceed.", "Pending", "", "", 0
 
         # Process each image
@@ -87,9 +87,9 @@ def process_image(images, project_name):
             milestone = VALID_MILESTONES[milestone_index]
 
             # Adjust milestone detection based on internal/external features
-            if milestone == "Walls Erected" and any("interior" in img_text.lower() for img_text in [img.filename] if img_text):
+            if milestone == "Walls Erected" and any("interior" in img_text.lower() for img_text in [os.path.basename(image)] if img_text):
                 # Check for visible internal works like electrical/plumbing
-                if any(keyword in str(img).lower() for keyword in ["electrical", "plumbing", "wiring", "pipes"]):
+                if any(keyword in str(os.path.basename(image)).lower() for keyword in ["electrical", "plumbing", "wiring", "pipes"]):
                     milestone = "Completed"  # Upgrade to Completed if internal works are detected
                 else:
                     milestone = "Walls Erected"  # Retain if no internal works visible
@@ -285,7 +285,7 @@ with gr.Blocks(css="""
 """) as demo:
     gr.Markdown("<h1 class='title'>Construction Progress Analyzer</h1>")
     with gr.Row():
-        image_input = gr.Image(type="filepath", label="Upload Construction Site Photos (JPG/PNG, ≤ 20MB)", file_count="multiple")
+        image_input = gr.Files(type="filepath", label="Upload Construction Site Photos (JPG/PNG, ≤ 20MB)")
         project_name_input = gr.Textbox(label="Project Name (Required)", placeholder="e.g. Project_12345")
 
     submit_button = gr.Button("Process Image")