add google slides - fragments

daras_ai_v2/gdrive_downloader.py

@@ -9,6 +9,8 @@
 def is_gdrive_url(f: furl) -> bool:
     return f.host in ["drive.google.com", "docs.google.com"]
 
+def is_gdrive_presentation_url(f: furl) -> bool:
+    return f.host in ["drive.google.com", "docs.google.com"] and "presentation" in f.path.segments
 
 def url_to_gdrive_file_id(f: furl) -> str:
     # extract google drive file ID

daras_ai_v2/vector_search.py

@@ -52,6 +52,7 @@
 from daras_ai_v2.gdrive_downloader import (
     gdrive_download,
     is_gdrive_url,
+    is_gdrive_presentation_url,
     url_to_gdrive_file_id,
     gdrive_metadata,
 )
@@ -972,6 +973,11 @@ def get_pdf_num_pages(f_bytes: bytes) -> int:
 
 
 def add_page_number_to_pdf(url: str | furl, page_num: int) -> furl:
+
+    # if it's a google drive presentation, add the slide number to the fragment
+    if is_gdrive_presentation_url(furl(url)):      
+        return furl(url).set(fragment_args={"slide": page_num} if page_num else {})
+
     return furl(url).set(fragment_args={"page": page_num} if page_num else {})