Update app.py

app.py

@@ -23,7 +23,7 @@ import tempfile
 import shutil
 
 # ========================== # Configuration and Setup # ==========================
-# Use a temporary directory for storage
+# Use a temporary directory for storage to avoid file system issues on Hugging Face Spaces
 TEMP_DIR = tempfile.mkdtemp(prefix="Ultralytics_")
 os.environ['YOLO_CONFIG_DIR'] = TEMP_DIR
 
@@ -31,10 +31,11 @@ os.environ['YOLO_CONFIG_DIR'] = TEMP_DIR
 OUTPUT_DIR = os.path.join(TEMP_DIR, "output")
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
+# Configure logging for better debugging
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 logger = logging.getLogger(__name__)
 
-# Check for FFmpeg availability
+# Check for FFmpeg availability to ensure video processing works
 def check_ffmpeg():
     try:
         subprocess.run(["ffmpeg", "-version"], stdout=subprocess.PIPE, stderr=subprocess.PIPE, check=True)
@@ -215,9 +216,9 @@ CONFIG = {
         "improper_tool_use": "Improper Tool Use"
     },
     "SF_CREDENTIALS": {
-        "username": "prashanth1ai@safety.com",
-        "password": "SaiPrash461",
-        "security_token": "AP4AQnPoidIKPvSvNEfAHyoK",
+        "username": os.getenv("SF_USERNAME", "prashanth1ai@safety.com"),
+        "password": os.getenv("SF_PASSWORD", "SaiPrash461"),
+        "security_token": os.getenv("SF_SECURITY_TOKEN", "AP4AQnPoidIKPvSvNEfAHyoK"),
         "domain": "login"
     },
     "PUBLIC_URL_BASE": "https://huggingface.co/spaces/PrashanthB461/AI_Safety_Demo2/resolve/main/static/output/",
@@ -635,7 +636,7 @@ def process_video(video_data):
                         detection = {
                             "worker_id": worker_id,
                             "violation": label,
-                            "confidence": round(conf, 2),
+                            "confidence": round(float(conf), 2),  # Ensure confidence is a float
                             "bounding_box": bbox,
                             "timestamp": current_time
                         }
@@ -667,7 +668,8 @@ def process_video(video_data):
                             "worker_id": worker_id,
                             "timestamp": current_time,
                             "snapshot_path": snapshot_path,
-                            "snapshot_url": f"{CONFIG['PUBLIC_URL_BASE']}{snapshot_filename}"
+                            "snapshot_url": f"{CONFIG['PUBLIC_URL_BASE']}{snapshot_filename}",
+                            "confidence": round(float(conf), 2)  # Ensure confidence is stored as float
                         })
                         
                         logger.info(f"Captured snapshot for {label} violation by worker {worker_id} at {current_time:.2f}s")
@@ -680,17 +682,28 @@ def process_video(video_data):
         processing_time = time.time() - start_time
         logger.info(f"Processing complete in {processing_time:.2f}s")
 
+        # Log the snapshots for debugging
+        logger.info(f"Snapshots: {snapshots}")
+
         violations = []
         for worker_id, worker_violations in unique_violations.items():
             for label, detection_time in worker_violations.items():
+                # Find the confidence from snapshots, ensuring it's a float
+                confidence = next(
+                    (float(s["confidence"]) for s in snapshots if s["worker_id"] == worker_id and s["violation"] == label),
+                    0.0
+                )
                 violation = {
                     "worker_id": worker_id,
                     "violation": label,
                     "timestamp": detection_time,
-                    "confidence": next((s["confidence"] for s in snapshots if s["worker_id"] == worker_id and s["violation"] == label), 0.0)
+                    "confidence": confidence
                 }
                 violations.append(violation)
 
+        # Log the violations for debugging
+        logger.info(f"Violations: {violations}")
+
         if not violations:
             logger.info("No violations detected after processing")
             yield "No violations detected in the video.", "Safety Score: 100%", "No snapshots captured.", "N/A", "N/A"
@@ -702,6 +715,7 @@ def process_video(video_data):
         # Push to Salesforce with fallback
         record_id, final_pdf_url = push_report_to_salesforce(violations, score, pdf_path, pdf_file)
 
+        # Generate violation table with robust error handling
         violation_table = "| Violation | Worker ID | Time (s) | Confidence |\n"
         violation_table += "|-----------|-----------|----------|------------|\n"
         
@@ -709,7 +723,12 @@ def process_video(video_data):
             display_name = CONFIG["DISPLAY_NAMES"].get(v.get("violation", "Unknown"), "Unknown")
             worker_id = v.get("worker_id", "Unknown")
             timestamp = v.get("timestamp", 0.0)
-            confidence = v.get("confidence", 0.0)
+            # Ensure confidence is a valid float
+            try:
+                confidence = float(v.get("confidence", 0.0))
+            except (ValueError, TypeError) as e:
+                logger.error(f"Invalid confidence value in violation {v}: {e}")
+                confidence = 0.0
             
             violation_table += f"| {display_name} | {worker_id} | {timestamp:.2f} | {confidence:.2f} |\n"