Update app.py

app.py

@@ -1,14 +1,15 @@
 import streamlit as st
-from salesforce_integration import sf  # Import Salesforce integration
+from salesforce_integration import get_quote_data  # Import Salesforce integration
 from generate_pdf import create_pdf  # Import PDF generation function
 from send_email import send_email  # Import email sending function
 from send_whatsapp import send_whatsapp  # Import WhatsApp sending function
 
+# Title of the app
 st.title('Generate and Send Quote Invoice')
 
 if st.button('Generate PDF and Send'):
     # Fetch the quote data from Salesforce
-    quote_data = sf.query("SELECT Customer_Name__c, Quote_Date__c, Origination_Location__c, Total_Amount__c FROM Quote__c WHERE Id = 'YOUR_QUOTE_ID'")
+    quote_data = get_quote_data()
     
     # Generate the PDF
     pdf_file = create_pdf(quote_data)