Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import time
 from dotenv import load_dotenv
 import pickle
 from huggingface_hub import Repository
@@ -135,12 +136,25 @@ def main():
             loading_message = st.empty()
             loading_message.text('Bot is thinking...')
 
+            # Start timing
+            start_time = time.time()
+
             VectorStore = load_pdf(pdf_path)
             chain = load_chatbot()
             docs = VectorStore.similarity_search(query=query, k=3)
             with get_openai_callback() as cb:
                 response = chain.run(input_documents=docs, question=query)
 
+            # Stop timing
+            end_time = time.time()
+
+            # Calculate duration
+            duration = end_time - start_time
+
+            # You can use Streamlit's text function to display the timing
+            st.text(f"Response time: {duration:.2f} seconds")
+
+
             st.session_state['chat_history'].append(("Bot", response, "new"))
 
             # Display new messages at the bottom