make static file route from request.url

daras_ai_v2/static_pages.py

@@ -1,4 +1,5 @@
 import io
+from starlette.responses import Response
 from static_pages.models import StaticPage
 from google.cloud import storage
 
@@ -51,12 +52,17 @@ def get_all_styles(links: list, uid: str):
     return f"<style>{styles}</style>"
 
 
-def serve(page_slug: str, file_path: str = None):
+def serve(request: Response, page_slug: str):
     static_page = StaticPage.objects.get(path=page_slug)
-
     if not static_page:
         return None
 
+    file_path = request.url.path
+    if not file_path == f"/{page_slug}/":
+        file_path = file_path[len(f"/{page_slug}/") :]
+    else:
+        file_path = None
+
     uid = static_page.uid
     bucket = gcs_bucket()
 

routers/root.py

@@ -578,23 +578,26 @@ def chat_lib_route(request: Request, integration_id: str, integration_name: str
     app,
     "/{page_slug}/",
     "/{page_slug}/{run_slug}/",
+    "/{page_slug}/{run_slug}/{path}",
     "/{page_slug}/{run_slug}-{example_id}/",
-    "/{page_slug}/{path:path}",
 )
 def recipe_page_or_handle(
     request: Request,
     page_slug: str,
     run_slug: str = None,
-    example_id: str = None,
     path: str = None,
+    example_id: str = None,
 ):
     try:
         handle = Handle.objects.get_by_name(page_slug)
     except Handle.DoesNotExist:
         try:
             import daras_ai_v2.static_pages as static_pages
 
-            html, file_path = static_pages.serve(page_slug=page_slug, file_path=path)
+            html, file_path = static_pages.serve(
+                request=request,
+                page_slug=page_slug,
+            )
 
             if file_path:
                 return RedirectResponse(file_path)