gsavin commited on
Commit
0005681
·
1 Parent(s): 17ab163

fix: remove check for disconnected request

Browse files
gradio-legacy/src/audio/audio_generator.py CHANGED
@@ -98,7 +98,7 @@ async def cleanup_music_session(user_hash: str):
98
  del sessions[user_hash]
99
 
100
 
101
- async def update_audio(user_hash: str, request: gr.Request):
102
  """
103
  Continuously stream audio from the queue as WAV bytes, and clean up
104
  when the user disconnects.
@@ -109,10 +109,6 @@ async def update_audio(user_hash: str, request: gr.Request):
109
  logger.info(f"Starting audio update loop for user hash: {user_hash}")
110
  try:
111
  while True:
112
- if await request.request.is_disconnected():
113
- logger.info(f"Client disconnected for user hash {user_hash}.")
114
- break
115
-
116
  if user_hash not in sessions:
117
  await asyncio.sleep(0.5)
118
  continue
 
98
  del sessions[user_hash]
99
 
100
 
101
+ async def update_audio(user_hash: str):
102
  """
103
  Continuously stream audio from the queue as WAV bytes, and clean up
104
  when the user disconnects.
 
109
  logger.info(f"Starting audio update loop for user hash: {user_hash}")
110
  try:
111
  while True:
 
 
 
 
112
  if user_hash not in sessions:
113
  await asyncio.sleep(0.5)
114
  continue