Update app.py

app.py

@@ -6,13 +6,13 @@ import bleach
 from simple_salesforce import Salesforce, SalesforceError
 import os
 
-# Salesforce configuration (loaded from environment variables for potential future use)
+# Salesforce configuration (loaded from environment variables for security)
 SF_USERNAME = os.getenv("SF_USERNAME")
 SF_PASSWORD = os.getenv("SF_PASSWORD")
 SF_SECURITY_TOKEN = os.getenv("SF_SECURITY_TOKEN")
 SF_INSTANCE_URL = os.getenv("SF_INSTANCE_URL")
 
-# Initialize Salesforce connection (for potential future use)
+# Initialize Salesforce connection
 def init_salesforce():
     try:
         sf = Salesforce(
@@ -25,7 +25,7 @@ def init_salesforce():
     except SalesforceError as e:
         return None, f"Salesforce connection failed: {str(e)}"
 
-# Create Salesforce record (for potential future use)
+# Create Salesforce record using Case object
 def create_salesforce_record(score, checklist_summary, missing_summary, status):
     sf, connection_message = init_salesforce()
     if not sf:
@@ -37,12 +37,13 @@ def create_salesforce_record(score, checklist_summary, missing_summary, status):
         sf_status = "Closed"
 
     try:
+        # Create a Case record in Salesforce with required fields
         record = {
             "Subject": "Project Readiness Evaluation",
             "Description": f"Readiness Score: {score}%\nChecklist Summary:\n{checklist_summary}\nMissing Items: {missing_summary}",
             "Status": sf_status,
-            "Origin": "Web",
-            "Priority": "Medium"
+            "Origin": "Web",  # Required field in many Salesforce orgs
+            "Priority": "Medium"  # Required field in many Salesforce orgs
         }
         result = sf.Case.create(record)
         return f"Case record created successfully in Salesforce. Case ID: {result['id']}"
@@ -133,7 +134,7 @@ with gr.Blocks(css="""progress { background-color: #f0f0f0; } .red { background-
     gr.Markdown(
         """
         # Project Closure Readiness Evaluator
-        Evaluate project readiness and generate a PDF report.
+        Evaluate project readiness, save to Salesforce, and generate a PDF report.
         """
     )
     with gr.Row():
@@ -148,6 +149,8 @@ with gr.Blocks(css="""progress { background-color: #f0f0f0; } .red { background-
             status_output = gr.Textbox(label="Overall Status")
             checklist_output = gr.Textbox(label="Checklist Summary")
             missing_output = gr.Textbox(label="Missing Items")
+            sf_create_btn = gr.Button("Save to Salesforce")
+            sf_output = gr.Textbox(label="Salesforce Status")
             pdf_btn = gr.Button("Download PDF Report")
             pdf_output = gr.File(label="PDF Report")
 
@@ -158,6 +161,13 @@ with gr.Blocks(css="""progress { background-color: #f0f0f0; } .red { background-
         outputs=[score_output, checklist_output, missing_output, status_output, progress_output]
     )
 
+    # Connect Salesforce create button to create_salesforce_record
+    sf_create_btn.click(
+        fn=create_salesforce_record,
+        inputs=[score_output, checklist_output, missing_output, status_output],
+        outputs=sf_output
+    )
+
     # Connect PDF button to generate_pdf
     pdf_btn.click(
         fn=generate_pdf,