Update app.py

app.py

@@ -10,13 +10,18 @@ from simple_salesforce import Salesforce
 from datetime import datetime
 import base64
 import io
+import logging
+
+# Set up logging to debug file writing issues
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 # Initialize PaddleOCR once with updated parameters
 ocr_model = PaddleOCR(use_textline_orientation=True, lang='en')
 
 def analyze_uv_coverage(img, brightness_threshold=150, kernel_size=5, apply_blur=True, adaptive_thresh=False):
     """
-    Analyze UV sterilization coverage by multiplexing the grayscale image.
+    Analyze UV sterilization coverage by thresholding the grayscale image.
     Optional adaptive thresholding and Gaussian blur for noise reduction.
     Morphological operations clean the mask for better accuracy.
     """
@@ -91,20 +96,39 @@ def upload_image_and_get_url(image_path):
     from datetime import datetime
     import uuid
 
-    # Define the uploads directory in the Hugging Face Space
-    public_dir = os.path.join(os.getcwd(), "uploads")  # Use current working directory + uploads
-    os.makedirs(public_dir, exist_ok=True)  # Create directory if it doesn't exist
+    # 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")
+    
+    # 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"
     destination_path = os.path.join(public_dir, unique_filename)
 
     # Copy the image to the uploads directory
-    shutil.copy(image_path, destination_path)
+    try:
+        shutil.copy(image_path, destination_path)
+        # Ensure the file is readable
+        os.chmod(destination_path, 0o664)
+        logger.info(f"Successfully copied image to: {destination_path}")
+    except Exception as e:
+        logger.error(f"Failed to copy image to {destination_path}: {e}")
+        raise
 
     # 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}")
 
     return public_url