Update app.py

app.py

@@ -4,6 +4,7 @@ from dotenv import load_dotenv
 from llama_index.core import StorageContext, load_index_from_storage, VectorStoreIndex, SimpleDirectoryReader, ChatPromptTemplate, Settings
 from llama_index.llms.huggingface import HuggingFaceInferenceAPI
 from llama_index.embeddings.huggingface import HuggingFaceEmbedding
+from simple_salesforce import Salesforce, SalesforceLogin
 import random
 import datetime
 import uuid
@@ -92,6 +93,26 @@ def save_chat_history(history):
         json.dump(history, f)
     print(f"Chat history saved as {chat_history_path}")
 
+    # Save to Salesforce
+    save_to_salesforce(current_chat_history)
+
+def save_to_salesforce(history):
+    username =os.getenv("username")
+    password =os.getenv("password")
+    security_token =os.getenv("security_token")
+    domain = 'test'  
+
+    session_id, sf_instance = SalesforceLogin(username=username, password=password, security_token=security_token, domain=domain)
+    sf = Salesforce(instance=sf_instance, session_id=session_id)
+    for past_query, response in history:
+        data = {
+            'Name': 'Chat with user',
+            'Bot_Message__c': response,
+            'User_Message__c': past_query,
+            'Date__c': str(datetime.datetime.now().date())
+        }
+        sf.Chat_History__c.create(data)
+
 # Define the function to handle predictions
 def predict(message, history):
     logo_html = '''
@@ -107,7 +128,7 @@ def predict(message, history):
     
     return response_with_logo
 
-# Define your Gradio chat interface function (replace with your actual logic)
+# Define your Gradio chat interface function
 def chat_interface(message, history):
     try:
         # Process the user message and generate a response
@@ -151,8 +172,14 @@ div.svelte-rk35yg {display: none;}
 div.progress-text.svelte-z7cif2.meta-text {display: none;}
 '''
 
-gr.ChatInterface(chat_interface,
-                 css=css,
-                 description="Lily",
-                 clear_btn=None, undo_btn=None, retry_btn=None,
-                 ).launch()
+demo = gr.ChatInterface(chat_interface, 
+                        css=css,
+                        description="Lily",
+                        clear_btn=None, undo_btn=None, retry_btn=None,
+)
+
+# Add a button to save chat history
+gr.Button("Close Chat").click(fn=save_chat_history)
+
+# Launch the interface
+demo.launch()