import gradio as gr

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(' -> ')
            srt_content += f"{i+1}\n{start.replace('.', ',')} --> {end.replace('.', ',')}\n{content.strip()}\n\n"
        except ValueError:
            continue  # Skip lines that don't match the expected format
    return srt_content

def download_srt(output):
    with open("output.srt", "w") as file:
        file.write(output)

with gr.Blocks() as app:
    gr.Markdown("### Text to SRT Converter")
    text_input = gr.TextArea(label="Enter text in the specified format")
    output = gr.TextArea(label="SRT Output")
    download_button = gr.Button(text="Download SRT", icon="download", onclick=download_srt, inputs=output)

    text_input.change(fn=text_to_srt, inputs=text_input, outputs=output)

app.launch()