Added retry logic in fetch audio

app.py

@@ -160,11 +160,19 @@ def send_task(payload):
         return response["text"]
 
 def fetch_audio():
-    filepath = random.choice(random_paths)
-    with fs.open(f"{ARENA_PATH}/{filepath}", 'rb') as f:
-        audio,sr = torchaudio.load(f)
-        audio = audio.numpy()
-        return audio,sr,filepath
+    num_tries = 3
+    iter_count = 0
+    while iter_count <= num_tries:
+        try:
+            filepath = random.choice(random_paths)
+            with fs.open(f"{ARENA_PATH}/{filepath}", 'rb') as f:
+                audio,sr = torchaudio.load(f)
+                audio = audio.numpy()
+                return audio,sr,filepath
+        except Exception:
+            iter_count += 1
+
+    return None,None,None
 
 def encode_audio_array(audio_array):
     buffer = io.BytesIO()
@@ -391,11 +399,14 @@ def on_random_click():
     reset_state()
     with st.spinner("Fetching random audio... please wait"):
         array, sampling_rate, filepath = fetch_audio()
-        st.session_state.audio = {"data":array,"sample_rate":sampling_rate,"format":"audio/wav"}
-        st.session_state.has_audio = True
-        st.session_state.current_audio_type = "random"
-        st.session_state.audio_path = filepath
-        st.session_state.option_selected = None
+        if filepath is None:
+            st.error("Error in fetching random audio please try uploading an audio or using the mic")
+        else:
+            st.session_state.audio = {"data":array,"sample_rate":sampling_rate,"format":"audio/wav"}
+            st.session_state.has_audio = True
+            st.session_state.current_audio_type = "random"
+            st.session_state.audio_path = filepath
+            st.session_state.option_selected = None
 
 def on_reset_click():
     reset_state()