Update app.py

app.py

@@ -5,11 +5,11 @@ from reportlab.pdfgen import canvas
 import bleach
 import logging
 import os
+from huggingface_hub import InferenceClient
 from retry import retry
 import time
 from datetime import datetime
 import base64
-import pytz  # For proper timezone handling
 
 # Print statement to confirm script initialization
 print("Starting Project Closure Readiness Evaluator app...")
@@ -49,6 +49,16 @@ if not all([SF_USERNAME, SF_PASSWORD, SF_SECURITY_TOKEN, SF_INSTANCE_URL]):
     logging.error("Please set these variables in Hugging Face Space Settings > Secrets.")
     SALESFORCE_AVAILABLE = False
 
+# Hugging Face configuration
+HF_API_TOKEN = os.getenv("HF_API_TOKEN")
+if not HF_API_TOKEN:
+    logging.error("Hugging Face API token (HF_API_TOKEN) not found. Hugging Face functionality will be disabled.")
+    logging.error("Please set HF_API_TOKEN in Hugging Face Space Settings > Secrets.")
+    HF_AVAILABLE = False
+else:
+    HF_AVAILABLE = True
+    hf_client = InferenceClient(token=HF_API_TOKEN)
+
 # Initialize Salesforce connection with retry mechanism
 @retry(tries=3, delay=2, backoff=2, logger=logger)
 def init_salesforce():
@@ -83,6 +93,26 @@ def init_salesforce():
         logging.error("Check your Salesforce org settings, network restrictions, or API access.")
         raise
 
+# Summarize text using Hugging Face Inference API
+def summarize_text(text, max_length=100, min_length=30):
+    if not HF_AVAILABLE:
+        logging.error("Hugging Face API not available. Returning original text.")
+        return text
+
+    if not text or text == "None":
+        return "No summary available"
+
+    try:
+        summary = hf_client.summarization(
+            text,
+            model="facebook/bart-large-cnn",
+            parameters={"max_length": max_length, "min_length": min_length}
+        )
+        return summary
+    except Exception as e:
+        logging.error(f"Failed to summarize text with Hugging Face: {str(e)}")
+        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):
     if not SALESFORCE_AVAILABLE:
@@ -95,37 +125,34 @@ def create_salesforce_record(score, checklist_summary, missing_summary, status,
             logging.error(f"Skipping Salesforce record creation due to connection failure: {connection_message}")
             return connection_message
 
-        # Generate a unique Project ID based on the current timestamp
-        ist_tz = pytz.timezone("Asia/Kolkata")
-        current_time = datetime.now(ist_tz)
-        project_id = f"PROJ-{current_time.strftime('%Y%m%d-%H%M%S')}"  # e.g., PROJ-20250526-103315
-        logging.info(f"Generated Project ID: {project_id}")
+        # Summarize checklist_summary and missing_summary using Hugging Face
+        summarized_checklist = summarize_text(checklist_summary)
+        summarized_missing = summarize_text(missing_summary)
 
         # 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
-        
-        # Set timezone to Asia/Kolkata (IST) and get current time
-        evaluated_at = current_time.strftime("%Y-%m-%dT%H:%M:%SZ")  # Salesforce Date/Time format in UTC with Z
+        evaluated_at = datetime.now().strftime("%Y-%m-%dT%H:%M:%SZ")  # Salesforce Date/Time format
         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,  # Use the generated Project ID
             "Readiness_Score__c": score,
             "Checklist_Summary__c": checklist_summary,
             "Missing_Documents__c": missing_documents,
             "Status__c": status,
-            "Summarized_Missing_Items__c": missing_summary,
+            "Summarized_Missing_Items__c": summarized_missing,
             "Alert_Sent__c": alert_sent,
             "Client_PDF_Pack_URL__c": "",  # Will be updated after attachment is created
             "Closure_Pack_URL__c": "",  # Placeholder; update if you have a closure pack URL
@@ -349,7 +376,7 @@ def generate_pdf(score, checklist_summary, missing_summary, checklist_status, lo
 
 # Gradio interface with updated UI
 with gr.Blocks(css="""
-    .progress-container { background-color: #f0f0f0; width: 100%; height: 20px; border-radius: 5px; overflowJavascript hidden; position: relative; }
+    .progress-container { background-color: #f0f0f0; width: 100%; height: 20px; border-radius: 5px; overflow: hidden; position: relative; }
     .progress-bar { height: 100%; text-align: center; line-height: 20px; color: #000; font-size: 12px; }
     .progress-bar.red { background-color: #FF0000; }
     .progress-bar.yellow { background-color: #FFFF00; }
@@ -363,7 +390,6 @@ with gr.Blocks(css="""
     )
     with gr.Row():
         with gr.Column(scale=2):
-            # Removed 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",