Update app.py

app.py

@@ -6,16 +6,18 @@ import bleach
 import logging
 import os
 from huggingface_hub import InferenceClient
+from retry import retry
+import time
 
 # Print statement to confirm script initialization
 print("Starting Project Closure Readiness Evaluator app...")
 
 # Set up console logging for debugging
-logging.basicConfig(level=logging.ERROR)
+logging.basicConfig(level=logging.DEBUG)
 logger = logging.getLogger()
 logger.handlers = []  # Clear existing handlers
 console_handler = logging.StreamHandler()
-console_handler.setLevel(logging.ERROR)
+console_handler.setLevel(logging.DEBUG)
 formatter = logging.Formatter("%(asctime)s - %(levelname)s - %(message)s")
 console_handler.setFormatter(formatter)
 logger.addHandler(console_handler)
@@ -34,6 +36,11 @@ SF_PASSWORD = os.getenv("SF_PASSWORD")
 SF_SECURITY_TOKEN = os.getenv("SF_SECURITY_TOKEN")
 SF_INSTANCE_URL = os.getenv("SF_INSTANCE_URL")
 
+# Validate Salesforce environment variables
+if not all([SF_USERNAME, SF_PASSWORD, SF_SECURITY_TOKEN, SF_INSTANCE_URL]):
+    logging.error("One or more Salesforce environment variables are missing. Salesforce functionality will be disabled.")
+    SALESFORCE_AVAILABLE = False
+
 # Hugging Face configuration
 HF_API_TOKEN = os.getenv("HF_API_TOKEN")
 if not HF_API_TOKEN:
@@ -43,7 +50,8 @@ else:
     HF_AVAILABLE = True
     hf_client = InferenceClient(token=HF_API_TOKEN)
 
-# Initialize Salesforce connection
+# Initialize Salesforce connection with retry mechanism
+@retry(tries=3, delay=2, backoff=2, logger=logger)
 def init_salesforce():
     if not SALESFORCE_AVAILABLE:
         logging.error("Salesforce library not available. Skipping connection.")
@@ -56,10 +64,14 @@ def init_salesforce():
             security_token=SF_SECURITY_TOKEN,
             instance_url=SF_INSTANCE_URL
         )
+        logging.info("Salesforce connected successfully")
         return sf, "Salesforce connected successfully"
+    except SalesforceError as e:
+        logging.error(f"Salesforce authentication failed: {str(e)}")
+        raise
     except Exception as e:
         logging.error(f"Failed to initialize Salesforce connection: {str(e)}")
-        return None, f"Salesforce connection failed: {str(e)}"
+        raise
 
 # Summarize text using Hugging Face Inference API
 def summarize_text(text, max_length=100, min_length=30):
@@ -85,32 +97,44 @@ def summarize_text(text, max_length=100, min_length=30):
 def create_salesforce_record(score, checklist_summary, missing_summary, status):
     if not SALESFORCE_AVAILABLE:
         logging.error("Salesforce library not available. Skipping record creation.")
-        return
+        return "Salesforce library not available"
 
     try:
         sf, connection_message = init_salesforce()
         if not sf:
             logging.error(f"Skipping Salesforce record creation due to connection failure: {connection_message}")
-            return
+            return connection_message
 
         # Summarize checklist_summary and missing_summary using Hugging Face
         summarized_checklist = summarize_text(checklist_summary)
         summarized_missing = summarize_text(missing_summary)
 
+        # Ensure inputs are strings and sanitized
+        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 ""
+
         # Create a record in the custom object Project_Closure_Handover__c
         record = {
             "Readiness_Score__c": score,
             "Checklist_Summary__c": checklist_summary,
             "Missing_Documents__c": missing_summary,
             "Status__c": status,
-            "Checklist_Summary__c": summarized_checklist,
             "Summarized_Missing_Items__c": summarized_missing
         }
-        sf.Project_Closure_Handover__c.create(record)
-        logging.info("Successfully created record in Project_Closure_Handover__c")
+        logging.debug(f"Attempting to create Salesforce record: {record}")
+        result = sf.Project_Closure_Handover__c.create(record)
+        logging.info(f"Successfully created Salesforce record: {result}")
+        return "Record created successfully"
+    except SalesforceError as e:
+        logging.error(f"Salesforce error while creating Project_Closure_Handover__c record: {str(e)}")
+        return f"Salesforce error: {str(e)}"
     except Exception as e:
         logging.error(f"Failed to create Salesforce Project_Closure_Handover__c record: {str(e)}")
-        pass
+        return f"Error creating record: {str(e)}"
 
 # Clean input to prevent injection attacks
 def sanitize_input(text):
@@ -192,7 +216,8 @@ def evaluate_readiness(logs, qa_report, punch_list_text):
         """
 
         # Automatically create Salesforce record
-        create_salesforce_record(score, checklist_summary, missing_summary, status)
+        sf_result = create_salesforce_record(score, checklist_summary, missing_summary, status)
+        logging.info(f"Salesforce record creation result: {sf_result}")
 
         return score, checklist_summary, missing_summary, status, progress_bar
     except Exception as e:
@@ -339,8 +364,6 @@ with gr.Blocks(css="""
         outputs=[pdf_output, pdf_debug]
     )
 
-# Launch the app (commented out for Hugging Face deployment)
-# demo.launch()
-
-if __name__ == "__main__":
-    demo.launch()
+# Comment out demo.launch() for Hugging Face Spaces deployment
+# if __name__ == "__main__":
+#     demo.launch()