Update app.py

app.py

@@ -70,7 +70,7 @@ def process_image(image, project_name):
         if not str(image).lower().endswith(('.jpg', '.jpeg', '.png')):
             return "Error: Only JPG/PNG images are supported.", "Failure", "", "", 0
 
-        # Save image to public folder
+        # Save image to a temporary file
         upload_dir = "public_uploads"
         os.makedirs(upload_dir, exist_ok=True)
         unique_id = datetime.now().strftime("%Y%m%d%H%M%S")
@@ -78,14 +78,23 @@ def process_image(image, project_name):
         saved_image_path = os.path.join(upload_dir, image_filename)
         shutil.copy(image, saved_image_path)
 
-        # Corrected public URL logic
-        if os.getenv("GRADIO_SERVER_NAME"):
-            public_url_base = f"https://{os.getenv('GRADIO_SERVER_NAME')}/file"
-        else:
-            public_url_base = "http://localhost:7860/file"
-
-        image_url = f"{public_url_base}/{upload_dir}/{image_filename}"
+        # Upload to Salesforce as a ContentVersion
+        try:
+            with open(saved_image_path, 'rb') as file:
+                content_version_data = {
+                    'Title': image_filename,
+                    'PathOnClient': image_filename,
+                    'VersionData': file,
+                    'FirstPublishLocationId': 'your_sobject_id',  # Replace with the appropriate Salesforce object ID (e.g., Project or Construction)
+                }
+                content_version = sf.ContentVersion.create(content_version_data)
+
+            # Generate download URL for the uploaded image
+            file_url = f"https://{sf.sf_instance}/sfc/servlet.shepherd/version/download/{content_version['id']}"
+        except Exception as e:
+            return f"Error: Failed to upload image to Salesforce - {str(e)}", "Failure", "", "", 0
 
+        # AI prediction
         milestone, percent_complete, confidence_score = mock_ai_model(img)
 
         record = {
@@ -95,9 +104,10 @@ def process_image(image, project_name):
             "Last_Updated_On__c": datetime.now().isoformat(),
             "Upload_Status__c": "Success",
             "Comments__c": f"AI Prediction: {milestone} with {confidence_score*100}% confidence",
-            "Last_Updated_Image__c": image_url
+            "Last_Updated_Image__c": file_url
         }
 
+        # Insert record into Salesforce
         try:
             sf.Construction__c.create(record)
         except Exception as e: