Update app.py

app.py

@@ -8,6 +8,7 @@ import os
 from huggingface_hub import InferenceClient
 from retry import retry
 import time
+from datetime import datetime
 
 # Print statement to confirm script initialization
 print("Starting Project Closure Readiness Evaluator app...")
@@ -94,7 +95,10 @@ 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):
+def create_salesforce_record(
+    score, checklist_summary, missing_summary, status, escalated, logs, qa_report, punch_list_text,
+    open_punch_items, project_id, start_date, end_date, owner_id, pdf_path=None
+):
     if not SALESFORCE_AVAILABLE:
         logging.error("Salesforce library not available. Skipping record creation.")
         return "Salesforce library not available"
@@ -109,22 +113,51 @@ def create_salesforce_record(score, checklist_summary, missing_summary, status):
         summarized_checklist = summarize_text(checklist_summary)
         summarized_missing = summarize_text(missing_summary)
 
-        # Ensure inputs are strings and sanitized
+        # Ensure inputs are properly formatted
         score = float(score) if score is not None else 0.0
         checklist_summary = str(checklist_summary) if checklist_summary else ""
         summarized_checklist = str(summarized_checklist) if summarized_checklist else ""
         missing_summary = str(missing_summary) if missing_summary else ""
         summarized_missing = str(summarized_missing) if summarized_missing else ""
         status = str(status) if status else ""
+        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 ""
+        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
+        project_id = str(project_id) if project_id else None
+        owner_id = str(owner_id) if owner_id else None
+        start_date = str(start_date) if start_date else None
+        end_date = str(end_date) if end_date else None
+        evaluated_at = datetime.now().strftime("%Y-%m-%dT%H:%M:%SZ")  # Salesforce Date/Time format
+        alert_sent = escalated  # Set Alert_Sent__c based on escalation flag
 
         # Create a record in the custom object Project_Closure_Handover__c
         record = {
             "Readiness_Score__c": score,
             "Checklist_Summary__c": checklist_summary,
-            "Missing_Documents__c": missing_summary,
+            "Missing_Documents__c": missing_documents,
             "Status__c": status,
-            "Summarized_Missing_Items__c": summarized_missing
+            "Summarized_Missing_Items__c": summarized_missing,
+            "Alert_Sent__c": alert_sent,
+            "Client_PDF_Pack_URL__c": pdf_path if pdf_path else "",
+            "Closure_Pack_URL__c": "",  # Placeholder; update if you have a closure pack URL
+            "End_Date__c": end_date,
+            "Escalation_Flag__c": escalated,
+            "Evaluated_At__c": evaluated_at,
+            "Logs__c": logs,
+            "Open_Punch_Items__c": open_punch_items,
+            "Punch_List__c": punch_list_text,
+            "QA_Report__c": qa_report,
+            "Start_Date__c": start_date
         }
+        
+        # Add optional fields if provided
+        if project_id:
+            record["Project_ID__c"] = project_id
+        if owner_id:
+            record["OwnerId"] = owner_id
+
         logging.debug(f"Attempting to create Salesforce record: {record}")
         result = sf.Project_Closure_Handover__c.create(record)
         logging.info(f"Successfully created Salesforce record: {result}")
@@ -143,10 +176,10 @@ def sanitize_input(text):
     return bleach.clean(text.strip())
 
 # Rule-based completeness engine with weighted scoring
-def evaluate_readiness(logs, qa_report, punch_list_text):
+def evaluate_readiness(logs, qa_report, punch_list_text, project_id, start_date, end_date, owner_id):
     try:
         # Log inputs for debugging
-        logging.info(f"Inputs - Logs: {logs}, QA Report: {qa_report}, Punch List: {punch_list_text}")
+        logging.info(f"Inputs - Logs: {logs}, QA Report: {qa_report}, Punch List: {punch_list_text}, Project ID: {project_id}, Start Date: {start_date}, End Date: {end_date}, Owner ID: {owner_id}")
 
         # Initialize score and lists for tracking
         score = 0
@@ -162,6 +195,10 @@ def evaluate_readiness(logs, qa_report, punch_list_text):
         logs = sanitize_input(logs)
         qa_report = sanitize_input(qa_report)
         punch_list_text = sanitize_input(punch_list_text)
+        project_id = sanitize_input(project_id)
+        start_date = sanitize_input(start_date)
+        end_date = sanitize_input(end_date)
+        owner_id = sanitize_input(owner_id)
 
         # Process Project Logs (30% weight)
         log_keywords = r"complete|handover done|finished|closed|successful"
@@ -196,6 +233,9 @@ def evaluate_readiness(logs, qa_report, punch_list_text):
             checklist_details.append("Punch List: Pending")
         logging.info(f"Punch List Check: {'Pass' if punch_pass else 'Fail'}, Final Score: {score}%")
 
+        # Calculate open punch items
+        open_punch_items = 0 if punch_pass else len(punch_list_text.split(",")) if punch_list_text else 1
+
         # Handle critical issues (escalation)
         escalated = any("incomplete" in item.lower() or "missing" in item.lower() for item in missing_items)
         status = "Escalated" if escalated else ("Completed" if not missing_items else "Pending")
