Update app.py

app.py

@@ -65,7 +65,7 @@ def extract_metadata(pdf_path):
 
     return title, author, email_str, affiliation_str
 
-# ----------------- PDF Selection -----------------
+# ----------------- Step 1: Choose PDF Source -----------------
 pdf_source = st.radio("Upload or provide a link to a PDF:", ["Upload a PDF file", "Enter a PDF URL"], index=0, horizontal=True)
 
 if pdf_source == "Upload a PDF file":
@@ -78,6 +78,26 @@ if pdf_source == "Upload a PDF file":
         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:")
+    if pdf_url and not st.session_state.pdf_loaded:
+        with st.spinner("🔄 Downloading PDF..."):
+            try:
+                response = requests.get(pdf_url)
+                if response.status_code == 200:
+                    st.session_state.pdf_path = "/mnt/data/temp.pdf"
+                    with open(st.session_state.pdf_path, "wb") as f:
+                        f.write(response.content)
+                    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}")
+
+
 # ----------------- Process PDF -----------------
 if not st.session_state.pdf_loaded and "pdf_path" in st.session_state:
     with st.spinner("🔄 Processing document... Please wait."):