Update app.py

app.py

@@ -3,6 +3,7 @@ import os
 
 HF_TOKEN = os.getenv('HW_Token')
 hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "save_audio")
+
 import gradio as gr
 import os
 
@@ -47,23 +48,27 @@ def save_audio_text(audio, text):
     # Debugging to print out the structure of the audio variable
     print("Received audio data:", audio)
 
-    # Check if audio is a dictionary and contains 'data'
-    if isinstance(audio, dict) and 'data' in audio:
-        audio_data = audio['data']
-        audio_path = f"/tmp/recordings/line_{reader.current_index}.wav"
-        text_path = f"/tmp/recordings/line_{reader.current_index}.txt"
-        
-        with open(audio_path, "wb") as f:
-            f.write(audio_data)
+    try:
+        # Check if audio is a dictionary and contains 'data'
+        if isinstance(audio, dict) and 'data' in audio:
+            audio_data = audio['data']
+            audio_path = f"/tmp/recordings/line_{reader.current_index}.wav"
+            text_path = f"/tmp/recordings/line_{reader.current_index}.txt"
+            
+            with open(audio_path, "wb") as f:
+                f.write(audio_data)
+                
+            with open(text_path, "w") as f:
+                f.write(text)
             
-        with open(text_path, "w") as f:
-            f.write(text)
-        
-        # Move to the next line after saving
-        next_line = reader.forward_line()
-        return next_line
-    else:
-        return "Audio data is not in the expected format."
+            # Move to the next line after saving
+            next_line = reader.forward_line()
+            return next_line
+        else:
+            return "Audio data is not in the expected format."
+    except Exception as e:
+        print(f"Error saving audio: {e}")
+        return "Error saving audio."
 
 # Define the Gradio interface
 with gr.Blocks() as demo: