Update app.py

app.py

@@ -1,47 +1,130 @@
 import gradio as gr
+import re
+from reportlab.lib.pagesizes import letter
+from reportlab.pdfgen import canvas
+import bleach
 
+# Clean input to prevent injection attacks
+def sanitize_input(text):
+    if not text or not isinstance(text, str):
+        return ""
+    return bleach.clean(text.strip())
+
+# Evaluate project readiness with enhanced logic
 def evaluate_readiness(logs, qa_report, punch_list_text):
-    score = 60  # Base score
+    score = 0  # Base score set to 0 for clearer range
     missing_items = []
+    checklist_details = []
+
+    # Sanitize inputs
+    logs = sanitize_input(logs)
+    qa_report = sanitize_input(qa_report)
+    punch_list_text = sanitize_input(punch_list_text)
 
     # Process Project Logs
-    if "complete" in logs.lower() or "handover done" in logs.lower():
-        score += 10
+    log_keywords = r"complete|handover done|finished|closed|successful"
+    if logs and re.search(log_keywords, logs.lower()):
+        score += 30
+        checklist_details.append("Logs: Completed")
     else:
         missing_items.append("Project Logs Incomplete")
+        checklist_details.append("Logs: Pending")
 
     # Process QA Report
-    if qa_report.strip().lower() in ["approved", "passed"]:
-        score += 20
+    qa_keywords = r"approved|passed|cleared"
+    if qa_report and re.search(qa_keywords, qa_report.lower()):
+        score += 40
+        checklist_details.append("QA Report: Approved")
     else:
         missing_items.append("QA Approval Missing")
+        checklist_details.append("QA Report: Pending")
 
-    # Process Punch List as plain text
-    if "none" in punch_list_text.lower() or "resolved" in punch_list_text.lower():
-        score += 10
+    # Process Punch List
+    punch_keywords = r"none|resolved|closed|no issues"
+    if punch_list_text and re.search(punch_keywords, punch_list_text.lower()):
+        score += 30
+        checklist_details.append("Punch List: Resolved")
     else:
         missing_items.append("Open Punch Points Detected")
+        checklist_details.append("Punch List: Pending")
+
+    # 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")
 
-    # Build output summary
-    checklist_summary = "All checklist items reviewed." if not missing_items else "Pending: " + ", ".join(missing_items)
+    # Build summaries
+    checklist_summary = "\n".join(checklist_details)
     missing_summary = "None" if not missing_items else ", ".join(missing_items)
 
-    return score, checklist_summary, missing_summary
-
-# Gradio interface setup
-demo = gr.Interface(
-    fn=evaluate_readiness,
-    inputs=[
-        gr.Textbox(label="Project Logs"),
-        gr.Textbox(label="QA Report"),
-        gr.Textbox(label="Punch List (Plain Text)")
-    ],
-    outputs=[
-        gr.Number(label="Readiness Score (%)"),
-        gr.Textbox(label="Checklist Summary"),
-        gr.Textbox(label="Missing Items")
-    ],
-    title="Project Closure Readiness Evaluator"
-)
-
-demo.launch()
+    # Generate progress bar HTML
+    color = "red" if score < 70 else "yellow" if score < 90 else "green"
+    progress_bar = f'<progress value="{score}" max="100" style="width:100%;height:20px;background-color:{color}">{score}%</progress>'
+
+    return score, checklist_summary, missing_summary, status, progress_bar
+
+# Generate PDF report
+def generate_pdf(score, checklist_summary, missing_summary, status):
+    pdf_file = "readiness_report.pdf"
+    c = canvas.Canvas(pdf_file, pagesize=letter)
+    c.setFont("Helvetica-Bold", 16)
+    c.drawString(50, 750, "Project Closure Readiness Report")
+    c.setFont("Helvetica", 12)
+    c.drawString(50, 720, f"Readiness Score: {score}%")
+    c.drawString(50, 700, f"Status: {status}")
+    c.drawString(50, 680, "Checklist Summary:")
+    y = 660
+    for line in checklist_summary.split("\n"):
+        c.drawString(50, y, line)
+        y -= 20
+    c.drawString(50, y-20, "Missing Items:")
+    c.drawString(50, y-40, missing_summary)
+    c.drawString(50, y-80, "Stakeholder Signature: ____________________")
+    c.drawString(50, y-100, "Date: ____________________")
+    c.save()
+    return pdf_file
+
+# Gradio interface with enhanced UI
+with gr.Blocks(css="""progress { background-color: #f0f0f0; } .red { background-color: red; }""") as demo:
+    gr.Markdown(
+        """
+        # Project Closure Readiness Evaluator
+        Evaluate project readiness based on logs, QA report, and punch list status.
+        """
+    )
+    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')")
+            submit_btn = gr.Button("Evaluate Readiness")
+        with gr.Column(scale=3):
+            score_output = gr.Number(label="Readiness Score (%)")
+            progress_output = gr.HTML(label="Progress")
+            status_output = gr.Textbox(label="Overall Status")
+            checklist_output = gr.Textbox(label="Checklist Summary")
+            missing_output = gr.Textbox(label="Missing Items")
+            with gr.Accordion("Details", open=False):
+                gr.Markdown("Detailed breakdown of checklist items and alerts.")
+                checklist_output_detailed = gr.Textbox(label="Checklist Details", interactive=False)
+            pdf_btn = gr.Button("Download PDF Report")
+            pdf_output = gr.File(label="PDF Report")
+
+    # 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]
+    )
+
+    # Connect PDF button to generate_pdf
+    pdf_btn.click(
+        fn=generate_pdf,
+        inputs=[score_output, checklist_output, missing_output, status_output],
+        outputs=pdf_output
+    )
+
+# Launch the app (commented out for Hugging Face deployment)
+# demo.launch()
+
+if __name__ == "__main__":
+    demo.launch()