Update app.py

app.py

@@ -140,12 +140,10 @@ def create_salesforce_record(score, checklist_summary, missing_summary, status,
         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  # Ensure correct value is used
+        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
-        # Explicitly convert escalated to lowercase boolean string for Salesforce
         alert_sent = str(bool(escalated)).lower()  # Converts True/False to "true"/"false"
         logging.info(f"Setting Alert_Sent__c to: {alert_sent}")
-        # Explicitly convert escalated to lowercase boolean string for Escalation_Flag__c
         escalation_flag = str(bool(escalated)).lower()  # Ensure this is also a proper boolean string
 
         # Create the record in Project_Closure_Handover__c without the PDF URL for now
@@ -280,9 +278,9 @@ def evaluate_readiness(logs, qa_report, punch_list_text):
         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
+        open_punch_items = 0 if punch_pass else 1  # Simplified for dropdown input
 
-        # Enhanced escalation logic: escalate if score is below 70% or open punch items are more than 2
+        # Enhanced escalation logic
         escalated = score < 70 or open_punch_items > 2
         logging.info(f"Escalation Check: Score < 70 ({score < 70}), Open Punch Items > 2 ({open_punch_items > 2}), Escalated: {escalated}")
 
@@ -294,7 +292,7 @@ def evaluate_readiness(logs, qa_report, punch_list_text):
         checklist_summary = "\n".join(checklist_details)
         missing_summary = "None" if not missing_items else ", ".join(missing_items)
 
-        # Generate progress bar HTML using div elements
+        # Generate progress bar HTML
         color_class = "red" if score < 70 else "yellow" if score <= 90 else "green"
         logging.info(f"Readiness Score: {score}%, Color Class: {color_class}")
         progress_bar = f"""
@@ -305,22 +303,21 @@ def evaluate_readiness(logs, qa_report, punch_list_text):
         </div>
         """
 
-        # Return necessary values 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, checklist_status)
     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
+# Generate PDF report with signature slots
 def generate_pdf(score, checklist_summary, missing_summary, checklist_status, logs, qa_report, punch_list_text):
     try:
-        # Sanitize inputs to ensure they are strings and handle None
+        # Sanitize inputs
         logging.info("Sanitizing inputs for PDF generation")
         score = str(float(score)) if score is not None else "0"
         checklist_summary = str(checklist_summary) if checklist_summary is not None else ""
         checklist_status = str(checklist_status) if checklist_status is not None else ""
 
-        # Remove non-ASCII characters to prevent rendering issues
+        # Remove non-ASCII characters
         checklist_summary = checklist_summary.encode('ascii', 'ignore').decode('ascii')
         checklist_status = checklist_status.encode('ascii', 'ignore').decode('ascii')
 
@@ -328,51 +325,33 @@ def generate_pdf(score, checklist_summary, missing_summary, checklist_status, lo
         pdf_path = "readiness_report.pdf"
         logging.info(f"Creating PDF at {pdf_path}")
 
-        # Create the PDF using reportlab
+        # Create the PDF
         c = canvas.Canvas(pdf_path, pagesize=letter)
-        
-        # Set initial font and draw title using Times-Bold
-        logging.info("Setting font to Times-Bold")
         c.setFont("Times-Bold", 16)
         c.drawString(50, 750, "Project Closure Readiness Report")
-        
-        # 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: {checklist_status}")
         c.drawString(50, 680, "Checklist Summary:")
-
-        # Draw checklist summary with Y-position overflow handling
-        logging.info("Drawing checklist summary")
         y = 660
-        BOTTOM_MARGIN = 50  # Minimum Y position before creating a new page
+        BOTTOM_MARGIN = 50
         for line in checklist_summary.split("\n"):
             if y < BOTTOM_MARGIN:
-                logging.info("Y position below margin, creating new page")
-                c.showPage()  # Create a new page
-                c.setFont("Times-Roman", 12)  # Reset font for new page
-                y = 750  # Reset Y position
+                c.showPage()
+                c.setFont("Times-Roman", 12)
+                y = 750
             c.drawString(50, y, line)
             y -= 20
-
-        # Draw signature slots
         if y - 40 < BOTTOM_MARGIN:
-            logging.info("Y position below margin, creating new page")
             c.showPage()
             c.setFont("Times-Roman", 12)
             y = 750
         c.drawString(50, y - 40, "Stakeholder Signature: ____________________")
-
         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, "Date: ____________________")
-
-        # Finalize the PDF
-        logging.info("Finalizing PDF")
         c.save()
 
         # Confirm the file exists
@@ -403,8 +382,18 @@ with gr.Blocks(css="""
     with gr.Row():
         with gr.Column(scale=2):
             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')")
+            qa_input = gr.Dropdown(
+                label="QA Report",
+                choices=["Approved", "Passed", "Cleared", "Pending", "Not Started"],
+                value="Pending",
+                allow_custom_value=False
+            )
+            punch_input = gr.Dropdown(
+                label="Punch List",
+                choices=["None", "Resolved", "Closed", "No Issues", "Open Items"],
+                value="Open Items",
+                allow_custom_value=False
+            )
             submit_btn = gr.Button("Evaluate and Generate PDF")
         with gr.Column(scale=3):
             score_output = gr.Number(label="Readiness Score (%)")
@@ -415,10 +404,9 @@ with gr.Blocks(css="""
                 gr.Markdown("### Handover Summary")
                 checklist_output = gr.Textbox(label="Checklist Summary")
                 missing_output = gr.Textbox(label="Missing Items")
-            # Add an output for Open Punch Items for debugging
             open_punch_items_output = gr.Number(label="Open Punch Items (Debug)")
-            pdf_output = gr.File(label="Download PDF Report", type="filepath", interactive=False)  # Output only, no auto-download
-            pdf_debug = gr.Textbox(label="PDF Debug Output")  # Temporary debug output
+            pdf_output = gr.File(label="Download PDF Report", type="filepath", interactive=False)
+            pdf_debug = gr.Textbox(label="PDF Debug Output")
 
     # Chain the evaluation, PDF generation, and Salesforce record creation
     submit_btn.click(
@@ -438,6 +426,6 @@ with gr.Blocks(css="""
             score_output, checklist_output, missing_output, status_output,
             gr.State(), gr.State(), gr.State(), gr.State(), open_punch_items_output, pdf_output
         ],
-        outputs=None  # No UI output for Salesforce status
+        outputs=None
     )
     demo.launch()