Update app.py

app.py

@@ -6,6 +6,7 @@ from simple_salesforce import Salesforce
 from datetime import datetime
 import hashlib
 import shutil
+import base64
 
 # Load environment variables
 load_dotenv()
@@ -78,16 +79,19 @@ def process_image(image, project_name):
         saved_image_path = os.path.join(upload_dir, image_filename)
         shutil.copy(image, saved_image_path)
 
+        # Open the image file and encode it as base64
+        with open(saved_image_path, "rb") as file:
+            encoded_image = base64.b64encode(file.read()).decode('utf-8')
+
         # 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)
+            content_version_data = {
+                'Title': image_filename,
+                'PathOnClient': image_filename,
+                'VersionData': encoded_image,  # Using the base64-encoded image data
+                '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']}"
@@ -144,4 +148,4 @@ with gr.Blocks(css=".gradio-container {background-color: #f0f4f8; font-family: A
         outputs=[output_text, upload_status, milestone, confidence, progress]
     )
 
-demo.launch(share=False)
+demo.launch(share=True)