Update app.py

app.py

@@ -47,6 +47,11 @@ def process_image(images, project_name):
         milestones = []
         for image in images:
             img = Image.open(image)
+            
+            # Debugging log for image filename
+            print(f"Processing image: {os.path.basename(image)}")
+
+            # Check if the image exceeds the size limit (20MB)
             image_size_mb = os.path.getsize(image) / (1024 * 1024)
             if image_size_mb > 20:
                 return "Error: One or more images exceed 20MB.", "Failure", "", "", 0
@@ -80,6 +85,9 @@ def process_image(images, project_name):
             except Exception as e:
                 return f"Error: Failed to upload image to Salesforce - {str(e)}", "Failure", "", "", 0
 
+            # Debugging log for detected milestone
+            print(f"Detected milestone for image: {os.path.basename(image)}")
+
             # Check if the image represents a completed building
             if "completed" in os.path.basename(image).lower() or "finished" in os.path.basename(image).lower():
                 milestone = "Completed"  # Directly detect completion if such features exist
@@ -90,10 +98,14 @@ def process_image(images, project_name):
                 milestone_index = img_hash % len(VALID_MILESTONES)
                 milestone = VALID_MILESTONES[milestone_index]
 
+            print(f"Milestone detected: {milestone}")
+
             milestones.append(milestone)
 
         # Determine overall milestone (most advanced detected)
         final_milestone = max(set(milestones), key=milestones.count) if milestones else "Planning"
+        print(f"Final milestone after processing all images: {final_milestone}")
+
         milestone_completion_map = {
             "Planning": 10,
             "Foundation": 30,
@@ -101,7 +113,9 @@ def process_image(images, project_name):
             "Completed": 100,
         }
         percent_complete = milestone_completion_map.get(final_milestone, 0)
+        print(f"Completion percentage: {percent_complete}%")
 
+        # Completion details for different milestones
         completion_details = {
             "Planning": {
                 "completed": [
@@ -221,6 +235,7 @@ def process_image(images, project_name):
         return result_html, "Success", final_milestone, f"{percent_complete}%"
 
     except Exception as e:
+        print(f"Error: {str(e)}")
         return f"Error: {str(e)}", "Failure", "", "", "0%"
 
 # Gradio UI