Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -54,8 +54,10 @@ if pdf_source == "Upload a PDF file":
|
|
| 54 |
st.session_state.vector_created = False
|
| 55 |
|
| 56 |
elif pdf_source == "Enter a PDF URL":
|
| 57 |
-
pdf_url = st.text_input("Enter PDF URL:", value
|
| 58 |
-
|
|
|
|
|
|
|
| 59 |
with st.spinner("Downloading PDF..."):
|
| 60 |
try:
|
| 61 |
response = requests.get(pdf_url)
|
|
@@ -63,9 +65,12 @@ elif pdf_source == "Enter a PDF URL":
|
|
| 63 |
st.session_state.pdf_path = "temp.pdf"
|
| 64 |
with open(st.session_state.pdf_path, "wb") as f:
|
| 65 |
f.write(response.content)
|
|
|
|
|
|
|
| 66 |
st.session_state.pdf_loaded = False
|
| 67 |
st.session_state.chunked = False
|
| 68 |
st.session_state.vector_created = False
|
|
|
|
| 69 |
st.success("β
PDF Downloaded Successfully!")
|
| 70 |
else:
|
| 71 |
st.error("β Failed to download PDF. Check the URL.")
|
|
|
|
| 54 |
st.session_state.vector_created = False
|
| 55 |
|
| 56 |
elif pdf_source == "Enter a PDF URL":
|
| 57 |
+
pdf_url = st.text_input("Enter PDF URL:", value="https://arxiv.org/pdf/2406.06998", key="pdf_url")
|
| 58 |
+
|
| 59 |
+
# Button to manually trigger the download
|
| 60 |
+
if st.button("Download and Process PDF") or pdf_url.strip() != "" and st.session_state.get("pdf_path") is None:
|
| 61 |
with st.spinner("Downloading PDF..."):
|
| 62 |
try:
|
| 63 |
response = requests.get(pdf_url)
|
|
|
|
| 65 |
st.session_state.pdf_path = "temp.pdf"
|
| 66 |
with open(st.session_state.pdf_path, "wb") as f:
|
| 67 |
f.write(response.content)
|
| 68 |
+
|
| 69 |
+
# Reset states
|
| 70 |
st.session_state.pdf_loaded = False
|
| 71 |
st.session_state.chunked = False
|
| 72 |
st.session_state.vector_created = False
|
| 73 |
+
|
| 74 |
st.success("β
PDF Downloaded Successfully!")
|
| 75 |
else:
|
| 76 |
st.error("β Failed to download PDF. Check the URL.")
|