Sajjo's picture
Update app.py
258d1de verified
raw
history blame
2.47 kB
import gradio as gr
import os
class TextFileReader:
def __init__(self):
self.lines = []
self.current_index = 0
def read_lines(self, file):
self.lines = file.decode('utf-8').splitlines()
self.current_index = 0
return self.get_current_line()
def get_current_line(self):
if 0 <= self.current_index < len(self.lines):
return self.lines[self.current_index]
else:
return "End of file reached."
def forward_line(self):
self.current_index = min(self.current_index + 1, len(self.lines) - 1)
return self.get_current_line()
def backward_line(self):
self.current_index = max(self.current_index - 1, 0)
return self.get_current_line()
# Initialize the text reader
reader = TextFileReader()
# Define a function to save the text lines to a file
def save_text_lines(file):
lines = reader.read_lines(file)
with open("text_lines.txt", "w") as f:
f.write("\n".join(reader.lines))
return lines
# Define a function to save the audio file
def save_audio_file(audio):
with open("recorded_audio.wav", "wb") as f:
f.write(audio['data'])
return audio
# Define the Gradio interface
with gr.Blocks() as demo:
with gr.Row():
file_upload = gr.File(label="Upload a text file", type="binary")
generate_button = gr.Button("Generate Lines")
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(source="microphone", type="file", label="Record Audio")
audio_output = gr.Audio(label="Recorded Audio")
def display_audio(audio):
save_audio_file(audio) # Save the audio file
return audio
audio_record.change(fn=display_audio, inputs=audio_record, outputs=audio_output)
demo.launch()