Update app.py

app.py

@@ -18,8 +18,10 @@ from simple_salesforce import Salesforce
 DEFAULT_MODEL_PATH = "models/yolov8_safety.pt"
 FALLBACK_MODEL = "yolov8n.pt"
 MODEL_PATH = os.getenv("SAFETY_MODEL_PATH", DEFAULT_MODEL_PATH)
-OUTPUT_DIR = "output"
-os.makedirs(OUTPUT_DIR, exist_ok=True)
+
+# Use static/output for publicly accessible files
+STATIC_OUTPUT_DIR = "static/output"
+os.makedirs(STATIC_OUTPUT_DIR, exist_ok=True)
 
 VIOLATION_LABELS = {
     0: "no_helmet",
@@ -46,14 +48,14 @@ except Exception as e:
     raise
 
 # ==========================
-# Salesforce connection with hardcoded credentials
+# Salesforce connection (hardcoded creds for testing)
 # ==========================
 def connect_to_salesforce():
     sf = Salesforce(
         username="prashanth1ai@safety.com",
         password="SaiPrash461",
         security_token="AP4AQnPoidIKPvSvNEfAHyoK",
-        domain="login"  # Use "test" if sandbox
+        domain="login"
     )
     return sf
 
@@ -70,7 +72,8 @@ def push_report_to_salesforce(score, violations, pdf_filename):
     violations_count = len(violations)
     violations_details = violations_to_text(violations)
 
-    PUBLIC_PDF_BASE_URL = "https://huggingface.co/spaces/PrashanthB461/AI_Safety_Demo1/resolve/main/output/"
+    # Updated public URL base with static/output path
+    PUBLIC_PDF_BASE_URL = "https://huggingface.co/spaces/PrashanthB461/AI_Safety_Demo1/resolve/main/static/output/"
     pdf_url = f"{PUBLIC_PDF_BASE_URL}{os.path.basename(pdf_filename)}"
 
     try:
@@ -107,7 +110,7 @@ def calculate_safety_score(violations):
 # ==========================
 def generate_pdf_report(violations, snapshots, score):
     try:
-        pdf_path = os.path.join(OUTPUT_DIR, f"report_{int(time.time())}.pdf")
+        pdf_path = os.path.join(STATIC_OUTPUT_DIR, f"report_{int(time.time())}.pdf")
         c = canvas.Canvas(pdf_path, pagesize=letter)
         width, height = letter
 
@@ -130,7 +133,7 @@ def generate_pdf_report(violations, snapshots, score):
 
             snapshot = next((s for s in snapshots if s["frame"] == v["frame"] and s["violation"] == v["violation"]), None)
             if snapshot:
-                local_img_path = os.path.join(OUTPUT_DIR, os.path.basename(snapshot["snapshot_url"]))
+                local_img_path = os.path.join(STATIC_OUTPUT_DIR, os.path.basename(snapshot["snapshot_url"]))
                 if os.path.exists(local_img_path):
                     img = ImageReader(local_img_path)
                     c.drawImage(img, 50, y - 100, width=200, height=150)
@@ -155,7 +158,7 @@ def generate_pdf_report(violations, snapshots, score):
 def process_video(video_data, frame_skip=5, max_frames=100):
     try:
         print("Processing video data...")
-        video_path = os.path.join(OUTPUT_DIR, f"temp_{int(time.time())}.mp4")
+        video_path = os.path.join(STATIC_OUTPUT_DIR, f"temp_{int(time.time())}.mp4")
         with open(video_path, "wb") as f:
             f.write(video_data)
         print(f"Video saved to {video_path}")
@@ -197,12 +200,12 @@ def process_video(video_data, frame_skip=5, max_frames=100):
                     }
                     violations.append(violation)
 
-                    snapshot_path = os.path.join(OUTPUT_DIR, f"snapshot_{frame_count}_{label}.jpg")
+                    snapshot_path = os.path.join(STATIC_OUTPUT_DIR, f"snapshot_{frame_count}_{label}.jpg")
                     cv2.imwrite(snapshot_path, frame)
                     snapshots.append({
                         "violation": label,
                         "frame": frame_count,
-                        "snapshot_url": f"https://huggingface.co/spaces/PrashanthB461/AI_Safety_Demo1/resolve/main/output/{os.path.basename(snapshot_path)}"
+                        "snapshot_url": f"https://huggingface.co/spaces/PrashanthB461/AI_Safety_Demo1/resolve/main/static/output/{os.path.basename(snapshot_path)}"
                     })
 
             frame_count += 1