Update app.py

app.py

@@ -239,6 +239,9 @@ from langchain_core.prompts import ChatPromptTemplate
 from llama_index.core import StorageContext, VectorStoreIndex, SimpleDirectoryReader, Settings
 from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 # Define Pydantic model for incoming request body
 class MessageRequest(BaseModel):
@@ -287,18 +290,19 @@ Settings.embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-small-en-v1.5")
 # Salesforce credentials
 username = os.getenv("username")
 password = os.getenv("password")
-security_token = os.getenv("security_token zasi")
+security_token = os.getenv("security_token")
 domain = os.getenv("domain")  # e.g., 'test' for sandbox
 
-# Initialize Salesforce connection
+# Initialize Salesforce connection (allow failure)
+sf = None
 try:
     session_id, sf_instance = SalesforceLogin(
         username=username, password=password, security_token=security_token, domain=domain
     )
     sf = Salesforce(instance=sf_instance, session_id=session_id)
+    logger.info("Salesforce connection established")
 except Exception as e:
-    logger.error(f"Failed to connect to Salesforce: {e}")
-    raise HTTPException(status_code=500, detail="Salesforce connection failed")
+    logger.error(f"Failed to connect to Salesforce: {e}. Continuing without Salesforce integration.")
 
 # Chat history
 chat_history = []
@@ -316,6 +320,9 @@ os.makedirs(PERSIST_DIR, exist_ok=True)
 def data_ingestion_from_directory():
     try:
         documents = SimpleDirectoryReader(PDF_DIRECTORY).load_data()
+        if not documents:
+            logger.warning("No documents found in PDF_DIRECTORY")
+            return
         storage_context = StorageContext.from_defaults()
         index = VectorStoreIndex.from_documents(documents)
         index.storage_context.persist(persist_dir=PERSIST_DIR)
@@ -389,10 +396,14 @@ async def load_chat(request: Request, id: str):
 
 @app.post("/hist/")
 async def save_chat_history(history: dict):
+    if not sf:
+        logger.error("Salesforce integration is disabled")
+        return {"error": "Salesforce integration is unavailable"}, 503
+
     user_id = history.get('userId')
     if not user_id:
         logger.error("userId is missing in history request")
-        raise HTTPException(status_code=400, detail="userId is required")
+        return {"error": "userId is required"}, 400
 
     hist = ''.join([f"'{entry['sender']}: {entry['message']}'\n" for entry in history['history']])
     hist = "You are a Redfernstech summarize model. Identify user interests from this conversation: " + hist
@@ -402,17 +413,21 @@ async def save_chat_history(history: dict):
         logger.info(f"Chat history updated for user {user_id}")
     except Exception as e:
         logger.error(f"Failed to update lead: {e}")
-        raise HTTPException(status_code=500, detail=f"Failed to update lead: {str(e)}")
+        return {"error": f"Failed to update lead: {str(e)}"}, 500
     
     return {"summary": hist, "message": "Chat history saved"}
 
 @app.post("/webhook")
 async def receive_form_data(request: Request):
+    if not sf:
+        logger.error("Salesforce integration is disabled")
+        return {"error": "Salesforce integration is unavailable"}, 503
+
     try:
         form_data = await request.json()
     except json.JSONDecodeError:
         logger.error("Invalid JSON in webhook request")
-        raise HTTPException(status_code=400, detail="Invalid JSON")
+        return {"error": "Invalid JSON"}, 400
 
     first_name, last_name = split_name(form_data.get('name', ''))
     data = {
@@ -431,7 +446,7 @@ async def receive_form_data(request: Request):
         return JSONResponse({"id": unique_id})
     except Exception as e:
         logger.error(f"Failed to create lead: {e}")
-        raise HTTPException(status_code=500, detail=f"Failed to create lead: {str(e)}")
+        return {"error": f"Failed to create lead: {str(e)}"}, 500
 
 @app.post("/chat/")
 async def chat(request: MessageRequest):