Update app.py

app.py

@@ -157,26 +157,27 @@ 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,
-            "Violations_Found__c": len(violations),
-            "Violations_Details__c": violations_text,
-            "Status__c": "Pending",
-            "PDF_Report_URL__c": pdf_url
-        }
-        record = sf.Safety_Video_Report__c.create(record_data)
-        record_id = record["id"]
-        logger.info(f"Salesforce record created: {record_id}")
+            "Compliance_Score__c": scorea) {
+    "Compliance_Score__c": score,
+    "Violations_Found__c": len(violations),
+    "Violations_Details__c": violations_text,
+    "Status__c": "Pending",
+    "PDF_Report_URL__c": pdf_url
+}
+record = sf.Safety_Video_Report__c.create(record_data)
+record_id = record["id"]
+logger.info(f"Salesforce record created: {record_id}")
 
-        if pdf_file:
-            uploaded_url = upload_pdf_to_salesforce(sf, pdf_file, record_id)
-            if uploaded_url:
-                sf.Safety_Video_Report__c.update(record_id, {"PDF_Report_URL__c": uploaded_url})
-                pdf_url = uploaded_url
+if pdf_file:
+    uploaded_url = upload_pdf_to_salesforce(sf, pdf_file, record_id)
+    if uploaded_url:
+        sf.Safety_Video_Report__c.update(record_id, {"PDF_Report_URL__c": uploaded_url})
+        pdf_url = uploaded_url
 
-        return record_id, pdf_url
-    except Exception as e:
-        logger.error(f"Salesforce record creation failed: {e}")
-        return None, ""
+return record_id, pdf_url
+except Exception as e:
+    logger.error(f"Salesforce record creation failed: {e}")
+    return None, ""
 
 # ==========================
 # Safety Score Calculation
@@ -279,26 +280,35 @@ def process_video(video_data):
 # ==========================
 def gradio_interface(video_file):
     if not video_file:
-        return "No video file uploaded.", "", "No snapshots.", "", "", "No file uploaded."
+        return "No file uploaded.", "", "No file uploaded.", "", "", "No file uploaded."
     try:
         with open(video_file, "rb") as f:
             video_data = f.read()
         result = process_video(video_data)
-        
-        # Convert violations to text format
-        violations_text = "\n".join(
-            f"Violation: {v['violation']}, Timestamp: {v['timestamp']:.2f}s, Confidence: {v['confidence']}, Bounding Box: {v['bounding_box']}"
-            for v in result["violations"]
-        ) or "No violations detected."
-        
-        # Convert snapshots to text format
-        snapshots_text = "\n".join(
-            f"Violation: {s['violation']}, Frame: {s['frame']}, Snapshot URL: {s['snapshot_url']}"
-            for s in result["snapshots"]
-        ) or "No snapshots captured."
-        
+
+        # Format violations as a Markdown table
+        violation_table = "No violations detected."
+        if result["violations"]:
+            header = "| Violation     | Timestamp | Confidence | Bounding Box             |\n"
+            separator = "|---------------|-----------|------------|--------------------------|\n"
+            rows = []
+            for v in result["violations"]:
+                # Simplify violation names (e.g., "no_helmet" to "helmet")
+                violation_name = v["violation"].replace("no_", "").replace("unsafe_", "")
+                row = f"| {violation_name:<13} | {v['timestamp']:.2f}s  | {v['confidence']:.2f}      | {v['bounding_box']} |"
+                rows.append(row)
+            violation ladies = header + separator + "\n".join(rows)
+
+        # 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']}: {s['snapshot_url']}"
+                for s in result["snapshots"]
+            )
+
         return (
-            violations_text,
+            violation_table,
             f"Safety Score: {result['score']}%",
             snapshots_text,
             f"Salesforce Record ID: {result['salesforce_record_id'] or 'N/A'}",
@@ -307,15 +317,15 @@ def gradio_interface(video_file):
         )
     except Exception as e:
         logger.error(f"Error in Gradio interface: {e}")
-        return f"Error: {str(e)}", "", "No snapshots.", "", "", "Error in processing."
+        return f"Error: {str(e)}", "", "Error in processing.", "", "", "Error in processing."
 
 interface = gr.Interface(
     fn=gradio_interface,
     inputs=gr.Video(label="Upload Site Video"),
     outputs=[
-        gr.Textbox(label="Detected Safety Violations"),
+        gr.Markdown(label="Detected Safety Violations"),
         gr.Textbox(label="Compliance Score"),
-        gr.Textbox(label="Snapshots"),
+        gr.Markdown(label="Snapshots"),
         gr.Textbox(label="Salesforce Record ID"),
         gr.Textbox(label="Violation Details URL"),
         gr.Textbox(label="Violation Details Content")