Update app.py

app.py

@@ -45,23 +45,22 @@ def process_image(images, project_name):
 
         milestones = []
         for image in images:
-            img = Image.open(image.name)  # Access image through the name attribute
+            img = Image.open(image)  # Open the image directly from binary data
             
             # Check if the image exceeds the size limit (20MB)
-            image_size_mb = len(image.read()) / (1024 * 1024)
+            image_size_mb = len(image) / (1024 * 1024)
             if image_size_mb > 20:
                 return "Error: One or more images exceed 20MB.", "Failure", "", "", 0
-            if not str(image.name).lower().endswith(('.jpg', '.jpeg', '.png')):
+            if not str(image).lower().endswith(('.jpg', '.jpeg', '.png')):
                 return "Error: Only JPG/PNG images are supported.", "Failure", "", "", 0
 
             # Convert image to base64 before uploading to Salesforce
-            with open(image.name, 'rb') as image_file:
-                image_data = base64.b64encode(image_file.read()).decode('utf-8')
+            image_data = base64.b64encode(image).decode('utf-8')
 
             # Create the ContentVersion record in Salesforce
             content_version = {
-                'Title': image.name,
-                'PathOnClient': image.name,
+                'Title': project_name,  # Assuming project name or a unique image title
+                'PathOnClient': project_name,  # Same here for simplicity
                 'VersionData': image_data
             }
 
@@ -74,7 +73,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.name)
+            milestone = detect_milestone(project_name)
 
             milestones.append(milestone)
 
@@ -230,7 +229,7 @@ with gr.Blocks(css="""
 """) as demo:
     gr.Markdown("<h1 class='title'>Construction Progress Analyzer</h1>")
     with gr.Row():
-        image_input = gr.Files(type="file", label="Upload Construction Site Photos (JPG/PNG, ≤ 20MB)")  # Remove 'multiple' as it's handled internally
+        image_input = gr.Files(type="binary", label="Upload Construction Site Photos (JPG/PNG, ≤ 20MB)")  # Use 'binary' to handle image data
         project_name_input = gr.Textbox(label="Project Name (Required)", placeholder="e.g. Project_12345")
 
     submit_button = gr.Button("Process Images")