Update app.py

app.py

@@ -152,8 +152,8 @@ class Chatbot():
         print(sources)
         return results.head(n)
     
-    def create_prompt(self, df, user_input):
-        result = self.search_embeddings(df, user_input, n=3)
+    def create_prompt(self, embeddings, df, user_input):
+        result = self.search_embeddings(df, embeddings, user_input, n=3)
         print(result)
         prompt = """You are a large language model whose expertise is reading and and providing answers about research papers. 
         You are given a query and a series of text embeddings from a paper in order of their cosine similarity to the query.
@@ -216,41 +216,42 @@ def show_pdf(file_content):
 
 def main():
     st.title("Research Paper Guru")
-    st.subheader("Ask a question about a research paper and get an answer with sources!")
     st.subheader("Upload PDF or Enter URL")
-    
-    pdf_option = st.selectbox("Choose an option:", ["Upload PDF", "Enter URL"])
-    chatbot = Chatbot()
-
-    if pdf_option == "Upload PDF":
-        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)
-            st.success("PDF uploaded and processed successfully!")
-            show_pdf(file_content)
+    col1, col2 = st.columns(2)
+    with col1:
+        pdf_option = st.selectbox("Choose an option:", ["Upload PDF", "Enter URL"])
+        chatbot = Chatbot()
 
-    elif pdf_option == "Enter URL":
-        url = st.text_input("Enter the URL of the PDF:")
-        if url:
-            if st.button("Download and process PDF"):
-                try:
-                    r = requests.get(str(url))
-                    content = r.content
-                    download_pdf(url)
-                    st.success("PDF downloaded and processed successfully!")
-                    show_pdf(content)
-                except Exception as e:
-                    st.error(f"An error occurred while processing the PDF: {e}")
+        if pdf_option == "Upload PDF":
+            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)
+                st.success("PDF uploaded and processed successfully!")
+                show_pdf(file_content)
 
-    query = st.text_input("Enter your query:")
-    if query:
-        if st.button("Get answer"):
-            response = chatbot.reply(query)
-            st.write(response['answer'])
-            st.write("Sources:")
-            for source in response['sources']:
-                st.write(source)
+        elif pdf_option == "Enter URL":
+            url = st.text_input("Enter the URL of the PDF:")
+            if url:
+                if st.button("Download and process PDF"):
+                    try:
+                        r = requests.get(str(url))
+                        content = r.content
+                        download_pdf(url)
+                        st.success("PDF downloaded and processed successfully!")
+                        show_pdf(content)
+                    except Exception as e:
+                        st.error(f"An error occurred while processing the PDF: {e}")
+    with col2:
+        st.subheader("Ask a question about a research paper and get an answer with sources!")
+        query = st.text_input("Enter your query:")
+        if query:
+            if st.button("Get answer"):
+                response = chatbot.reply(query)
+                st.write(response['answer'])
+                st.write("Sources:")
+                for source in response['sources']:
+                    st.write(source)
 
 if __name__ == "__main__":
     main()