Update app.py

app.py

@@ -256,11 +256,11 @@ class MessageRequest(BaseModel):
     message: str
 
 # Environment variables for Salesforce
-os.environ["HF_TOKEN"] = os.getenv("HF_TOKEN")  # Optional: Needed for model download if gated
+os.environ["HF_TOKEN"] = os.getenv("HF_TOKEN", "")  # Optional: Needed for model download if gated
 username = os.getenv("username")
 password = os.getenv("password")
 security_token = os.getenv("security_token")
-domain = os.getenv("domain")  # Using sandbox environment
+domain = os.getenv("domain", "test")  # Default to sandbox environment
 session_id, sf_instance = SalesforceLogin(username=username, password=password, security_token=security_token, domain=domain)
 
 # Create Salesforce object
@@ -368,10 +368,10 @@ def handle_query(query):
     
     storage_context = StorageContext.from_defaults(persist_dir=PERSIST_DIR)
     index = load_index_from_storage(storage_context)
-    context_str = ""
-    for past_query, response in reversed(current_chat_history):
-        if past_query.strip():
-            context_str += f"User asked: '{past_query}'\nBot answered: '{response}'\n"
+    
+    # Build context from chat history
+    context_str = "\n".join([f"User asked: '{past_query}'\nBot answered: '{response}'" 
+                             for past_query, response in reversed(current_chat_history)])
     
     logger.info(f"Query: {query}")
     logger.info(f"Context: {context_str}")
@@ -411,16 +411,13 @@ async def save_chat_history(history: dict):
     hist = "You are a Redfernstech summarize model. Your aim is to use this conversation to identify user interests solely based on that conversation: " + hist
     logger.info(f"Chat history: {hist}")
 
-    # Get the summarized result
-    result = hist
-
     try:
-        sf.Lead.update(user_id, {"Description": result})
+        sf.Lead.update(user_id, {"Description": hist})
     except Exception as e:
         logger.error(f"Failed to update lead: {str(e)}")
         return {"error": f"Failed to update lead: {str(e)}"}, 500
     
-    return {"summary": result, "message": "Chat history saved"}
+    return {"summary": hist, "message": "Chat history saved"}
 
 @app.post("/webhook")
 async def receive_form_data(request: Request):
@@ -430,8 +427,8 @@ async def receive_form_data(request: Request):
     data = {
         "FirstName": first_name,
         "LastName": last_name,
-        "Description": "hii",  # Static description
-        "Company": form_data["company"],  # Assuming company is available in form_data
+        "Description": f"New lead created via webhook: {form_data['name']}",
+        "Company": form_data.get("company", ""),  # Assuming company might be optional
         "Phone": form_data["phone"].strip(),  # Phone from form data
         "Email": form_data["email"],  # Email from form data
     }