Upload front_end.py

front_end.py

@@ -0,0 +1,67 @@
+# import streamlit as st
+# import requests
+# import json
+
+# st.set_page_config(page_title="AskPDF", page_icon="📄")
+
+
+
+# st.title("📄 AskPDF")
+# uploaded_file = st.file_uploader("Upload a PDF file", type="pdf")
+# st.session_state=False
+# if uploaded_file is not None:
+#     uploaded_file=None
+#     upload_file = {'file':uploaded_file}
+#     pdf = requests.post(url='http://127.0.0.1:8000/upload_pdf', files=upload_file)
+#     st.session_state=True
+# else:
+#     st.write("Waiting for vectors to be uploaded...")
+
+# if st.session_state!=False and pdf.status_code==200:
+#     st.title("🤖 Ask a Question about the PDF Content")
+#     add_selectbox = st.sidebar.selectbox("Select Model", 
+#                                             ("Default", "deepset/roberta-base-squad2", "llama"))
+#     question = st.text_input("Ask a question about the PDF content:")
+#     url = f"http://127.0.0.1:8000/ask_question/?question={question}&model={add_selectbox}"
+#     payload = {}
+#     headers = {
+#     'accept': 'application/json'
+#     }
+#     response = requests.request("POST", url, headers=headers, data=payload)
+#     st.write(response)
+#     print(response.text)
+
+
+import streamlit as st
+import requests
+
+st.set_page_config(page_title="ChatPDF", page_icon="📄")
+
+st.title("📄 ChatPDF")
+uploaded_file = st.file_uploader("Upload a PDF file", type="pdf")
+
+if 'pdf_processed' not in st.session_state:
+    st.session_state['pdf_processed'] = False
+
+if uploaded_file is not None and not st.session_state['pdf_processed']:
+    upload_file = {'file': uploaded_file}
+    response = requests.post(url='http://127.0.0.1:8000/upload_pdf', files=upload_file)
+    if response.status_code == 200:
+        st.session_state['pdf_processed'] = True
+        st.write("PDF processed and vectors upserted!")
+    else:
+        st.write("Failed to process PDF")
+
+if st.session_state['pdf_processed']:
+    st.title("🤖 Ask a Question about the PDF Content")
+    add_selectbox = st.sidebar.selectbox("Select Model", ("Default", "deepset/roberta-base-squad2", "llama"))
+    question = st.text_input("Ask a question about the PDF content:")
+    if question:
+        url = f"http://127.0.0.1:8000/ask_question/?question={question}&model={add_selectbox}"
+        response = requests.post(url)
+        if response.status_code == 200:
+            st.write(response.json()['answer'])
+        else:
+            st.write("Failed to get an answer")
+
+## streamlit run C:\Yash\Streamlit\front_end.py