Fixed pop and add exp backoff

app.py

@@ -51,7 +51,6 @@ async def main(page: ft.Page):
     async def stop_recording_flet(e):
         record_button.disabled = False
         stop_button.disabled = True
-        play_button.disabled = False
 
         page.update()
         nonlocal audio_player
@@ -73,12 +72,13 @@ async def main(page: ft.Page):
 
         # Create audio_player using the base64 string from audio_storage
         if page.session_id in audio_storage:
-            base64_audio = audio_storage[page.session_id]
+            base64_audio = audio_storage.pop(page.session_id)
             audio_player = ft.Audio(src=f"data:audio/wav;base64,{base64_audio}")
             page.overlay.append(audio_player)
         else:
             logger.error(f"Audio data not found for session {page.session_id}")
 
+        play_button.disabled = False
         page.update()
         logger.info(page.controls)
 
@@ -283,11 +283,17 @@ async def get_audio(session_id: str):
     audio_data = audio_storage.pop(session_id)
     
     return JSONResponse(content={"audio_data": audio_data})
-
 # Add this helper function to wait for the audio data
-async def wait_for_audio_data(session_id, max_attempts=10, delay=0.5):
-    for _ in range(max_attempts):
+async def wait_for_audio_data(session_id, max_wait_time=5, initial_interval=0.01):
+    start_time = time.time()
+    interval = initial_interval
+    while time.time() - start_time < max_wait_time:
         if session_id in audio_storage:
             return
-        await asyncio.sleep(delay)
+        await asyncio.sleep(interval)
+        # Exponential backoff with a maximum interval of 0.5 seconds
+        interval = min(interval * 1.5, 0.5)
+    
+    # If we've reached this point, we've timed out
     logger.error(f"Timeout waiting for audio data for session {session_id}")
+