Fix streaming Audio/Video Output (#9656)

* Fix * add changeset * lint * add changeset * type ignore --------- Co-authored-by: gradio-pr-bot <[email protected]> Co-authored-by: Abubakar Abid <[email protected]>
gradio-app · Oct 11, 2024 · 8f6626c · 8f6626c
1 parent b1c5a68
commit 8f6626c
Show file tree

Hide file tree

Showing 6 changed files with 10 additions and 7 deletions.
diff --git a/.changeset/silly-hairs-burn.md b/.changeset/silly-hairs-burn.md
@@ -0,0 +1,5 @@
+---
+"gradio": patch
+---
+
+feat:Fix streaming Audio/Video Output
diff --git a/gradio/blocks.py b/gradio/blocks.py
@@ -391,9 +391,7 @@ def serve_static_file(
                 return client_utils.synchronize_async(
                     processing_utils.async_move_files_to_cache, data, self
                 )
-            except (
-                AttributeError
-            ):  # Can be raised if this function is called before the Block is fully initialized.
+            except AttributeError:  # Can be raised if this function is called before the Block is fully initialized.
                 return data
 
 

diff --git a/gradio/chat_interface.py b/gradio/chat_interface.py
@@ -466,7 +466,7 @@ def _setup_stop_events(
                 None,
                 None,
                 None,
-                cancels=events_to_cancel,
+                cancels=events_to_cancel,  # type: ignore
                 show_api=False,
             )
 

diff --git a/gradio/components/audio.py b/gradio/components/audio.py
@@ -339,6 +339,7 @@ async def stream_output(
             "path": output_id,
             "is_stream": True,
             "orig_name": "audio-stream.mp3",
+            "meta": {"_type": "gradio.FileData"},
         }
         if value is None:
             return None, output_file

diff --git a/gradio/components/video.py b/gradio/components/video.py
@@ -559,6 +559,7 @@ async def stream_output(
                 # Need to set orig_name so that downloaded file has correct
                 # extension
                 "orig_name": "video-stream.mp4",
+                "meta": {"_type": "gradio.FileData"},
             }
         }
         if value is None:

diff --git a/gradio/routes.py b/gradio/routes.py
@@ -461,9 +461,7 @@ def login(form_data: OAuth2PasswordRequestForm = Depends()):
                 not callable(app.auth)
                 and username in app.auth
                 and compare_passwords_securely(password, app.auth[username])  # type: ignore
-            ) or (
-                callable(app.auth) and app.auth.__call__(username, password)
-            ):  # type: ignore
+            ) or (callable(app.auth) and app.auth.__call__(username, password)):  # type: ignore
                 token = secrets.token_urlsafe(16)
                 app.tokens[token] = username
                 response = JSONResponse(content={"success": True})