Update app.py

app.py

@@ -10,6 +10,7 @@ from retry import retry
 import time
 from datetime import datetime
 import base64
+import pytz  # Added for proper timezone handling
 
 # Print statement to confirm script initialization
 print("Starting Project Closure Readiness Evaluator app...")
@@ -114,7 +115,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(project_id, score, checklist_summary, missing_summary, status, escalated, logs, qa_report, punch_list_text, open_punch_items, pdf_path=None):
     if not SALESFORCE_AVAILABLE:
         logging.error("Salesforce library not available. Skipping record creation.")
         return "Salesforce library not available"
@@ -141,13 +142,17 @@ def create_salesforce_record(score, checklist_summary, missing_summary, status,
         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
-        evaluated_at = datetime.now().strftime("%Y-%m-%dT%H:%M:%SZ")  # Salesforce Date/Time format
+        
+        # Set timezone to Asia/Kolkata (IST) and get current time
+        ist_tz = pytz.timezone("Asia/Kolkata")
+        evaluated_at = datetime.now(ist_tz).strftime("%Y-%m-%dT%H:%M:%SZ")  # Salesforce Date/Time format in UTC with Z
         alert_sent = str(bool(escalated)).lower()  # Converts True/False to "true"/"false"
         logging.info(f"Setting Alert_Sent__c to: {alert_sent}")
         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
         record = {
+            "Project_ID__c": project_id,  # Added Project ID field
             "Readiness_Score__c": score,
             "Checklist_Summary__c": checklist_summary,
             "Missing_Documents__c": missing_documents,
@@ -224,10 +229,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(project_id, logs, qa_report, punch_list_text):
     try:
         # Log inputs for debugging
-        logging.info(f"Inputs - Logs: {logs}, QA Report: {qa_report}, Punch List: {punch_list_text}")
+        logging.info(f"Inputs - Project ID: {project_id}, Logs: {logs}, QA Report: {qa_report}, Punch List: {punch_list_text}")
 
         # Initialize score and lists for tracking
         score = 0
@@ -240,6 +245,7 @@ def evaluate_readiness(logs, qa_report, punch_list_text):
         PUNCH_WEIGHT = 30 # 30% weight for punch list
 
         # Sanitize inputs
+        project_id = sanitize_input(project_id)
         logs = sanitize_input(logs)
         qa_report = sanitize_input(qa_report)
         punch_list_text = sanitize_input(punch_list_text)
@@ -390,6 +396,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-12345)")  # Added Project ID input
             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",
@@ -420,7 +427,7 @@ with gr.Blocks(css="""
     # Chain the evaluation, PDF generation, and Salesforce record creation
     submit_btn.click(
         fn=evaluate_readiness,
-        inputs=[logs_input, qa_input, punch_input],
+        inputs=[project_id_input, logs_input, qa_input, punch_input],
         outputs=[
             score_output, checklist_output, missing_output, status_output, progress_output,
             gr.State(), gr.State(), gr.State(), gr.State(), open_punch_items_output, status_output
@@ -432,7 +439,7 @@ with gr.Blocks(css="""
     ).then(
         fn=create_salesforce_record,
         inputs=[
-            score_output, checklist_output, missing_output, status_output,
+            project_id_input, score_output, checklist_output, missing_output, status_output,
             gr.State(), gr.State(), gr.State(), gr.State(), open_punch_items_output, pdf_output
         ],
         outputs=None