@@ -215,17 +255,14 @@ def evaluate_readiness(logs, qa_report, punch_list_text):
         </div>
         """
 
-        # Automatically create Salesforce record
-        sf_result = create_salesforce_record(score, checklist_summary, missing_summary, status)
-        logging.info(f"Salesforce record creation result: {sf_result}")
-
-        return score, checklist_summary, missing_summary, status, progress_bar
+        # Store inputs for PDF generation and Salesforce record
+        return (score, checklist_summary, missing_summary, status, progress_bar, escalated, logs, qa_report, punch_list_text, open_punch_items, project_id, start_date, end_date, owner_id)
     except Exception as e:
         logging.error(f"Error in evaluate_readiness: {str(e)}")
         raise
 
 # Generate PDF report with signature slots and write to a temporary file
-def generate_pdf(score, checklist_summary, missing_summary, status):
+def generate_pdf(score, checklist_summary, missing_summary, status, logs, qa_report, punch_list_text, project_id, start_date, end_date, owner_id):
     try:
         # Sanitize inputs to ensure they are strings and handle None
         logging.info("Sanitizing inputs for PDF generation")
@@ -233,11 +270,25 @@ def generate_pdf(score, checklist_summary, missing_summary, status):
         checklist_summary = str(checklist_summary) if checklist_summary is not None else ""
         missing_summary = str(missing_summary) if missing_summary is not None else ""
         status = str(status) if status is not None else ""
+        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 "Not Provided"
+        start_date = str(start_date) if start_date else "Not Provided"
+        end_date = str(end_date) if end_date else "Not Provided"
+        owner_id = str(owner_id) if owner_id else "Not Provided"
 
         # Remove non-ASCII characters to prevent rendering issues
         checklist_summary = checklist_summary.encode('ascii', 'ignore').decode('ascii')
         missing_summary = missing_summary.encode('ascii', 'ignore').decode('ascii')
         status = status.encode('ascii', 'ignore').decode('ascii')
+        logs = logs.encode('ascii', 'ignore').decode('ascii')
+        qa_report = qa_report.encode('ascii', 'ignore').decode('ascii')
+        punch_list_text = punch_list_text.encode('ascii', 'ignore').decode('ascii')
+        project_id = project_id.encode('ascii', 'ignore').decode('ascii')
+        start_date = start_date.encode('ascii', 'ignore').decode('ascii')
+        end_date = end_date.encode('ascii', 'ignore').decode('ascii')
+        owner_id = owner_id.encode('ascii', 'ignore').decode('ascii')
 
         # Define the temporary file path
         pdf_path = "temp_readiness_report.pdf"
@@ -254,13 +305,17 @@ def generate_pdf(score, checklist_summary, missing_summary, status):
         # Draw basic fields using Times-Roman
         logging.info("Drawing basic fields")
         c.setFont("Times-Roman", 12)
-        c.drawString(50, 720, f"Readiness Score: {score}%")
-        c.drawString(50, 700, f"Status: {status}")
-        c.drawString(50, 680, "Checklist Summary:")
+        c.drawString(50, 720, f"Project ID: {project_id}")
+        c.drawString(50, 700, f"Readiness Score: {score}%")
+        c.drawString(50, 680, f"Status: {status}")
+        c.drawString(50, 660, f"Start Date: {start_date}")
+        c.drawString(50, 640, f"End Date: {end_date}")
+        c.drawString(50, 620, f"Owner ID: {owner_id}")
+        c.drawString(50, 600, "Checklist Summary:")
 
         # Draw checklist summary with Y-position overflow handling
         logging.info("Drawing checklist summary")
-        y = 660
+        y = 580
         BOTTOM_MARGIN = 50  # Minimum Y position before creating a new page
         for line in checklist_summary.split("\n"):
             if y < BOTTOM_MARGIN:
@@ -279,7 +334,7 @@ def generate_pdf(score, checklist_summary, missing_summary, status):
             c.setFont("Times-Roman", 12)
             y = 750
         c.drawString(50, y - 20, "Missing Items:")
-
+        
         if y - 40 < BOTTOM_MARGIN:
             logging.info("Y position below margin, creating new page")
             c.showPage()
@@ -287,19 +342,61 @@ def generate_pdf(score, checklist_summary, missing_summary, status):
             y = 750
         c.drawString(50, y - 40, missing_summary)
 
+        if y - 60 < BOTTOM_MARGIN:
+            logging.info("Y position below margin, creating new page")
+            c.showPage()
+            c.setFont("Times-Roman", 12)
+            y = 750
+        c.drawString(50, y - 60, "Logs:")
+        
         if y - 80 < BOTTOM_MARGIN:
             logging.info("Y position below margin, creating new page")
             c.showPage()
             c.setFont("Times-Roman", 12)
             y = 750
-        c.drawString(50, y - 80, "Stakeholder Signature: ____________________")
+        c.drawString(50, y - 80, logs[:100] + "..." if len(logs) > 100 else logs)
 
         if y - 100 < BOTTOM_MARGIN:
             logging.info("Y position below margin, creating new page")
             c.showPage()
             c.setFont("Times-Roman", 12)
             y = 750
-        c.drawString(50, y - 100, "Date: ____________________")
+        c.drawString(50, y - 100, "QA Report:")
+        
+        if y - 120 < BOTTOM_MARGIN:
+            logging.info("Y position below margin, creating new page")
+            c.showPage()
+            c.setFont("Times-Roman", 12)
+            y = 750
+        c.drawString(50, y - 120, qa_report[:100] + "..." if len(qa_report) > 100 else qa_report)
+
+        if y - 140 < BOTTOM_MARGIN:
+            logging.info("Y position below margin, creating new page")
+            c.showPage()
+            c.setFont("Times-Roman", 12)
+            y = 750
+        c.drawString(50, y - 140, "Punch List:")
+        
+        if y - 160 < BOTTOM_MARGIN:
+            logging.info("Y position below margin, creating new page")
+            c.showPage()
+            c.setFont("Times-Roman", 12)
+            y = 750
+        c.drawString(50, y - 160, punch_list_text[:100] + "..." if len(punch_list_text) > 100 else punch_list_text)
+
+        if y - 180 < BOTTOM_MARGIN:
+            logging.info("Y position below margin, creating new page")
+            c.showPage()
+            c.setFont("Times-Roman", 12)
+            y = 750
+        c.drawString(50, y - 180, "Stakeholder Signature: ____________________")
+
+        if y - 200 < BOTTOM_MARGIN:
+            logging.info("Y position below margin, creating new page")
+            c.showPage()
+            c.setFont("Times-Roman", 12)
+            y = 750
+        c.drawString(50, y - 200, "Date: ____________________")
 
         # Finalize the PDF
         logging.info("Finalizing PDF")
@@ -311,7 +408,7 @@ def generate_pdf(score, checklist_summary, missing_summary, status):
             raise FileNotFoundError(f"PDF file {pdf_path} was not created")
 
         logging.info(f"PDF generated successfully at {pdf_path}")
-        return pdf_path, "PDF generation triggered successfully"  # Return the file path and debug message
+        return pdf_path, "PDF generation triggered successfully"
     except Exception as e:
         logging.error(f"Error in generate_pdf: {str(e)}")
         raise
@@ -336,6 +433,10 @@ with gr.Blocks(css="""
             logs_input = gr.Textbox(label="Project Logs", lines=5, placeholder="Enter project logs (e.g., 'Project complete, handover done')")
             qa_input = gr.Textbox(label="QA Report", placeholder="Enter QA status (e.g., 'Approved')")
             punch_input = gr.Textbox(label="Punch List (Plain Text)", placeholder="Enter punch list status (e.g., 'None' or 'Resolved')")
