Update app.py

app.py

@@ -315,7 +315,8 @@ def process_video(video_data):
 
 def gradio_interface(video_file):
     if not video_file:
-        return "No file uploaded.", "", "No file uploaded.", "", ""
+        return "", "", "", "", ""  # Return empty outputs when no file uploaded
+    
     try:
         yield "Processing video... please wait.", "", "", "", ""
 
@@ -324,25 +325,26 @@ def gradio_interface(video_file):
 
         result = process_video(video_data)
 
-        if result.get("message"):
-            yield result["message"], "", "", "", ""
+        # Only output violations if detected, else output nothing or minimal
+        if not result["violations"]:
+            # No violations detected — return empty or minimal outputs
+            yield "", "", "", "", ""
             return
 
-        violation_table = "No violations detected."
-        if result["violations"]:
-            header = "| Violation              | Timestamp (s) | Confidence | Bounding Box             |\n"
-            separator = "|------------------------|---------------|------------|--------------------------|\n"
-            rows = []
-            violation_name_map = CONFIG["DISPLAY_NAMES"]
-            for v in result["violations"]:
-                display_name = violation_name_map.get(v["violation"], v["violation"])
-                row = f"| {display_name:<22} | {v['timestamp']:.2f}        | {v['confidence']:.2f}      | {v['bounding_box']} |"
-                rows.append(row)
-            violation_table = header + separator + "\n".join(rows)
-
-        snapshots_text = "No snapshots captured."
+        # Build violation table only if violations exist
+        header = "| Violation              | Timestamp (s) | Confidence | Bounding Box             |\n"
+        separator = "|------------------------|---------------|------------|--------------------------|\n"
+        rows = []
+        violation_name_map = CONFIG["DISPLAY_NAMES"]
+        for v in result["violations"]:
+            display_name = violation_name_map.get(v["violation"], v["violation"])
+            row = f"| {display_name:<22} | {v['timestamp']:.2f}        | {v['confidence']:.2f}      | {v['bounding_box']} |"
+            rows.append(row)
+        violation_table = header + separator + "\n".join(rows)
+
+        # Prepare snapshots if available
+        snapshots_text = ""
         if result["snapshots"]:
-            violation_name_map = CONFIG["DISPLAY_NAMES"]
             snapshots_text = "\n".join(
                 f"- Snapshot for {violation_name_map.get(s['violation'], s['violation'])} at frame {s['frame']}: ![]({s['snapshot_base64']})"
                 for s in result["snapshots"]
@@ -357,7 +359,7 @@ def gradio_interface(video_file):
         )
     except Exception as e:
         logger.error(f"Error in Gradio interface: {e}", exc_info=True)
-        yield f"Error: {str(e)}", "", "Error in processing.", "", ""
+        yield f"Error: {str(e)}", "", "", "", ""
 
 interface = gr.Interface(
     fn=gradio_interface,