WIP: Optimize /file route (#7416)

* Improve File route * file route fix * add changeset * add changeset --------- Co-authored-by: gradio-pr-bot <[email protected]>
gradio-app · Feb 15, 2024 · c88290d · c88290d
1 parent a6371bb
commit c88290d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/.changeset/evil-lamps-refuse.md b/.changeset/evil-lamps-refuse.md
@@ -0,0 +1,5 @@
+---
+"gradio": patch
+---
+
+feat:WIP: Optimize /file route
diff --git a/gradio/routes.py b/gradio/routes.py
@@ -439,7 +439,11 @@ async def file(path_or_url: str, request: fastapi.Request):
             if in_blocklist or is_dir:
                 raise HTTPException(403, f"File not allowed: {path_or_url}.")
 
-            created_by_app = str(abs_path) in set().union(*blocks.temp_file_sets)
+            created_by_app = False
+            for temp_file_set in blocks.temp_file_sets:
+                if abs_path in temp_file_set:
+                    created_by_app = True
+                    break
             in_allowlist = any(
                 utils.is_in_or_equal(abs_path, allowed_path)
                 for allowed_path in blocks.allowed_paths