Update app.py

app.py

@@ -22,8 +22,8 @@ MODEL_PATH = os.getenv("SAFETY_MODEL_PATH", DEFAULT_MODEL_PATH)
 STATIC_OUTPUT_DIR = "static/output"
 os.makedirs(STATIC_OUTPUT_DIR, exist_ok=True)
 
-# Base URL for publicly accessible files (adjust based on your hosting platform)
-BASE_PUBLIC_URL = "https://huggingface.co/spaces/PrashanthB461/AI_Safety_Demo1/resolve/main/static/output/"
+# Base URL for publicly accessible files (Hugging Face Spaces)
+BASE_PUBLIC_URL = "https://huggingface.co/spaces/PrashanthB461/AI_Safety_Demo1/raw/main/static/output/"
 
 VIOLATION_LABELS = {
     0: "no_helmet",
@@ -163,6 +163,10 @@ def generate_pdf_report(violations, snapshots, score):
         c.save()
         print(f"PDF generated at {pdf_path}")
 
+        # Verify the file exists
+        if not os.path.exists(pdf_path):
+            raise Exception(f"PDF file not found at {pdf_path}")
+
         # Generate public URL for the PDF
         pdf_url = f"{BASE_PUBLIC_URL}{pdf_filename}"
         print(f"Public PDF URL: {pdf_url}")
@@ -251,7 +255,8 @@ def process_video(video_data, frame_skip=5, max_frames=100):
 
         # Create Salesforce record with the PDF URL
         record_id = push_report_to_salesforce(score, violations, pdf_url)
-        print(f"Salesforce record created with Id: {record_id}")
+        if not record_id:
+            print("⚠️ Salesforce record creation failed, but PDF URL is available")
 
         with open(pdf_path, "rb") as f:
             pdf_base64 = base64.b64encode(f.read()).decode('utf-8')