Update app.py

app.py

@@ -34,7 +34,6 @@ CONFIG = {
     },
     "PUBLIC_URL_BASE": "https://huggingface.co/spaces/PrashanthB461/AI_Safety_Demo1/resolve/main/static/output/",
     "FRAME_SKIP": 5,
-    "MAX_FRAMES": 100,
     "MAX_PROCESSING_TIME": 30
 }
 
@@ -206,10 +205,10 @@ def process_video(video_data):
             raise ValueError("Could not open video file")
 
         violations, snapshots = [], []
-        frame_count, processed_frames = 0, 0
+        frame_count = 0
         start_time = time.time()
 
-        while processed_frames < CONFIG["MAX_FRAMES"]:
+        while True:
             ret, frame = video.read()
             if not ret:
                 break
@@ -241,7 +240,6 @@ def process_video(video_data):
                     })
 
             frame_count += 1
-            processed_frames += 1
             if time.time() - start_time > CONFIG["MAX_PROCESSING_TIME"]:
                 logger.warning("Processing time limit exceeded")
                 break
@@ -288,24 +286,19 @@ def gradio_interface(video_file):
             rows = []
             for v in result["violations"]:
                 violation_name = v["violation"]
-                # Replace specific violation names for display
                 if violation_name == "no_helmet":
-                    violation_name = "no helmet"
-                elif violation_name == "no_harness":
-                    violation_name = "no harness"
-                elif violation_name == "unsafe_posture":
-                    violation_name = "unsafe posture"
-                elif violation_name == "unsafe_zone":
-                    violation_name = "unsafe zone"
-                row = f"| {violation_name:<13} | {v['timestamp']:.2f}s  | {v['confidence']:.2f}      | {v['bounding_box']} | [Details]({result['violation_details_url']}) |"
+                    violation_name = "no_helmet"
+                else:
+                    violation_name = violation_name.replace("no_", "").replace("unsafe_", "")
+                row = f"| {violation_name:<13} | {v['timestamp']:.2f}s  | {v['confidence']:.2f}      | {v['bounding_box']} | {result['violation_details_url']} |"
                 rows.append(row)
             violation_table = header + separator + "\n".join(rows)
 
-        # Format snapshots as a bullet list with clickable image links
+        # Format snapshots as a bullet list
         snapshots_text = "No snapshots captured."
         if result["snapshots"]:
             snapshots_text = "\n".join(
-                f"- Snapshot for {s['violation'].replace('no_', '').replace('unsafe_', '')} at frame {s['frame']}: [![Snapshot]({s['snapshot_url']})]({s['snapshot_url']})"
+                f"- Snapshot for {s['violation'].replace('no_', '').replace('unsafe_', '')} at frame {s['frame']}: [{s['snapshot_url']}]({s['snapshot_url']})"
                 for s in result["snapshots"]
             )
 
@@ -314,7 +307,7 @@ def gradio_interface(video_file):
             f"Safety Score: {result['score']}%",
             snapshots_text,
             f"Salesforce Record ID: {result['salesforce_record_id'] or 'N/A'}",
-            result["violation_details_url"] or 'N/A'
+            result["violation_details_url"] or "N/A"
         )
     except Exception as e:
         logger.error(f"Error in Gradio interface: {e}")