Update app.py

app.py

@@ -22,7 +22,6 @@ SALESFORCE_USERNAME = "vaneshdevarapalli866@agentforce.com"
 SALESFORCE_PASSWORD = "vanesh@331"
 SALESFORCE_SECURITY_TOKEN = "VRUVbBOdG0s9Q4xy0W6DB1Y6b"
 
-HUGGINGFACE_SPACE_URL = "https://huggingface.co/spaces/YOUR_USERNAME/YOUR_SPACE_NAME"
 HUGGINGFACE_API_URL = "https://your-huggingface-model-endpoint"
 
 # Connect to Salesforce
@@ -54,11 +53,11 @@ project_choices = [
 
 ai_suggestion_choices = ["Move", "Pause Rent", "Repair", "Replace"]
 
-# Generate PDF and return public Hugging Face URL
+# Generate PDF and return filename and path
 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 = Path(f"reports/{report_filename}")
     report_path.parent.mkdir(parents=True, exist_ok=True)
 
     c = canvas.Canvas(str(report_path), pagesize=letter)
@@ -72,7 +71,7 @@ def generate_pdf_report(record_id, data_dict):
         y -= 20
 
     c.save()
-    return f"{HUGGINGFACE_SPACE_URL}/file/static/reports/{report_filename}", str(report_path)
+    return report_filename, str(report_path)
 
 def call_ai_model(usage_hours, idle_hours, movement_frequency, cost_per_hour, last_maintenance_str):
     try:
@@ -143,9 +142,9 @@ def process_equipment_utilization(equipment_name, project_name, usage_hours, idl
             "Utilization Score": utilization_score
         }
 
-        public_url, file_path = generate_pdf_report(record_id, summary_data)
+        report_filename, file_path = generate_pdf_report(record_id, summary_data)
 
-        sf.Equipment_Utilization_Record__c.update(record_id, {"Report_Link__c": public_url})
+        sf.Equipment_Utilization_Record__c.update(record_id, {"Report_Link__c": report_filename})
 
         return {
             "Salesforce_Record_Id": record_id,
@@ -153,7 +152,7 @@ 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": public_url,
+            "Report_Link": report_filename,
             "report_file_path": file_path
         }
     except Exception as e:
@@ -262,4 +261,4 @@ with gr.Blocks() as app:
     """
 
 if __name__ == "__main__":
-    app.launch()
+    app.launch()