Update app.py

app.py

@@ -257,11 +257,7 @@ def process_video(video_data):
             "snapshots": snapshots,
             "score": score,
             "salesforce_record_id": report_id,
-            "violation_details_url": final_pdf_url,
-            "violation_details_content": "\n".join(
-                f"{v['violation']} at {v['timestamp']:.2f}s (Confidence: {v['confidence']})"
-                for v in violations
-            ) or "No violations detected."
+            "violation_details_url": final_pdf_url
         }
     except Exception as e:
         logger.error(f"Error processing video: {e}")
@@ -270,8 +266,7 @@ def process_video(video_data):
             "snapshots": [],
             "score": 0,
             "salesforce_record_id": None,
-            "violation_details_url": "",
-            "violation_details_content": f"Error: {str(e)}"
+            "violation_details_url": ""
         }
 
 # ==========================
@@ -279,7 +274,7 @@ def process_video(video_data):
 # ==========================
 def gradio_interface(video_file):
     if not video_file:
-        return "No file uploaded.", "", "No file uploaded.", "", "", "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()
@@ -311,12 +306,11 @@ def gradio_interface(video_file):
             f"Safety Score: {result['score']}%",
             snapshots_text,
             f"Salesforce Record ID: {result['salesforce_record_id'] or 'N/A'}",
-            f"Violation Details URL: {result['violation_details_url'] or 'N/A'}",
-            result["violation_details_content"]
+            f"Violation Details URL: {result['violation_details_url'] or 'N/A'}"
         )
     except Exception as e:
         logger.error(f"Error in Gradio interface: {e}")
-        return f"Error: {str(e)}", "", "Error in processing.", "", "", "Error in processing."
+        return f"Error: {str(e)}", "", "Error in processing.", "", "Error in processing."
 
 interface = gr.Interface(
     fn=gradio_interface,
@@ -326,8 +320,7 @@ interface = gr.Interface(
         gr.Textbox(label="Compliance Score"),
         gr.Markdown(label="Snapshots"),
         gr.Textbox(label="Salesforce Record ID"),
-        gr.Textbox(label="Violation Details URL"),
-        gr.Textbox(label="Violation Details Content")
+        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)."