Update app.py

app.py

@@ -51,18 +51,17 @@ def detect_milestone_from_image(image_path):
 def process_image(images, project_name):
     try:
         if not images or len(images) == 0:
-            return "Error: Please upload at least one image to proceed.", "Pending", "", "", 0, None  # Added None for image output
+            return "Error: Please upload at least one image to proceed.", "Pending", "", "", 0
 
         milestones = []
-        last_image_path = None  # Track the last saved image path
         for image in images:
             img = Image.open(image)
 
             image_size_mb = os.path.getsize(image) / (1024 * 1024)
             if image_size_mb > 20:
-                return "Error: One or more images exceed 20MB.", "Failure", "", "", 0, None
+                return "Error: One or more images exceed 20MB.", "Failure", "", "", 0
             if not str(image).lower().endswith(('.jpg', '.jpeg', '.png')):
-                return "Error: Only JPG/PNG images are supported.", "Failure", "", "", 0, None
+                return "Error: Only JPG/PNG images are supported.", "Failure", "", "", 0
 
             upload_dir = "public_uploads"
             os.makedirs(upload_dir, exist_ok=True)
@@ -70,7 +69,6 @@ def process_image(images, project_name):
             image_filename = f"{unique_id}_{os.path.basename(image)}"
             saved_image_path = os.path.join(upload_dir, image_filename)
             shutil.copy(image, saved_image_path)
-            last_image_path = saved_image_path  # Update last image path
 
             with open(saved_image_path, 'rb') as image_file:
                 image_data = base64.b64encode(image_file.read()).decode('utf-8')
@@ -85,7 +83,7 @@ def process_image(images, project_name):
                 content_version_id = content_version_result['id']
                 file_url = f"https://sathkruthatechsolutionspri8-dev-ed.develop.lightning.force.com/{content_version_id}"
             except Exception as e:
-                return f"Error: Failed to upload image to Salesforce - {str(e)}", "Failure", "", "", 0, None
+                return f"Error: Failed to upload image to Salesforce - {str(e)}", "Failure", "", "", 0
 
             milestone = detect_milestone_from_image(saved_image_path)
             milestones.append(milestone)
@@ -115,25 +113,25 @@ def process_image(images, project_name):
         try:
             sf.Construction__c.create(record)
         except Exception as e:
-            return f"Error: Failed to update Salesforce - {str(e)}", "Failure", "", "", 0, None
-
-        # Return last image path for display (filename not shown in UI)
-        return (
-            f"<h3 style='color:green;'>Milestone: {final_milestone}</h3>",
-            "Success",
-            final_milestone,
-            f"{percent_complete}%",
-            last_image_path  # Added to return the last image for display
-        )
+            return f"Error: Failed to update Salesforce - {str(e)}", "Failure", "", "", 0
+
+        return f"<h3 style='color:green;'>Milestone: {final_milestone}</h3>", "Success", final_milestone, f"{percent_complete}%"
 
     except Exception as e:
-        return f"Error: {str(e)}", "Failure", "", "", "0%", None
+        return f"Error: {str(e)}", "Failure", "", "", "0%"
 
-# Gradio UI with added image output
+# Gradio UI with modified input to show image preview
 with gr.Blocks(css="") as demo:
     gr.Markdown("<h1 class='title'>Construction Progress Analyzer</h1>")
     with gr.Row():
-        image_input = gr.Files(type="filepath", label="Upload Construction Site Photos (JPG/PNG, ≤ 20MB)")
+        # 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_share_button=False
+        )
         project_name_input = gr.Textbox(label="Project Name (Required)", placeholder="e.g. Project_12345")
 
     submit_button = gr.Button("Process Image")
@@ -141,12 +139,11 @@ with gr.Blocks(css="") as demo:
     upload_status = gr.Textbox(label="Upload Status")
     milestone = gr.Textbox(label="Detected Milestone")
     progress = gr.Textbox(label="Completion Percentage", interactive=False)
-    image_output = gr.Image(label="Uploaded Image", interactive=False)  # Added to display image
 
     submit_button.click(
         fn=process_image,
         inputs=[image_input, project_name_input],
-        outputs=[output_html, upload_status, milestone, progress, image_output]  # Added image_output
+        outputs=[output_html, upload_status, milestone, progress]
     )
 
 demo.launch(share=True)