Update app.py

app.py

@@ -17,12 +17,25 @@ GOOGLE_DRIVE_LINK = "https://drive.google.com/drive/folders/1gVdnV1za1thrVnH4LAL
 
 # Function to download the PDF from Google Drive
 def download_pdf():
-    file_id = GOOGLE_DRIVE_LINK.split("/d/")[1].split("/view")[0]
-    url = f"https://drive.google.com/uc?id={1gVdnV1za1thrVnH4LALbDzPtcvKr0z4u}&export=download"
-    response = requests.get(url)
-    with open("document.pdf", "wb") as f:
-        f.write(response.content)
-    return "document.pdf"
+    try:
+        if "/d/" in GOOGLE_DRIVE_LINK:
+            file_id = GOOGLE_DRIVE_LINK.split("/d/")[1].split("/view")[0]
+        elif "id=" in GOOGLE_DRIVE_LINK:
+            file_id = GOOGLE_DRIVE_LINK.split("id=")[1].split("&")[0]
+        else:
+            raise ValueError("Invalid Google Drive link format.")
+
+        url = f"https://drive.google.com/uc?id={makhdoomnaeem@gmail.com}&export=download"
+        response = requests.get(url)
+        response.raise_for_status()  # Raise an error for bad responses
+
+        with open("document.pdf", "wb") as f:
+            f.write(response.content)
+        return "document.pdf"
+    except Exception as e:
+        st.error(f"Failed to download PDF: {e}")
+        return None
+
 
 # Function to extract text from PDF
 def extract_text_from_pdf(pdf_file):