+            project_id_input = gr.Textbox(label="Project ID", placeholder="Enter Salesforce Project ID (e.g., 'a0Axx000000xxxx')")
+            start_date_input = gr.Textbox(label="Start Date (YYYY-MM-DD)", placeholder="e.g., 2025-05-01")
+            end_date_input = gr.Textbox(label="End Date (YYYY-MM-DD)", placeholder="e.g., 2025-05-19")
+            owner_id_input = gr.Textbox(label="Owner ID", placeholder="Enter Salesforce User/Group ID (e.g., '005xx000000xxxx')")
             submit_btn = gr.Button("Evaluate Readiness")
         with gr.Column(scale=3):
             score_output = gr.Number(label="Readiness Score (%)")
@@ -353,18 +454,23 @@ with gr.Blocks(css="""
     # Connect inputs to evaluate_readiness
     submit_btn.click(
         fn=evaluate_readiness,
-        inputs=[logs_input, qa_input, punch_input],
-        outputs=[score_output, checklist_output, missing_output, status_output, progress_output]
+        inputs=[logs_input, qa_input, punch_input, project_id_input, start_date_input, end_date_input, owner_id_input],
+        outputs=[score_output, checklist_output, missing_output, status_output, progress_output, 
+                 gr.State(), gr.State(), gr.State(), gr.State(), gr.State(), gr.State(), gr.State(), gr.State(), gr.State()]
     )
 
-    # Connect PDF button to generate_pdf
+    # Connect PDF button to generate_pdf and create Salesforce record
     pdf_btn.click(
         fn=generate_pdf,
-        inputs=[score_output, checklist_output, missing_output, status_output],
+        inputs=[score_output, checklist_output, missing_output, status_output, gr.State(), gr.State(), gr.State(), gr.State(), gr.State(), gr.State(), gr.State()],
         outputs=[pdf_output, pdf_debug]
+    ).then(
+        fn=create_salesforce_record,
+        inputs=[score_output, checklist_output, missing_output, status_output, gr.State(), gr.State(), gr.State(), gr.State(), gr.State(), gr.State(), gr.State(), gr.State(), gr.State(), pdf_output],
+        outputs=gr.Textbox(label="Salesforce Record Creation Status")
     )
     demo.launch()
 
 # Comment out demo.launch() for Hugging Face Spaces deployment
 # if __name__ == "__main__":
-     
+#     demo.launch()