Update app.py

app.py

@@ -5,6 +5,8 @@ import uuid
 from pathlib import Path
 import requests
 import gradio as gr
+from reportlab.lib.pagesizes import letter
+from reportlab.pdfgen import canvas
 from simple_salesforce import Salesforce
 
 # Configure logging
@@ -20,7 +22,6 @@ SALESFORCE_USERNAME = "vaneshdevarapalli866@agentforce.com"
 SALESFORCE_PASSWORD = "vanesh@331"
 SALESFORCE_SECURITY_TOKEN = "VRUVbBOdG0s9Q4xy0W6DB1Y6b"
 
-# Hugging Face API Endpoint (replace with your actual endpoint)
 HUGGINGFACE_API_URL = "https://your-huggingface-model-endpoint"
 
 # Connect to Salesforce
@@ -53,20 +54,25 @@ project_choices = [
 
 ai_suggestion_choices = ["Move", "Pause Rent", "Repair", "Replace"]
 
-# Generate report link (simulate report as HTML file)
-def generate_report_link(record_id, data_dict):
+# Generate PDF report
+def generate_pdf_report(record_id, data_dict):
     report_id = str(uuid.uuid4())[:8]
-    report_filename = f"report_{report_id}.html"
+    report_filename = f"report_{report_id}.pdf"
     report_path = Path(f"reports/{report_filename}")
     report_path.parent.mkdir(exist_ok=True)
 
-    with open(report_path, "w") as f:
-        f.write(f"<h1>Equipment Utilization Report</h1>")
-        f.write(f"<p>Salesforce Record ID: {record_id}</p>")
-        for key, val in data_dict.items():
-            f.write(f"<p><strong>{key}</strong>: {val}</p>")
+    c = canvas.Canvas(str(report_path), pagesize=letter)
+    c.setFont("Helvetica", 12)
+    c.drawString(100, 750, f"Equipment Utilization Report")
+    c.drawString(100, 735, f"Salesforce Record ID: {record_id}")
 
-    return f"https://yourdomain.com/reports/{report_filename}"
+    y = 710
+    for key, val in data_dict.items():
+        c.drawString(100, y, f"{key}: {val}")
+        y -= 20
+
+    c.save()
+    return str(report_path)
 
 # Call Hugging Face AI model
 def call_ai_model(usage_hours, idle_hours, movement_frequency, cost_per_hour, last_maintenance_str):
@@ -99,7 +105,6 @@ def call_ai_model(usage_hours, idle_hours, movement_frequency, cost_per_hour, la
         return "No Action", 0.0, 0.0
 
 # Main process
-
 def process_equipment_utilization(equipment_name, project_name, usage_hours, idle_hours,
                                   movement_frequency, cost_per_hour, report_link, last_maintenance, ai_suggestion):
     last_maintenance_str = last_maintenance.strftime('%Y-%m-%d') if last_maintenance else None
@@ -120,7 +125,7 @@ def process_equipment_utilization(equipment_name, project_name, usage_hours, idl
         "AI_Suggestion__c": ai_suggestion,
         "Suggestion_Confidence__c": suggestion_confidence * 100,
         "Utilization_Score__c": utilization_score * 100,
-        "Report_Link__c": report_link.strip() if report_link.strip() else "N/A",
+        "Report_Link__c": "Pending",
         "Last_Maintenance__c": last_maintenance_str,
         "Dashboard_Flag__c": False
     }
@@ -130,7 +135,6 @@ def process_equipment_utilization(equipment_name, project_name, usage_hours, idl
         record_id = response.get("id")
         logger.info(f"Salesforce record created: {record_id}")
 
-        # Now generate report and update the record with the actual link
         summary_data = {
             "Equipment Name": equipment_name,
             "Project": project_name,
@@ -140,9 +144,9 @@ def process_equipment_utilization(equipment_name, project_name, usage_hours, idl
             "Confidence": suggestion_confidence,
             "Utilization Score": utilization_score
         }
-        generated_report_link = generate_report_link(record_id, summary_data)
+        report_file_path = generate_pdf_report(record_id, summary_data)
 
-        sf.Equipment_Utilization_Record__c.update(record_id, {"Report_Link__c": generated_report_link})
+        sf.Equipment_Utilization_Record__c.update(record_id, {"Report_Link__c": report_file_path})
 
         return {
             "Salesforce_Record_Id": record_id,
@@ -150,7 +154,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": generated_report_link
+            "Report_Link": report_file_path,
+            "report_file_path": report_file_path
         }
     except Exception as e:
         logger.error(f"Error creating or updating Salesforce record: {e}")
@@ -166,14 +171,14 @@ def gradio_upload_process(equipment_name, project_name, usage_hours, idle_hours,
         movement_frequency = float(movement_frequency)
         cost_per_hour = float(cost_per_hour)
     except Exception as e:
-        return {"error": f"Invalid numeric input: {str(e)}"}
+        return {"error": f"Invalid numeric input: {str(e)}"}, None
 
     try:
         last_maintenance_dt = datetime.datetime.strptime(last_maintenance, "%Y-%m-%d") if last_maintenance else None
     except Exception as e:
-        return {"error": f"Invalid date format for Last Maintenance (expected YYYY-MM-DD): {str(e)}"}
+        return {"error": f"Invalid date format for Last Maintenance (expected YYYY-MM-DD): {str(e)}"}, None
 
-    return process_equipment_utilization(
+    result = process_equipment_utilization(
         equipment_name,
         project_name,
         usage_hours,
@@ -184,6 +189,7 @@ def gradio_upload_process(equipment_name, project_name, usage_hours, idle_hours,
         last_maintenance_dt,
         ai_suggestion
     )
+    return result, result.get("report_file_path")
 
 # UI
 with gr.Blocks() as app:
@@ -215,6 +221,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")
 
         submit_button.click(
             fn=gradio_upload_process,
@@ -225,7 +232,7 @@ with gr.Blocks() as app:
                 report_link, last_maintenance,
                 ai_suggestion_dropdown
             ],
-            outputs=output
+            outputs=[output, report_file_output]
         )
 
     app.css = """