Update app.py

app.py

@@ -3,8 +3,9 @@ import re
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
 import bleach
-import os
+import io
 import logging
+import os
 
 # Print statement to confirm script initialization
 print("Starting Project Closure Readiness Evaluator app...")
@@ -87,41 +88,46 @@ def sanitize_input(text):
         return ""
     return bleach.clean(text.strip())
 
-# Rule-based completeness engine to evaluate readiness and create Salesforce record
-def evaluate_readiness_and_save(logs, qa_report, punch_list_text):
+# Rule-based completeness engine with weighted scoring
+def evaluate_readiness(logs, qa_report, punch_list_text):
     try:
-        # Evaluate readiness
-        score = 0  # Base score set to 0 for clearer range
+        # Initialize score and lists for tracking
+        score = 0
         missing_items = []
         checklist_details = []
 
+        # Define weights for scoring
+        LOGS_WEIGHT = 30  # 30% weight for logs
+        QA_WEIGHT = 40    # 40% weight for QA report
+        PUNCH_WEIGHT = 30 # 30% weight for punch list
+
         # Sanitize inputs
         logs = sanitize_input(logs)
         qa_report = sanitize_input(qa_report)
         punch_list_text = sanitize_input(punch_list_text)
 
-        # Process Project Logs
+        # Process Project Logs (30% weight)
         log_keywords = r"complete|handover done|finished|closed|successful"
         if logs and re.search(log_keywords, logs.lower()):
-            score += 30
+            score += LOGS_WEIGHT
             checklist_details.append("Logs: Completed")
         else:
             missing_items.append("Project Logs Incomplete")
             checklist_details.append("Logs: Pending")
 
-        # Process QA Report
+        # Process QA Report (40% weight)
         qa_keywords = r"approved|passed|cleared"
         if qa_report and re.search(qa_keywords, qa_report.lower()):
-            score += 40
+            score += QA_WEIGHT
             checklist_details.append("QA Report: Approved")
         else:
             missing_items.append("QA Approval Missing")
             checklist_details.append("QA Report: Pending")
 
-        # Process Punch List
+        # Process Punch List (30% weight)
         punch_keywords = r"none|resolved|closed|no issues"
         if punch_list_text and re.search(punch_keywords, punch_list_text.lower()):
-            score += 30
+            score += PUNCH_WEIGHT
             checklist_details.append("Punch List: Resolved")
         else:
             missing_items.append("Open Punch Points Detected")
@@ -144,14 +150,15 @@ def evaluate_readiness_and_save(logs, qa_report, punch_list_text):
 
         return score, checklist_summary, missing_summary, status, progress_bar
     except Exception as e:
-        logging.error(f"Error in evaluate_readiness_and_save: {str(e)}")
+        logging.error(f"Error in evaluate_readiness: {str(e)}")
         raise
 
-# Generate PDF report with signature slots
+# Generate PDF report with signature slots using in-memory buffer
 def generate_pdf(score, checklist_summary, missing_summary, status):
     try:
-        pdf_file = "readiness_report.pdf"
-        c = canvas.Canvas(pdf_file, pagesize=letter)
+        # Use an in-memory buffer instead of writing to disk
+        buffer = io.BytesIO()
+        c = canvas.Canvas(buffer, pagesize=letter)
         c.setFont("Helvetica-Bold", 16)
         c.drawString(50, 750, "Project Closure Readiness Report")
         c.setFont("Helvetica", 12)
@@ -167,7 +174,8 @@ def generate_pdf(score, checklist_summary, missing_summary, status):
         c.drawString(50, y-80, "Stakeholder Signature: ____________________")
         c.drawString(50, y-100, "Date: ____________________")
         c.save()
-        return pdf_file
+        buffer.seek(0)
+        return buffer, "readiness_report.pdf"
     except Exception as e:
         logging.error(f"Error in generate_pdf: {str(e)}")
         raise
@@ -201,9 +209,9 @@ with gr.Blocks(css="""progress { background-color: #f0f0f0; } .red { background-
             pdf_btn = gr.Button("Download PDF Report")
             pdf_output = gr.File(label="PDF Report")
 
-    # Connect inputs to evaluate_readiness_and_save
+    # Connect inputs to evaluate_readiness
     submit_btn.click(
-        fn=evaluate_readiness_and_save,
+        fn=evaluate_readiness,
         inputs=[logs_input, qa_input, punch_input],
         outputs=[score_output, checklist_output, missing_output, status_output, progress_output]
     )
@@ -212,7 +220,7 @@ with gr.Blocks(css="""progress { background-color: #f0f0f0; } .red { background-
     pdf_btn.click(
         fn=generate_pdf,
         inputs=[score_output, checklist_output, missing_output, status_output],
-        outputs=pdf_output
+        outputs=[pdf_output]
     )
 
 # Launch the app (commented out for Hugging Face deployment)