Update app.py

app.py

@@ -84,12 +84,14 @@ def create_pdf_report(coverage_percent, extracted_texts, annotated_image_path, o
 
 def upload_image_and_get_url(image_path):
     """
-    TODO: Implement your image upload to public storage here.
-    For now, returns a placeholder URL.
+    Instead of uploading, encode image file as base64 data URI
+    and return as a valid URL string to store in Salesforce.
     """
-    # Example: upload to AWS S3, Azure Blob Storage, or other service
-    # Return the public URL to the uploaded image
-    return "https://example.com/path/to/your/annotated_image.jpg"
+    with open(image_path, "rb") as img_file:
+        img_bytes = img_file.read()
+        encoded_str = base64.b64encode(img_bytes).decode('utf-8')
+        data_uri = f"data:image/jpeg;base64,{encoded_str}"
+    return data_uri
 
 def save_record_to_salesforce(annotated_image_url, coverage_percent, original_image_pil, compliance_threshold=80):
     sf = Salesforce(
@@ -99,10 +101,12 @@ def save_record_to_salesforce(annotated_image_url, coverage_percent, original_im
         domain=os.environ.get('SF_DOMAIN', 'login')  # 'test' for sandbox
     )
     
-    # Encode original image to base64 string for storage
+    # Encode original image to base64 data URI string for storage
     buffered = io.BytesIO()
     original_image_pil.save(buffered, format="JPEG")
-    original_img_b64 = base64.b64encode(buffered.getvalue()).decode('utf-8')
+    original_img_bytes = buffered.getvalue()
+    original_img_b64 = base64.b64encode(original_img_bytes).decode('utf-8')
+    original_img_data_uri = f"data:image/jpeg;base64,{original_img_b64}"
 
     compliance_status = 'Pass' if coverage_percent >= compliance_threshold else 'Fail'
     technician_id = os.environ.get('SF_TECHNICIAN_ID')  # Salesforce UserId lookup
@@ -113,7 +117,7 @@ def save_record_to_salesforce(annotated_image_url, coverage_percent, original_im
         'Name': record_name,
         'Annotated_Image__c': annotated_image_url,
         'Coverage_Percentage__c': round(coverage_percent, 2),
-        'Original_Image__c': original_img_b64,
+        'Original_Image__c': original_img_data_uri,
         'Compliance_Status__c': compliance_status,
         'Technician_ID__c': technician_id,
         'Verified_On__c': datetime.utcnow().isoformat()
@@ -151,7 +155,7 @@ def process_image(input_img, brightness_threshold=150):
     temp_pdf_file.close()
     create_pdf_report(coverage_percent, extracted_texts, annotated_img_path, temp_pdf_file.name)
 
-    # Upload annotated image and get URL
+    # Upload annotated image and get URL (now base64 data URI)
     annotated_image_url = upload_image_and_get_url(annotated_img_path)
 
     # Save record in Salesforce