Update app.py

app.py

@@ -48,7 +48,7 @@ except Exception as e:
     raise
 
 # ==========================
-# Salesforce connection (hardcoded creds for testing)
+# Salesforce connection
 # ==========================
 def connect_to_salesforce():
     sf = Salesforce(
@@ -66,15 +66,29 @@ def violations_to_text(violations):
         lines.append(line)
     return "\n".join(lines)
 
-def push_report_to_salesforce(score, violations, pdf_filename):
+def generate_violation_text_file(violations):
+    try:
+        text_filename = f"violations_{int(time.time())}.txt"
+        text_path = os.path.join(STATIC_OUTPUT_DIR, text_filename)
+        with open(text_path, "w") as f:
+            f.write("Worksite Safety Violation Details\n")
+            f.write("=" * 40 + "\n")
+            f.write(violations_to_text(violations))
+        print(f"Text file generated at {text_path}")
+        return text_path
+    except Exception as e:
+        print(f"❌ Error generating text file: {e}")
+        return ""
+
+def push_report_to_salesforce(score, violations, pdf_filename, text_filename):
     sf = connect_to_salesforce()
 
     violations_count = len(violations)
     violations_details = violations_to_text(violations)
 
-    # Updated public URL base with static/output path
-    PUBLIC_PDF_BASE_URL = "https://huggingface.co/spaces/PrashanthB461/AI_Safety_Demo1/resolve/main/static/output/"
-    pdf_url = f"{PUBLIC_PDF_BASE_URL}{os.path.basename(pdf_filename)}"
+    PUBLIC_BASE_URL = "https://huggingface.co/spaces/PrashanthB461/AI_Safety_Demo1/resolve/main/static/output/"
+    pdf_url = f"{PUBLIC_BASE_URL}{os.path.basename(pdf_filename)}"
+    text_url = f"{PUBLIC_BASE_URL}{os.path.basename(text_filename)}" if text_filename else ""
 
     try:
         record = sf.Safety_Video_Report__c.create({
@@ -82,7 +96,8 @@ def push_report_to_salesforce(score, violations, pdf_filename):
             "Violations_Found__c": violations_count,
             "Violations_Details__c": violations_details,
             "Status__c": "Pending",
-            "PDF_Report_URL__c": pdf_url
+            "PDF_Report_URL__c": pdf_url,
+            "Violation_Details_URL__c": text_url  # New field for text file URL
         })
         print(f"✅ Salesforce record created with Id: {record.get('id')}")
         return record.get("id")
@@ -145,9 +160,7 @@ def generate_pdf_report(violations, snapshots, score):
 
         c.save()
         print(f"PDF generated at {pdf_path}")
-
         return pdf_path
-
     except Exception as e:
         print(f"❌ Error generating PDF: {e}")
         return ""
@@ -223,9 +236,10 @@ def process_video(video_data, frame_skip=5, max_frames=100):
 
         score = calculate_safety_score(violations)
         pdf_path = generate_pdf_report(violations, snapshots, score)
+        text_path = generate_violation_text_file(violations)
 
-        if pdf_path:
-            report_id = push_report_to_salesforce(score, violations, pdf_path)
+        if pdf_path and text_path:
+            report_id = push_report_to_salesforce(score, violations, pdf_path, text_path)
             print(f"Salesforce record created with Id: {report_id}")
         else:
             report_id = None
@@ -291,4 +305,4 @@ interface = gr.Interface(
 
 if __name__ == "__main__":
     print("🚀 Launching Safety Analyzer App...")
-    interface.launch()
+    interface.launch()