Update app.py

app.py

@@ -43,35 +43,47 @@ st.title("Blah-2")
 # Step 1: Choose PDF Source
 pdf_source = st.radio("Upload or provide a link to a PDF:", ["Enter a PDF URL", "Upload a PDF file"], index=0, horizontal=True)
 
-if pdf_source == "Upload a PDF file":
-    uploaded_file = st.file_uploader("Upload your PDF file", type="pdf")
-    if uploaded_file:
-        st.session_state.pdf_path = "temp.pdf"
-        with open(st.session_state.pdf_path, "wb") as f:
-            f.write(uploaded_file.getbuffer())
-        st.session_state.pdf_loaded = False
-        st.session_state.chunked = False
-        st.session_state.vector_created = False
-
-elif pdf_source == "Enter a PDF URL":
-    pdf_url = st.text_input("Enter PDF URL:", value = "https://arxiv.org/pdf/2406.06998")
-    if pdf_url and not st.session_state.pdf_path:
+# Function to download and process the PDF
+def download_pdf():
+    if st.session_state.pdf_url and not st.session_state.pdf_path:
         with st.spinner("Downloading PDF..."):
             try:
-                response = requests.get(pdf_url)
+                response = requests.get(st.session_state.pdf_url)
                 if response.status_code == 200:
                     st.session_state.pdf_path = "temp.pdf"
                     with open(st.session_state.pdf_path, "wb") as f:
                         f.write(response.content)
+
+                    # Reset processing state
                     st.session_state.pdf_loaded = False
                     st.session_state.chunked = False
                     st.session_state.vector_created = False
+
                     st.success("✅ PDF Downloaded Successfully!")
                 else:
                     st.error("❌ Failed to download PDF. Check the URL.")
             except Exception as e:
                 st.error(f"❌ Error downloading PDF: {e}")
 
+if pdf_source == "Upload a PDF file":
+    uploaded_file = st.file_uploader("Upload your PDF file", type="pdf")
+    if uploaded_file:
+        st.session_state.pdf_path = "temp.pdf"
+        with open(st.session_state.pdf_path, "wb") as f:
+            f.write(uploaded_file.getbuffer())
+        st.session_state.pdf_loaded = False
+        st.session_state.chunked = False
+        st.session_state.vector_created = False
+
+elif pdf_source == "Enter a PDF URL":
+    # ✅ Text input with Enter support
+    st.text_input("Enter PDF URL:", value="https://arxiv.org/pdf/2406.06998", key="pdf_url", on_change=download_pdf)
+
+    # ✅ Button support
+    if st.button("Load PDF"):
+        download_pdf()
+
+
 # Step 2: Load & Process PDF (Only Once)
 if st.session_state.pdf_path and not st.session_state.pdf_loaded:
     with st.spinner("Loading PDF..."):