Update app.py

app.py

@@ -45,31 +45,23 @@ def process_image(images, project_name):
 
         milestones = []
         for image in images:
-            img = Image.open(image)
+            img = Image.open(image.name)  # Access image through the name attribute
             
             # Check if the image exceeds the size limit (20MB)
-            image_size_mb = os.path.getsize(image) / (1024 * 1024)
+            image_size_mb = len(image.read()) / (1024 * 1024)
             if image_size_mb > 20:
                 return "Error: One or more images exceed 20MB.", "Failure", "", "", 0
-            if not str(image).lower().endswith(('.jpg', '.jpeg', '.png')):
+            if not str(image.name).lower().endswith(('.jpg', '.jpeg', '.png')):
                 return "Error: Only JPG/PNG images are supported.", "Failure", "", "", 0
 
-            # Save image to public folder temporarily before uploading to Salesforce
-            upload_dir = "public_uploads"
-            os.makedirs(upload_dir, exist_ok=True)
-            unique_id = datetime.now().strftime("%Y%m%d%H%M%S")
-            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)
-
             # Convert image to base64 before uploading to Salesforce
-            with open(saved_image_path, 'rb') as image_file:
+            with open(image.name, 'rb') as image_file:
                 image_data = base64.b64encode(image_file.read()).decode('utf-8')
 
             # Create the ContentVersion record in Salesforce
             content_version = {
-                'Title': image_filename,
-                'PathOnClient': saved_image_path,
+                'Title': image.name,
+                'PathOnClient': image.name,
                 'VersionData': image_data
             }
 
@@ -82,7 +74,7 @@ def process_image(images, project_name):
                 return f"Error: Failed to upload image to Salesforce - {str(e)}", "Failure", "", "", 0
 
             # Enhanced Milestone Detection
-            milestone = detect_milestone(image)
+            milestone = detect_milestone(image.name)
 
             milestones.append(milestone)
 
@@ -97,71 +89,6 @@ def process_image(images, project_name):
         }
         percent_complete = milestone_completion_map.get(final_milestone, 0)
 
-        completion_details = {
-            "Planning": {
-                "completed": [
-                    "Initial project outline and objectives have been established.",
-                    "Preliminary designs and architectural plans are drafted.",
-                    "Stakeholder meetings and initial approvals are completed."
-                ],
-                "not_completed": [
-                    "Detailed construction plans and blueprints are pending finalization.",
-                    "Permits and regulatory approvals are yet to be obtained.",
-                    "Contractor selection and procurement processes are not yet complete."
-                ]
-            },
-            "Foundation": {
-                "completed": [
-                    "Site preparation, including clearing and leveling, is finished.",
-                    "Excavation for the foundation has been completed.",
-                    "Concrete pouring for the foundation, including footings and slabs, is done.",
-                    "Initial structural inspections for the foundation have been passed."
-                ],
-                "not_completed": [
-                    "Plumbing and electrical groundwork installations are pending.",
-                    "Backfilling and site grading around the foundation are not yet done.",
-                    "Above-ground structural work, such as columns and walls, has not started."
-                ]
-            },
-            "Walls Erected": {
-                "completed": [
-                    "The concrete framework, including columns and beams, is in place.",
-                    "All structural walls have been erected and stabilized.",
-                    "Temporary scaffolding and safety measures are installed for ongoing work.",
-                    "Initial inspections for structural integrity have been completed."
-                ],
-                "not_completed": [
-                    "Roofing installation and weatherproofing are pending.",
-                    "Windows, doors, and exterior cladding are not yet installed.",
-                    "Interior walls, electrical, and plumbing systems are still to be implemented."
-                ]
-            },
-            "Interior Furnishing": {
-                "completed": [
-                    "Interior walls are fully constructed and painted.",
-                    "Windows, doors, and roofing are complete.",
-                    "HVAC, electrical, and plumbing are fully installed.",
-                    "Flooring, fixtures, and finishes are nearing completion."
-                ],
-                "not_completed": [
-                    "Furniture installation and final touches are pending.",
-                    "Final inspection and approval from authorities are yet to be completed."
-                ]
-            },
-            "Completed": {
-                "completed": [
-                    "The concrete framework, including columns, beams, and floor slabs, is fully constructed.",
-                    "Exterior walls, windows, and cladding are installed, completing the building's facade.",
-                    "Interior work, including electrical, plumbing, and HVAC systems, is fully implemented.",
-                    "Finishing touches, such as flooring, painting, and fixtures, are completed.",
-                    "All phases of the project are finished, including final inspections and approvals."
-                ],
-                "not_completed": [
-                    "There should be no more pending work as the project is fully completed."
-                ]
-            }
-        }
-
         completed_tasks = completion_details[final_milestone]["completed"]
         not_completed_tasks = completion_details[final_milestone]["not_completed"]
 
@@ -303,10 +230,10 @@ 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)")
+        image_input = gr.Files(type="file", label="Upload Construction Site Photos (JPG/PNG, ≤ 20MB)", multi=True)  # Allow multiple uploads
         project_name_input = gr.Textbox(label="Project Name (Required)", placeholder="e.g. Project_12345")
 
-    submit_button = gr.Button("Process Image")
+    submit_button = gr.Button("Process Images")
     output_html = gr.HTML(label="Result")
     upload_status = gr.Textbox(label="Upload Status")
     milestone = gr.Textbox(label="Detected Milestone")