import gradio as gr
import base64

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(srt_content):
    b64 = base64.b64encode(srt_content.encode()).decode()
    return f'<a download="output.srt" href="data:text/srt;base64,{b64}">Download SRT</a>'

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(label="Download SRT")
    text_input.change(fn=text_to_srt, inputs=text_input, outputs=output)
    download_button.click(fn=download_srt, inputs=output, outputs=download_button)

app.launch()