|
import gradio as gr |
|
import os |
|
|
|
def text_to_srt(text): |
|
lines = text.split('\n') |
|
srt_content = "" |
|
for i, line in enumerate(lines): |
|
if line.strip() == "": |
|
continue |
|
try: |
|
times, content = line.split(']', 1) |
|
start, end = times[1:].split(' -> ') |
|
|
|
if start.count(":") == 1: |
|
start = "00:" + start |
|
if end.count(":") == 1: |
|
end = "00:" + end |
|
|
|
srt_content += f"{i+1}\n{start.replace('.', ',')} --> {end.replace('.', ',')}\n{content.strip()}\n\n" |
|
except ValueError: |
|
continue |
|
|
|
temp_file_path = '/tmp/output.srt' |
|
with open(temp_file_path, 'w', encoding='utf-8') as file: |
|
file.write(srt_content) |
|
return temp_file_path |
|
|
|
with gr.Blocks() as app: |
|
gr.Markdown(""" |
|
# Text to SRT Converter |
|
|
|
Convert your timestamped text into SRT format easily. Please format your input text as follows (this is format for [Whisper JAX](https://huggingface.co/spaces/sanchit-gandhi/whisper-jax)): |
|
|
|
``` |
|
[00:00:00 -> 00:00:05] Line of dialogue 1 |
|
[00:00:06 -> 00:00:10] Line of dialogue 2 |
|
``` |
|
|
|
Each line should include a start and end time, followed by the dialogue, separated by ' -> '. Times should be in HH:MM:SS format, or MM:SS if under an hour. |
|
|
|
For web use, please visit [this space](https://huggingface.co/spaces/Lenylvt/Text_to_SRT). |
|
""") |
|
text_input = gr.TextArea(label="Enter text") |
|
output = gr.File(label="Download SRT File", file_count="single") |
|
text_input.change(fn=text_to_srt, inputs=text_input, outputs=output) |
|
|
|
if __name__ == "__main__": |
|
app.launch() |
|
|