Update app.py

app.py

@@ -16,7 +16,7 @@ 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 thresholding the grayscale image.
+    Analyze UV sterilization coverage by multiplexing the grayscale image.
     Optional adaptive thresholding and Gaussian blur for noise reduction.
     Morphological operations clean the mask for better accuracy.
     """
@@ -84,28 +84,27 @@ def create_pdf_report(coverage_percent, extracted_texts, annotated_image_path, o
 
 def upload_image_and_get_url(image_path):
     """
-    Upload the image to a public directory in Hugging Face Space and return a public URL.
-    Assumes the app is hosted in a Hugging Face Space with a public /uploads folder.
+    Upload the image to the uploads directory in Hugging Face Space and return a public URL.
     """
     import shutil
     import os
     from datetime import datetime
     import uuid
 
-    # Define the public uploads directory in the Hugging Face Space
-    public_dir = os.path.join("/home/user/app/uploads")  # Adjust path based on Hugging Face Space structure
+    # 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
 
     # 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 public directory
+    # Copy the image to the uploads directory
     shutil.copy(image_path, destination_path)
 
-    # Construct the public URL (adjust the base URL to match your Hugging Face Space)
-    space_url = os.environ.get('HF_SPACE_URL', 'https://your-username-your-space-name.hf.space')  # Replace with your Space URL
-    public_url = f"{space_url}/uploads/{unique_filename}"
+    # 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}"
 
     return public_url