Update app.py

app.py

@@ -6,38 +6,39 @@ from datetime import datetime
 from simple_salesforce import Salesforce, SalesforceAuthenticationFailed
 from flask import Flask, jsonify, request, render_template, redirect, url_for
 
-# Configure logging with more detail
-logging.basicConfig(
-    level=logging.DEBUG,  # Set to DEBUG for detailed logs
-    format='%(asctime)s - %(levelname)s - %(message)s',
-    handlers=[
-        logging.StreamHandler()  # Output logs to console
-    ])
+# Set up logging to capture errors and debug information
+logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Hugging Face API configuration
-HUGGING_FACE_API_URL = "https://api-inference.huggingface.co/models/distilgpt2"
-HUGGING_FACE_API_TOKEN = "your_hugging_face_api_token_here"  # Replace with your actual Hugging Face API token
+app = FastAPI()
 
-# Salesforce credentials (use environment variables for security)
-#SF_USERNAME = os.getenv('Ai@Coach.com')
-#SF_PASSWORD = os.getenv('Teja90325@')
-#SF_SECURITY_TOKEN = os.getenv('clceSdBgQ30Rx9BSC66gAcRx')
-#SF_DOMAIN = os.getenv('SF_DOMAIN', 'login')  # Default to 'login' if not set
-#HUGGINGFACE_API_KEY = os.getenv('HUGGINGFACE_API_KEY')
-
-# Salesforce configuration
-#SALESFORCE_USERNAME = "Ai@Coach.com"
-#SALESFORCE_PASSWORD = "Teja90325@"
-#SALESFORCE_SECURITY_TOKEN = "clceSdBgQ30Rx9BSC66gAcRx"
-#SALESFORCE_DOMAIN = "login.salesforce.com"
-
-#Salesforce credentials
+# Salesforce credentials
 SF_USERNAME = os.getenv("SF_USERNAME", "Ai@Coach.com")
 SF_PASSWORD = os.getenv("SF_PASSWORD", "Teja90325@")
 SF_SECURITY_TOKEN = os.getenv("SF_SECURITY_TOKEN", "clceSdBgQ30Rx9BSC66gAcRx")
 SF_DOMAIN = os.getenv("SF_DOMAIN", "login")
 
+# Verify API key is set
+API_KEY = os.getenv("HUGGING_FACE_API_KEY")
+if not API_KEY:
+    logger.error("HUGGING_FACE_API_KEY environment variable not set")
+    raise ValueError("HUGGING_FACE_API_KEY environment variable not set")
+
+# Connect to Salesforce
+try:
+    sf = Salesforce(
+        username=SF_USERNAME,
+        password=SF_PASSWORD,
+        security_token=SF_SECURITY_TOKEN,
+        domain=SF_DOMAIN
+    )
+    logger.info("Successfully connected to Salesforce")
+except Exception as e:
+    logger.error(f"Failed to connect to Salesforce: {str(e)}")
+    raise
+
+
+
 
 # Validate configuration
 if not HUGGING_FACE_API_TOKEN: