Update app.py

app.py

@@ -4,6 +4,7 @@ import numpy as np
 from fpdf import FPDF
 import tempfile
 import os
+import shutil
 from paddleocr import PaddleOCR
 import time
 from simple_salesforce import Salesforce
@@ -90,42 +91,60 @@ def create_pdf_report(coverage_percent, extracted_texts, annotated_image_path, o
 def upload_image_and_get_url(image_path):
     """
     Upload the image to the uploads directory in Hugging Face Space and return a public URL.
+    Use /tmp as a temporary storage location, then copy to the persistent uploads folder.
     """
     import shutil
     import os
     from datetime import datetime
     import uuid
 
-    # Define the uploads directory in the Hugging Face Space's persistent storage
-    # Use the repository root (where the Space files are stored)
-    space_root = os.getenv("HF_SPACE_STORAGE", "/home/user/app")  # Fallback to default path
-    public_dir = os.path.join(space_root, "uploads")
-    
+    # Step 1: Write to /tmp (guaranteed writable in Hugging Face Spaces)
+    temp_dir = "/tmp/uploads"
+    try:
+        os.makedirs(temp_dir, exist_ok=True)
+        logger.info(f"Created temporary directory at: {temp_dir}")
+    except Exception as e:
+        logger.error(f"Failed to create temporary directory {temp_dir}: {e}")
+        raise
+
+    # Generate a unique filename to avoid conflicts
+    unique_filename = f"{uuid.uuid4().hex}_{datetime.utcnow().strftime('%Y%m%d_%H%M%S')}.jpg"
+    temp_path = os.path.join(temp_dir, unique_filename)
+
+    # Copy the image to the temporary directory
+    try:
+        shutil.copy(image_path, temp_path)
+        os.chmod(temp_path, 0o664)
+        logger.info(f"Successfully copied image to temporary path: {temp_path}")
+    except Exception as e:
+        logger.error(f"Failed to copy image to {temp_path}: {e}")
+        raise
+
+    # Step 2: Define the uploads directory in the Space's repository root
+    # The repository root is typically the current working directory in Hugging Face Spaces
+    repo_root = os.getcwd()  # Should be the root of the Space (e.g., where app.py is)
+    public_dir = os.path.join(repo_root, "uploads")
+
     # Create the uploads directory if it doesn't exist
     try:
         os.makedirs(public_dir, exist_ok=True)
-        # Ensure the directory is writable
         os.chmod(public_dir, 0o775)
         logger.info(f"Created or verified uploads directory at: {public_dir}")
     except Exception as e:
         logger.error(f"Failed to create uploads directory: {e}")
         raise
 
-    # Generate a unique filename to avoid conflicts
-    unique_filename = f"{uuid.uuid4().hex}_{datetime.utcnow().strftime('%Y%m%d_%H%M%S')}.jpg"
+    # Step 3: Copy the file from /tmp to the persistent uploads folder
     destination_path = os.path.join(public_dir, unique_filename)
-
-    # Copy the image to the uploads directory
     try:
-        shutil.copy(image_path, destination_path)
-        # Ensure the file is readable
+        shutil.move(temp_path, destination_path)
         os.chmod(destination_path, 0o664)
-        logger.info(f"Successfully copied image to: {destination_path}")
+        logger.info(f"Successfully moved image to: {destination_path}")
     except Exception as e:
-        logger.error(f"Failed to copy image to {destination_path}: {e}")
+        logger.error(f"Failed to move image to {destination_path}: {e}")
         raise
 
-    # Construct the public URL using the HF_SPACE_URL from .env
+    # Step 4: Construct the public URL using the HF_SPACE_URL from .env
     space_url = os.environ.get('HF_SPACE_URL', 'https://huggingface.co/spaces/pavansuresh/Smartlab_tech')
     public_url = f"{space_url.rstrip('/')}/uploads/{unique_filename}"
     logger.info(f"Generated public URL: {public_url}")
@@ -157,7 +176,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_image_url,  # Correct field API name here
+        'Original_Image__c': original_image_url,
         'Compliance_Status__c': compliance_status,
         'Technician_ID__c': technician_id,
         'Verified_On__c': datetime.utcnow().isoformat()