Update app.py

app.py

@@ -5,6 +5,7 @@ from reportlab.pdfgen import canvas
 import bleach
 import logging
 import os
+from huggingface_hub import InferenceClient
 
 # Print statement to confirm script initialization
 print("Starting Project Closure Readiness Evaluator app...")
@@ -33,6 +34,15 @@ SF_PASSWORD = os.getenv("SF_PASSWORD")
 SF_SECURITY_TOKEN = os.getenv("SF_SECURITY_TOKEN")
 SF_INSTANCE_URL = os.getenv("SF_INSTANCE_URL")
 
+# 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.")
+    HF_AVAILABLE = False
+else:
+    HF_AVAILABLE = True
+    hf_client = InferenceClient(token=HF_API_TOKEN)
+
 # Initialize Salesforce connection
 def init_salesforce():
     if not SALESFORCE_AVAILABLE:
@@ -51,7 +61,27 @@ def init_salesforce():
         logging.error(f"Failed to initialize Salesforce connection: {str(e)}")
         return None, f"Salesforce connection failed: {str(e)}"
 
-# Create Salesforce record using Case object
+# 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):
     if not SALESFORCE_AVAILABLE:
         logging.error("Salesforce library not available. Skipping record creation.")
@@ -63,22 +93,23 @@ def create_salesforce_record(score, checklist_summary, missing_summary, status):
             logging.error(f"Skipping Salesforce record creation due to connection failure: {connection_message}")
             return
 
-        # Map app status to Salesforce Case.Status values
-        sf_status = "New"  # Default for Pending and Escalated
-        if status == "Completed":
-            sf_status = "Closed"
+        # Summarize checklist_summary and missing_summary using Hugging Face
+        summarized_checklist = summarize_text(checklist_summary)
+        summarized_missing = summarize_text(missing_summary)
 
-        # Create a Case record in Salesforce with required fields
+        # Create a record in the custom object Project_Closure_Handover__c
         record = {
-            "Subject": "Project Readiness Evaluation",
-            "Description": f"Readiness Score: {score}%\nChecklist Summary:\n{checklist_summary}\nMissing Items: {missing_summary}",
-            "Status": sf_status,
-            "Origin": "Web",  # Required field in many Salesforce orgs
-            "Priority": "Medium"  # Required field in many Salesforce orgs
+            "Readiness_Score__c": score,
+            "Checklist_Summary__c": checklist_summary,
+            "Missing_Items__c": missing_summary,
+            "Overall_Status__c": status,
+            "Summarized_Checklist__c": summarized_checklist,
+            "Summarized_Missing_Items__c": summarized_missing
         }
-        sf.Case.create(record)
+        sf.Project_Closure_Handover__c.create(record)
+        logging.info("Successfully created record in Project_Closure_Handover__c")
     except Exception as e:
-        logging.error(f"Failed to create Salesforce Case record: {str(e)}")
+        logging.error(f"Failed to create Salesforce Project_Closure_Handover__c record: {str(e)}")
         pass
 
 # Clean input to prevent injection attacks
@@ -272,7 +303,7 @@ with gr.Blocks(css="""
         """
         # Project Closure Readiness Evaluator
         Evaluate project readiness and generate a PDF report.
-        Results are automatically saved to Salesforce.
+        Results are automatically saved to Salesforce in Project_Closure_Handover__c.
         """
     )
     with gr.Row():