Update app.py

app.py

@@ -1,12 +1,10 @@
 import os
-import tempfile
 import streamlit as st
 from streamlit_chat import message
 from pdfquery import PDFQuery
 from dotenv import load_dotenv
 from gtts import gTTS
 import io
-import base64
 
 # Load environment variables from .env file
 load_dotenv()
@@ -26,40 +24,28 @@ def generate_tts_audio(text):
     audio_bytes.seek(0)
     return audio_bytes
 
-from datetime import datetime
-
 def play_audio(audio_bytes):
     audio_bytes.seek(0)  # Ensure the pointer is at the beginning of the IO stream
     st.audio(audio_bytes, format='audio/mp3', start_time=0)
 
-
-
 def process_input():
     if st.session_state["user_input"] and len(st.session_state["user_input"].strip()) > 0:
         user_text = st.session_state["user_input"].strip()
         with st.session_state["thinking_spinner"], st.spinner("Thinking..."):
             query_text = st.session_state["pdfquery"].ask(user_text)
-
-        # Generate and play audio for the response with autoplay
+        
         audio_response = generate_tts_audio(query_text)
         play_audio(audio_response)
-
+        
         st.session_state["messages"].append((user_text, True))
         st.session_state["messages"].append((query_text, False))
 
-def read_and_save_file():
-    st.session_state["pdfquery"].forget()  # to reset the knowledge base
-    st.session_state["messages"] = []
-    st.session_state["user_input"] = ""
-
-    for file in st.session_state["file_uploader"]:
-        with tempfile.NamedTemporaryFile(delete=False) as tf:
-            tf.write(file.getbuffer())
-            file_path = tf.name
-
-        with st.session_state["ingestion_spinner"], st.spinner(f"Ingesting {file.name}"):
-            st.session_state["pdfquery"].ingest(file_path)
-        os.remove(file_path)
+def load_and_ingest_document():
+    file_path = 'osha.pdf'  # PDF file is assumed to be in the same directory as this script
+    if os.path.exists(file_path):
+        st.session_state["pdfquery"].ingest(file_path)
+    else:
+        st.error("PDF file not found. Please ensure 'data.pdf' is in the root directory.")
 
 def is_openai_api_key_set() -> bool:
     return len(st.session_state["OPENAI_API_KEY"]) > 0
@@ -70,27 +56,14 @@ def main():
         st.session_state["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY", "")
         if is_openai_api_key_set():
             st.session_state["pdfquery"] = PDFQuery(st.session_state["OPENAI_API_KEY"])
+            load_and_ingest_document()
         else:
             st.session_state["pdfquery"] = None
 
     st.header("Wise Chat Labs - Your Ultimate Chatbot")
 
-    st.subheader("Upload document")
-    st.file_uploader(
-        "Upload document",
-        type=["pdf"],
-        key="file_uploader",
-        on_change=read_and_save_file,
-        label_visibility="collapsed",
-        accept_multiple_files=True,
-        disabled=not is_openai_api_key_set(),
-    )
-
-    st.session_state["ingestion_spinner"] = st.empty()
-
     display_messages()
     st.text_input("Message", key="user_input", disabled=not is_openai_api_key_set(), on_change=process_input)
 
-
 if __name__ == "__main__":
-    main()
+    main()