Update app.py

app.py

@@ -5,7 +5,6 @@ import torch
 import numpy as np
 from ultralytics import YOLO
 import time
-import base64
 from simple_salesforce import Salesforce
 
 # ==========================
@@ -69,19 +68,31 @@ def violations_to_text(violations):
 
 def generate_violation_text_file(violations):
     try:
+        # Ensure the directory is writable
+        if not os.access(STATIC_OUTPUT_DIR, os.W_OK):
+            raise PermissionError(f"Cannot write to directory: {STATIC_OUTPUT_DIR}")
+
         text_filename = f"violations_{int(time.time())}.txt"
         text_path = os.path.join(STATIC_OUTPUT_DIR, text_filename)
+        
+        # Generate the content
+        content = "Worksite Safety Violation Details\n" + "=" * 40 + "\n"
+        content += violations_to_text(violations)
+        
+        # Write the file
         with open(text_path, "w") as f:
-            f.write("Worksite Safety Violation Details\n")
-            f.write("=" * 40 + "\n")
-            f.write(violations_to_text(violations))
+            f.write(content)
+        
         print(f"✅ Text file generated at {text_path}")
+        if not os.path.exists(text_path):
+            raise FileNotFoundError(f"Text file was not created at {text_path}")
+        
         public_url = f"https://huggingface.co/spaces/PrashanthB461/AI_Safety_Demo1/resolve/main/static/output/{text_filename}"
         print(f"✅ Text file URL: {public_url}")
-        return text_path, public_url
+        return text_path, public_url, content
     except Exception as e:
         print(f"❌ Error generating text file: {e}")
-        return "", ""
+        return "", "", "Error: Could not generate violation details."
 
 def push_report_to_salesforce(score, violations, text_filename):
     try:
@@ -103,7 +114,7 @@ def push_report_to_salesforce(score, violations, text_filename):
         print(f"Attempting to create Salesforce record with data: {record_data}")
         record = sf.Safety_Video_Report__c.create(record_data)
         print(f"✅ Salesforce record created with Id: {record.get('id')}")
-        return record.get("id")
+        return record.get('id')
     except Exception as e:
         print(f"❌ Salesforce record creation failed: {e}")
         return None
@@ -193,7 +204,7 @@ def process_video(video_data, frame_skip=5, max_frames=100):
         os.remove(video_path)
 
         score = calculate_safety_score(violations)
-        text_path, text_url = generate_violation_text_file(violations)
+        text_path, text_url, text_content = generate_violation_text_file(violations)
 
         if text_path:
             report_id = push_report_to_salesforce(score, violations, text_path)
@@ -207,7 +218,8 @@ def process_video(video_data, frame_skip=5, max_frames=100):
             "snapshots": snapshots,
             "score": score,
             "salesforce_record_id": report_id,
-            "violation_details_url": text_url
+            "violation_details_url": text_url,
+            "violation_details_content": text_content
         }
 
     except Exception as e:
@@ -218,6 +230,7 @@ def process_video(video_data, frame_skip=5, max_frames=100):
             "score": 0,
             "salesforce_record_id": None,
             "violation_details_url": "",
+            "violation_details_content": "Error: Could not process video.",
             "error": str(e)
         }
 
@@ -227,7 +240,7 @@ def process_video(video_data, frame_skip=5, max_frames=100):
 def gradio_interface(video_file):
     try:
         if not video_file:
-            return {"error": "Please upload a video file."}, "", [], ""
+            return {"error": "Please upload a video file."}, "", [], "", "", "No file uploaded."
 
         with open(video_file, "rb") as f:
             video_data = f.read()
@@ -238,11 +251,12 @@ def gradio_interface(video_file):
             f"Safety Score: {result.get('score', 0)}%",
             result.get("snapshots", []),
             f"Salesforce Record ID: {result.get('salesforce_record_id', '')}",
-            f"Violation Details URL: {result.get('violation_details_url', '')}"
+            f"Violation Details URL: {result.get('violation_details_url', '')}",
+            result.get("violation_details_content", "No content available.")
         )
     except Exception as e:
         print(f"❌ Error in gradio_interface: {e}")
-        return {"error": str(e)}, "", [], "", ""
+        return {"error": str(e)}, "", [], "", "", "Error in processing."
 
 interface = gr.Interface(
     fn=gradio_interface,
@@ -252,7 +266,8 @@ interface = gr.Interface(
         gr.Textbox(label="Compliance Score"),
         gr.JSON(label="Snapshots"),
         gr.Textbox(label="Salesforce Record ID"),
-        gr.Textbox(label="Violation Details URL")
+        gr.Textbox(label="Violation Details URL"),
+        gr.Textbox(label="Violation Details Content (Preview)")
     ],
     title="Worksite Safety Violation Analyzer",
     description="Upload short site videos to detect safety violations (e.g., no helmet, no harness, unsafe posture)."