Update app.py

app.py

@@ -2,6 +2,7 @@ import datetime
 import logging
 import sys
 import uuid
+import os
 from pathlib import Path
 import csv
 import gradio as gr
@@ -16,12 +17,14 @@ logging.basicConfig(
     handlers=[logging.FileHandler('app_log.txt'), logging.StreamHandler(sys.stdout)]
 )
 logger = logging.getLogger(__name__)
-
 # Salesforce credentials
 SALESFORCE_USERNAME = "vaneshdevarapalli866@agentforce.com"
 SALESFORCE_PASSWORD = "vanesh@331"
 SALESFORCE_SECURITY_TOKEN = "VRUVbBOdG0s9Q4xy0W6DB1Y6b"
 
+# Base URL for your application (replace with your actual domain)
+BASE_URL = "https://huggingface.co/spaces/VaneshDev/EquipmentDashboard"  # Change this to your actual domain or use ngrok for testing
+
 # Connect to Salesforce
 def connect_to_salesforce():
     try:
@@ -51,13 +54,15 @@ project_choices = [
 
 ai_suggestion_choices = ["Move", "Pause Rent", "Repair", "Replace"]
 
-# Generate PDF report
+# Generate PDF report with public URL
 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"static/reports/{report_filename}")
-    report_path.parent.mkdir(parents=True, exist_ok=True)
+    report_dir = Path("static/reports/")
+    report_dir.mkdir(parents=True, exist_ok=True)
+    report_path = report_dir / report_filename
 
+    # Create PDF content
     c = canvas.Canvas(str(report_path), pagesize=letter)
     c.setFont("Helvetica", 12)
     c.drawString(100, 750, f"Equipment Utilization Report")
@@ -69,14 +74,18 @@ def generate_pdf_report(record_id, data_dict):
         y -= 20
 
     c.save()
-    return str(report_path)
+    
+    # Generate public URL
+    public_url = f"{BASE_URL}/static/reports/{report_filename}"
+    return str(report_path), public_url
 
-# Generate CSV report (without public URL)
+# Generate CSV report with public URL
 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)
+    csv_dir = Path("static/reports/")
+    csv_dir.mkdir(parents=True, exist_ok=True)
+    csv_path = csv_dir / csv_filename
 
     with open(csv_path, mode='w', newline='') as f:
         writer = csv.writer(f)
@@ -85,15 +94,15 @@ def generate_csv_report(record_id, data_dict):
         for k, v in data_dict.items():
             writer.writerow([k, v])
 
-    return str(csv_path)
+    public_url = f"{BASE_URL}/static/reports/{csv_filename}"
+    return str(csv_path), public_url
 
 def call_ai_model(usage_hours, idle_hours, movement_frequency, cost_per_hour, last_maintenance_str):
-    # Placeholder response since API endpoint is removed
     logger.info("AI model endpoint removed; returning placeholder response.")
     return "No Action", 0.0, 0.0
 
 def process_equipment_utilization(equipment_name, project_name, usage_hours, idle_hours,
-                                  movement_frequency, cost_per_hour, last_maintenance, ai_suggestion):
+                                movement_frequency, cost_per_hour, last_maintenance, ai_suggestion):
     last_maintenance_str = last_maintenance.strftime('%Y-%m-%d') if last_maintenance else None
 
     if not ai_suggestion:
@@ -112,11 +121,10 @@ def process_equipment_utilization(equipment_name, project_name, usage_hours, idl
         "Suggestion": ai_suggestion,
         "Confidence": suggestion_confidence,
         "Utilization Score": utilization_score,
-        "Cost per Hour": cost_per_hour  # Added to summary_data for the report
+        "Cost per Hour": cost_per_hour
     }
 
     try:
-        # Log the data being sent to Salesforce
         record_data = {
             "Equipment_Name__c": equipment_name,
             "Project_Name__c": project_name,
@@ -133,17 +141,16 @@ def process_equipment_utilization(equipment_name, project_name, usage_hours, idl
         logger.info(f"Creating Salesforce record with data: {record_data}")
 
         response = sf.Equipment_Utilization_Record__c.create(record_data)
-
         record_id = response.get("id")
         logger.info(f"Successfully created Salesforce record with ID: {record_id}")
 
-        # Generate both CSV and PDF reports
-        csv_path = generate_csv_report(record_id, summary_data)
-        pdf_path = generate_pdf_report(record_id, summary_data)
+        # Generate reports and get public URLs
+        csv_path, csv_url = generate_csv_report(record_id, summary_data)
+        pdf_path, pdf_url = generate_pdf_report(record_id, summary_data)
 
-        # Update the Salesforce record with the local PDF path
-        sf.Equipment_Utilization_Record__c.update(record_id, {"Report_Link__c": pdf_path})
-        logger.info(f"Updated Salesforce record {record_id} with Report_Link__c: {pdf_path}")
+        # Update the Salesforce record with the public PDF URL
+        sf.Equipment_Utilization_Record__c.update(record_id, {"Report_Link__c": pdf_url})
+        logger.info(f"Updated Salesforce record {record_id} with Report_Link__c: {pdf_url}")
 
         return {
             "Salesforce_Record_Id": record_id,
@@ -151,15 +158,16 @@ 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": pdf_path,
-            "report_file_path": pdf_path
+            "Report_Link": pdf_url,
+            "report_file_path": pdf_path,
+            "public_report_url": pdf_url
         }
     except Exception as e:
         logger.error(f"Error creating or updating Salesforce record: {e}")
         return {"error": str(e)}
 
 def gradio_upload_process(equipment_name, project_name, usage_hours, idle_hours,
-                         movement_frequency, cost_per_hour, last_maintenance, ai_suggestion):
+                        movement_frequency, cost_per_hour, last_maintenance, ai_suggestion):
     try:
         usage_hours = float(usage_hours)
         idle_hours = float(idle_hours)
@@ -185,6 +193,12 @@ def gradio_upload_process(equipment_name, project_name, usage_hours, idle_hours,
     )
     return result, result.get("report_file_path")
 
+# Serve static files through Gradio
+def serve_static_files():
+    static_dir = Path("static")
+    static_dir.mkdir(exist_ok=True)
+    return static_dir
+
 with gr.Blocks() as app:
     gr.Markdown("## 📋 Equipment Utilization Record Uploader")
     gr.Markdown("Fill in the details below to generate AI suggestions and save them to Salesforce.")
@@ -214,6 +228,7 @@ with gr.Blocks() as app:
             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_url_output = gr.Textbox(label="🌐 Public Report URL", interactive=False)
 
         submit_button.click(
             fn=gradio_upload_process,
@@ -223,7 +238,7 @@ with gr.Blocks() as app:
                 movement_frequency, cost_per_hour,
                 last_maintenance, ai_suggestion_dropdown
             ],
-            outputs=[output, report_file_output]
+            outputs=[output, report_file_output, report_url_output]
         )
 
     app.css = """
@@ -256,4 +271,6 @@ with gr.Blocks() as app:
     """
 
 if __name__ == "__main__":
-    app.launch()
+    # Ensure static directory exists
+    serve_static_files()
+    app.launch(server_name="0.0.0.0", server_port=7860)