mirror/gradio
2
0
Fork 0
mirror of https://github.com/gradio-app/gradio.git synced 2025-04-06 12:30:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove session if browser closed on mobile (#6241)

Browse Source 
* changes

* changes

* add changeset

---------

Co-authored-by: gradio-pr-bot <gradio-pr-bot@users.noreply.github.com>
This commit is contained in:
aliabid94 2023-11-02 11:59:29 -07:00 committed by GitHub
parent d7a1a65590
commit 61c155e9ba
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 29 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
.changeset/cute-poems-cover.md Normal file
View File

@ -0,0 +1,5 @@
---
"gradio": minor
---
feat:Remove session if browser closed on mobile

44
gradio/routes.py
View File

@ -615,28 +615,32 @@ class App(FastAPI):
await blocks._queue.send_estimation(event, estimation, rank)
async def sse_stream(request: fastapi.Request):
last_heartbeat = time.perf_counter()
while True:
if await request.is_disconnected():
await blocks._queue.clean_event(event)
if not event.alive:
return
heartbeat_rate = 15
check_rate = 0.05
message = None
try:
message = event.message_queue.get_nowait()
if message is None: # end of stream marker
try:
last_heartbeat = time.perf_counter()
while True:
if await request.is_disconnected():
await blocks._queue.clean_event(event)
if not event.alive:
return
except EmptyQueue:
await asyncio.sleep(check_rate)
if time.perf_counter() - last_heartbeat > heartbeat_rate:
message = {"msg": "heartbeat"}
last_heartbeat = time.time()
if message:
yield f"data: {json.dumps(message)}\n\n"
heartbeat_rate = 15
check_rate = 0.05
message = None
try:
message = event.message_queue.get_nowait()
if message is None: # end of stream marker
return
except EmptyQueue:
await asyncio.sleep(check_rate)
if time.perf_counter() - last_heartbeat > heartbeat_rate:
message = {"msg": "heartbeat"}
last_heartbeat = time.time()
if message:
yield f"data: {json.dumps(message)}\n\n"
except asyncio.CancelledError as e:
await blocks._queue.clean_event(event)
raise e
return StreamingResponse(
sse_stream(request),