Update app.py

app.py

@@ -196,6 +196,7 @@ def process_pdf(file):
     df = chatbot.paper_df(paper_text)
     df = chatbot.calculate_embeddings(df)
     print("Done processing pdf")
+    return chatbot.embeddings
 
 def download_pdf(url):
     chatbot = Chatbot()
@@ -207,6 +208,7 @@ def download_pdf(url):
     df = chatbot.paper_df(paper_text)
     df = chatbot.calculate_embeddings(df)
     print("Done processing pdf")
+    return chatbot.embeddings
 
 def show_pdf(file_content):
     base64_pdf = base64.b64encode(file_content).decode('utf-8')
@@ -217,6 +219,7 @@ def show_pdf(file_content):
 def main():
     st.title("Research Paper Guru")
     st.subheader("Upload PDF or Enter URL")
+    embeddings = None
     col1, col2 = st.columns(2)
     with col1:
         pdf_option = st.selectbox("Choose an option:", ["Upload PDF", "Enter URL"])
@@ -226,7 +229,7 @@ def main():
             uploaded_file = st.file_uploader("Choose a PDF file", type="pdf")
             if uploaded_file is not None:
                 file_content = uploaded_file.read()
-                process_pdf(file_content)
+                embeddings = process_pdf(file_content)
                 st.success("PDF uploaded and processed successfully!")
                 show_pdf(file_content)
 
@@ -237,7 +240,7 @@ def main():
                     try:
                         r = requests.get(str(url))
                         content = r.content
-                        download_pdf(url)
+                        embeddings = download_pdf(url)
                         st.success("PDF downloaded and processed successfully!")
                         show_pdf(content)
                     except Exception as e:
@@ -247,7 +250,10 @@ def main():
         query = st.text_input("Enter your query:")
         if query:
             if st.button("Get answer"):
-                response = chatbot.reply(query)
+                if embeddings is not None:
+                    response = chatbot.reply(embeddings, query)
+                else:
+                    st.warning("Please upload a PDF or enter a URL first.")
                 st.write(response['answer'])
                 st.write("Sources:")
                 for source in response['sources']: