Update app.py

app.py

@@ -156,7 +156,7 @@ def push_report_to_salesforce(violations, score, pdf_path, pdf_file):
         pdf_url = f"{CONFIG['PUBLIC_URL_BASE']}{os.path.basename(pdf_path)}" if pdf_path else ""
 
         record_data = {
-            "Compliance_Score__c": score,
+            "Compliance Score__c": score,
             "Violations_Found__c": len(violations),
             "Violations_Details__c": violations_text,
             "Status__c": "Pending",
@@ -281,10 +281,11 @@ def gradio_interface(video_file):
         # Format violations as a Markdown table
         violation_table = "No violations detected."
         if result["violations"]:
-            header = "| Violation     | Timestamp | Confidence | Bounding Box             | Violation Details |\n"
-            separator = "|---------------|-----------|------------|--------------------------|-------------------|\n"
+            header = "| Violation     | Timestamp | Confidence | Bounding Box             | Violation Details       |\n"
+            separator = "|---------------|-----------|------------|--------------------------|-------------------------|\n"
             rows = []
             for v in result["violations"]:
+                # Simplify violation names but preserve "no_helmet" for table
                 violation_name = v["violation"]
                 if violation_name == "no_helmet":
                     violation_name = "no_helmet"
@@ -294,7 +295,7 @@ def gradio_interface(video_file):
                 rows.append(row)
             violation_table = header + separator + "\n".join(rows)
 
-        # Format snapshots as a bullet list
+        # Format snapshots as a bullet list with direct image links
         snapshots_text = "No snapshots captured."
         if result["snapshots"]:
             snapshots_text = "\n".join(
@@ -321,7 +322,7 @@ interface = gr.Interface(
         gr.Textbox(label="Compliance Score"),
         gr.Markdown(label="Snapshots"),
         gr.Textbox(label="Salesforce Record ID"),
-        gr.Textbox(label="URL")
+        gr.Textbox(label="Violation Details URL")
     ],
     title="Worksite Safety Violation Analyzer",
     description="Upload short site videos to detect safety violations (e.g., no helmet, no harness, unsafe posture)."