Update app.py

app.py

@@ -3,6 +3,7 @@ import logging
 import sys
 import uuid
 from pathlib import Path
+import csv
 import requests
 import gradio as gr
 from reportlab.lib.pagesizes import letter
@@ -22,6 +23,7 @@ SALESFORCE_USERNAME = "vaneshdevarapalli866@agentforce.com"
 SALESFORCE_PASSWORD = "vanesh@331"
 SALESFORCE_SECURITY_TOKEN = "VRUVbBOdG0s9Q4xy0W6DB1Y6b"
 
+HUGGINGFACE_SPACE_URL = "https://huggingface.co/spaces/vaneshdev/equipment-utilization-app"
 HUGGINGFACE_API_URL = "https://your-huggingface-model-endpoint"
 
 # Connect to Salesforce
@@ -53,11 +55,12 @@ project_choices = [
 
 ai_suggestion_choices = ["Move", "Pause Rent", "Repair", "Replace"]
 
-# Generate PDF and return filename and path
+# Generate PDF report
+
 def generate_pdf_report(record_id, data_dict):
     report_id = str(uuid.uuid4())[:8]
     report_filename = f"report_{report_id}.pdf"
-    report_path = Path(f"reports/{report_filename}")
+    report_path = Path(f"static/reports/{report_filename}")
     report_path.parent.mkdir(parents=True, exist_ok=True)
 
     c = canvas.Canvas(str(report_path), pagesize=letter)
@@ -71,7 +74,25 @@ def generate_pdf_report(record_id, data_dict):
         y -= 20
 
     c.save()
-    return report_filename, str(report_path)
+    return str(report_path)
+
+# Generate CSV report and public link
+
+def generate_csv_report(record_id, data_dict):
+    report_id = str(uuid.uuid4())[:8]
+    csv_filename = f"report_{report_id}.csv"
+    csv_path = Path(f"static/reports/{csv_filename}")
+    csv_path.parent.mkdir(parents=True, exist_ok=True)
+
+    with open(csv_path, mode='w', newline='') as f:
+        writer = csv.writer(f)
+        writer.writerow(["Field", "Value"])
+        writer.writerow(["Salesforce Record ID", record_id])
+        for k, v in data_dict.items():
+            writer.writerow([k, v])
+
+    public_csv_url = f"{HUGGINGFACE_SPACE_URL}/file/static/reports/{csv_filename}"
+    return public_csv_url, str(csv_path)
 
 def call_ai_model(usage_hours, idle_hours, movement_frequency, cost_per_hour, last_maintenance_str):
     try:
@@ -114,37 +135,35 @@ def process_equipment_utilization(equipment_name, project_name, usage_hours, idl
         suggestion_confidence = 0.9
         utilization_score = 0.85
 
-    record_data = {
-        "Equipment_Name__c": equipment_name,
-        "Project_Name__c": project_name,
-        "Usage_Hours__c": usage_hours,
-        "Idle_Hours__c": idle_hours,
-        "AI_Suggestion__c": ai_suggestion,
-        "Suggestion_Confidence__c": suggestion_confidence * 100,
-        "Utilization_Score__c": utilization_score * 100,
-        "Report_Link__c": "Pending",
-        "Last_Maintenance__c": last_maintenance_str,
-        "Dashboard_Flag__c": False
+    summary_data = {
+        "Equipment Name": equipment_name,
+        "Project": project_name,
+        "Usage Hours": usage_hours,
+        "Idle Hours": idle_hours,
+        "Suggestion": ai_suggestion,
+        "Confidence": suggestion_confidence,
+        "Utilization Score": utilization_score
     }
 
     try:
-        response = sf.Equipment_Utilization_Record__c.create(record_data)
-        record_id = response.get("id")
-        logger.info(f"Salesforce record created: {record_id}")
-
-        summary_data = {
-            "Equipment Name": equipment_name,
-            "Project": project_name,
-            "Usage Hours": usage_hours,
-            "Idle Hours": idle_hours,
-            "Suggestion": ai_suggestion,
-            "Confidence": suggestion_confidence,
-            "Utilization Score": utilization_score
-        }
+        response = sf.Equipment_Utilization_Record__c.create({
+            "Equipment_Name__c": equipment_name,
+            "Project_Name__c": project_name,
+            "Usage_Hours__c": usage_hours,
+            "Idle_Hours__c": idle_hours,
+            "AI_Suggestion__c": ai_suggestion,
+            "Suggestion_Confidence__c": suggestion_confidence * 100,
+            "Utilization_Score__c": utilization_score * 100,
+            "Report_Link__c": "Pending",
+            "Last_Maintenance__c": last_maintenance_str,
+            "Dashboard_Flag__c": False
+        })
 
-        report_filename, file_path = generate_pdf_report(record_id, summary_data)
+        record_id = response.get("id")
+        csv_url, csv_path = generate_csv_report(record_id, summary_data)
+        generate_pdf_report(record_id, summary_data)
 
-        sf.Equipment_Utilization_Record__c.update(record_id, {"Report_Link__c": report_filename})
+        sf.Equipment_Utilization_Record__c.update(record_id, {"Report_Link__c": csv_url})
 
         return {
             "Salesforce_Record_Id": record_id,
@@ -152,8 +171,8 @@ def process_equipment_utilization(equipment_name, project_name, usage_hours, idl
             "AI_Suggestion": ai_suggestion,
             "Suggestion_Confidence": suggestion_confidence,
             "Utilization_Score": utilization_score,
-            "Report_Link": report_filename,
-            "report_file_path": file_path
+            "Report_Link": csv_url,
+            "report_file_path": csv_path
         }
     except Exception as e:
         logger.error(f"Error creating or updating Salesforce record: {e}")
@@ -217,7 +236,7 @@ with gr.Blocks() as app:
         with gr.Row():
             submit_button = gr.Button("🚀 Submit", variant="primary")
             output = gr.JSON(label="📄 Salesforce Record Creation Result")
-            report_file_output = gr.File(label="📃 Download PDF Report")
+            report_file_output = gr.File(label="📃 Download CSV Report")
 
         submit_button.click(
             fn=gradio_upload_process,