Update app.py

app.py

@@ -1,7 +1,9 @@
-import gradio as gr
-import os
+
 HF_TOKEN = os.getenv('HW_Token')
 hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "save_audio")
+import gradio as gr
+import os
+
 class TextFileReader:
     def __init__(self):
         self.lines = []
@@ -35,12 +37,22 @@ def save_text_lines(file):
         f.write("\n".join(reader.lines))
     return lines
 
-# Define a function to save the audio file
-def save_audio_file(audio):
-    audio_path = "/tmp/recorded_audio.wav"
+# Define a function to save the audio file and corresponding text
+def save_audio_text(audio, text):
+    if not os.path.exists("/tmp/recordings"):
+        os.makedirs("/tmp/recordings")
+    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'])
-    return audio_path
+        
+    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
 
 # Define the Gradio interface
 with gr.Blocks() as demo:
@@ -50,33 +62,17 @@ with gr.Blocks() as demo:
     
     current_line = gr.Textbox(label="Current Line")
     
-    with gr.Row():
-        prev_button = gr.Button("Previous")
-        next_button = gr.Button("Next")
-
     def update_output(file):
         lines = reader.read_lines(file)
         save_text_lines(file)  # Save the text lines to a file
         return lines
 
-    def forward_output():
-        return reader.forward_line()
-
-    def backward_output():
-        return reader.backward_line()
-
     generate_button.click(fn=update_output, inputs=file_upload, outputs=current_line)
-    prev_button.click(fn=backward_output, inputs=None, outputs=current_line)
-    next_button.click(fn=forward_output, inputs=None, outputs=current_line)
     
     with gr.Row():
-        audio_record = gr.Audio(sources=["microphone","upload"], type="filepath")
-        audio_output = gr.Audio(label="Recorded Audio")
-
-    def display_audio(audio):
-        audio_path = save_audio_file(audio)  # Save the audio file
-        return audio_path
+        audio_record = gr.Audio(source="microphone", type="file", label="Record Audio")
+        save_button = gr.Button("Save Audio and Next Line")
 
-    audio_record.change(fn=display_audio, inputs=audio_record, outputs=audio_output)
+    save_button.click(fn=save_audio_text, inputs=[audio_record, current_line], outputs=current_line)
 
 demo.launch()