Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 from dotenv import load_dotenv
-import streamlit as st
 from PyPDF2 import PdfReader
 from langchain.text_splitter import CharacterTextSplitter
 from langchain.embeddings.openai import OpenAIEmbeddings
@@ -11,9 +10,10 @@ import time
 import logging
 import pdfplumber
 import os
+import docx  # Importing docx for Word document processing
 
 def process_pdf(pdf):
-    start_time = time.time()
+	start_time = time.time()
     text = ""
     with pdfplumber.open(pdf) as pdf_reader:
         for page in pdf_reader.pages:
@@ -23,11 +23,23 @@ def process_pdf(pdf):
     return text
 
 def display_chat_history():
+    if 'chat_history' not in st.session_state:
+        st.session_state.chat_history = []
     history_text = ""
     for chat in st.session_state.chat_history:
         history_text += f"Q: {chat['question']}\nA: {chat['answer']}\n{chat['time']}\n---\n"
     st.text_area("Chat History", history_text, height=300)
 
+def update_chat_history(question, answer):
+    if 'chat_history' not in st.session_state:
+        st.session_state.chat_history = []
+    st.session_state.chat_history.append({
+        "question": question,
+        "answer": answer,
+        "time": time.strftime("%Y-%m-%d %H:%M:%S")
+    })
+
+
 def read_pdf(file_path):
     with open(file_path, "rb") as file:
         pdf_reader = PdfReader(file)
@@ -52,11 +64,11 @@ def read_documents_from_directory(directory):
         elif filename.endswith(".docx"):
             combined_text += read_word(file_path)
     return combined_text
-
+	
+	
 train_directory = r'C:\Users\writa\Downloads\Crypto'
 
 def main():
-    
     load_dotenv()
     st.set_page_config(page_title="EstateSphere")
     st.header("🏢 EstateSphere")
@@ -75,13 +87,16 @@ def main():
 
     # Chat interface
     query = st.text_input("Type your question:", key="query")
+
     if query:
         with st.spinner("Finding your answer..."):
-            docs = docsearch.similarity_search(query)
-            response = chain.run(input_documents=docs, question=query)
-            # Update chat history here
-            update_chat_history(query, response)
-            display_chat_history()
+            try:
+                docs = docsearch.similarity_search(query)
+                response = chain.run(input_documents=docs, question=query)
+                update_chat_history(query, response)
+                display_chat_history()
+            except Exception as e:
+                st.error(f"An error occurred: {e}")
 
     # Help and support in the sidebar
     st.sidebar.header("Help & Support")
@@ -89,10 +104,8 @@ def main():
 
     # Footer
     st.sidebar.text("© 2024 MosaicAI")
-
-    #def update_chat_history(question, answer):
-    # Implement function to update chat history
-
+	
+	update_chat_history(question, answer)
 
 if __name__ == "__main__":
     main()