Update app.py

app.py

@@ -1,16 +1,15 @@
-import os
 import gradio as gr
-import zipfile
-import requests
-import json
-from pathlib import Path
 import logging
+import os
 import numpy as np
 from sentence_transformers import SentenceTransformer
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 import faiss
 from simple_salesforce import Salesforce
-from dotenv import load_dotenv  # Import dotenv to load environment variables
+from dotenv import load_dotenv
+import json
+import zipfile
+from pathlib import Path
 
 # Setup logging
 logging.basicConfig(level=logging.INFO)
@@ -134,8 +133,6 @@ def create_salesforce_record(query, answer, confidence_percentage, source_link):
     except Exception as e:
         # Log any error during record creation
         logger.error(f"Error creating Salesforce record: {str(e)}")
-        if 'response' in locals():
-            logger.error(f"Salesforce API Response: {str(response)}")
         return None
 
 # --- Search & Answer ---
@@ -184,14 +181,19 @@ def answer_query(query):
         logger.error(f"Error in answer_query: {str(e)}")
         return f"Error: {str(e)}", "", "", ""
 
-# --- Gradio UI ---
-def process_question(q):
+# --- Gradio Chatbot UI ---
+def process_question(q, chat_history):
     if not q.strip():
-        return "Please enter a question.", "", ""
-    return answer_query(q)
+        return chat_history + [("User", "Please enter a question.")], "", ""
+    
+    answer, confidence, source, record_id = answer_query(q)
+    chat_history.append(("User", q))
+    chat_history.append(("Bot", answer))
+    
+    return chat_history, confidence, source, record_id
 
 with gr.Blocks(title="Company Documents Q&A", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("## 📚 Company Documents Q&A System")
+    gr.Markdown("## 📚 Company Documents Q&A Chatbot")
     
     with gr.Row():
         with gr.Column(scale=3):
@@ -205,11 +207,11 @@ with gr.Blocks(title="Company Documents Q&A", theme=gr.themes.Soft()) as demo:
 
     with gr.Row():
         with gr.Column():
-            answer_out = gr.Markdown(label="Answer")
+            chat_history = gr.Chatbot(label="Chat History").style(height=400)
             conf_out = gr.Markdown(label="Confidence")
             source_out = gr.Markdown(label="Source Link")
-            record_out = gr.Markdown(label="Salesforce Record ID")  # To show the Salesforce Record ID
+            record_out = gr.Markdown(label="Salesforce Record ID")
 
-    submit_btn.click(fn=process_question, inputs=question, outputs=[answer_out, conf_out, source_out, record_out])
+    submit_btn.click(fn=process_question, inputs=[question, chat_history], outputs=[chat_history, conf_out, source_out, record_out])
 
 demo.launch(server_name="0.0.0.0", server_port=7860, share=True)