Update app.py

app.py

@@ -114,7 +114,7 @@ def summarize_text(text, max_length=100, min_length=30):
         return text  # Fallback to original text
 
 # Create Salesforce record in custom object Project_Closure_Handover__c
-def create_salesforce_record(score, checklist_summary, missing_summary, status, escalated, logs, qa_report, punch_list_text, open_punch_items, pdf_path=None):
+def create_salesforce_record(score, checklist_summary, missing_summary, status, escalated, logs, qa_report, punch_list_text, open_punch_items, pdf_path=None, project_id=None):
     if not SALESFORCE_AVAILABLE:
         logging.error("Salesforce library not available. Skipping record creation.")
         return "Salesforce library not available"
@@ -139,6 +139,7 @@ def create_salesforce_record(score, checklist_summary, missing_summary, status,
         logs = str(logs) if logs else ""
         qa_report = str(qa_report) if qa_report else ""
         punch_list_text = str(punch_list_text) if punch_list_text else ""
+        project_id = str(project_id) if project_id else ""  # Ensure project_id is a string
         missing_documents = len(missing_summary.split(", ")) if missing_summary and missing_summary != "None" else 0
         open_punch_items = int(open_punch_items) if open_punch_items is not None else 0
         evaluated_at = datetime.now().strftime("%Y-%m-%dT%H:%M:%SZ")  # Salesforce Date/Time format
@@ -161,7 +162,8 @@ def create_salesforce_record(score, checklist_summary, missing_summary, status,
             "Logs__c": logs,
             "Open_Punch_Items__c": open_punch_items,
             "Punch_List__c": punch_list_text,
-            "QA_Report__c": qa_report
+            "QA_Report__c": qa_report,
+            "Project_ID__c": project_id  # Added Project ID field
         }
 
         logging.debug(f"Attempting to create Salesforce record in Project_Closure_Handover__c with data: {record}")
@@ -210,7 +212,7 @@ def create_salesforce_record(score, checklist_summary, missing_summary, status,
         logging.error("Check the following in your Salesforce org:")
         logging.error("- Ensure the user has Create and Edit permission on Project_Closure_Handover__c.")
         logging.error("- Ensure the user has permission to create and read Attachments.")
-        logging.error("- Verify field-level security for all fields in the record.")
+        logging.error("- Verify field-level security for all fields in the record, including Project_ID__c.")
         logging.error("- Check for validation rules or required fields that might be failing.")
         return f"Salesforce error: {str(e)}"
     except Exception as e:
@@ -390,6 +392,7 @@ with gr.Blocks(css="""
     )
     with gr.Row():
         with gr.Column(scale=2):
+            project_id_input = gr.Textbox(label="Project ID", placeholder="Enter Project ID (e.g., PROJ-1234)")
             logs_input = gr.Textbox(label="Project Logs", lines=5, placeholder="Enter project logs (e.g., 'Project complete, handover done')")
             qa_input = gr.Dropdown(
                 label="QA Report",
@@ -433,7 +436,7 @@ with gr.Blocks(css="""
         fn=create_salesforce_record,
         inputs=[
             score_output, checklist_output, missing_output, status_output,
-            gr.State(), gr.State(), gr.State(), gr.State(), open_punch_items_output, pdf_output
+            gr.State(), gr.State(), gr.State(), gr.State(), open_punch_items_output, pdf_output, project_id_input
         ],
         outputs=None
     )