Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
-import io
+import os
+from tempfile import NamedTemporaryFile
 
 def text_to_srt(text):
     lines = text.split('\n')
@@ -18,14 +19,31 @@ def text_to_srt(text):
 def export_file(srt_content):
     if srt_content.strip() == "":
         return None
-    return gr.File(content=srt_content.encode('utf-8'), file_name="output.srt")
+    # Write SRT content to a temporary file
+    with NamedTemporaryFile(delete=False, suffix=".srt", mode='w+', dir="./") as tmp_file:
+        tmp_file.write(srt_content)
+        return tmp_file.name  # Return the path for Gradio to handle
+
+# Use a stateful text component to hold the path to the temporary SRT file
+def update_output(text_input):
+    srt_content = text_to_srt(text_input)
+    file_path = export_file(srt_content)
+    return file_path
 
 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")
-    export_button = gr.Button("Export as SRT")
-    text_input.change(fn=text_to_srt, inputs=text_input, outputs=output)
-    export_button.click(fn=export_file, inputs=output, outputs=gr.File(label="Download SRT"))
+    export_btn = gr.Button("Export as SRT")
+    output_file = gr.File(label="Download SRT", visible=False)  # Initially hidden
+
+    # Update to use a state variable for holding the SRT file path
+    def on_export_click(_, file_path):
+        if file_path:
+            output_file.update(value=file_path, visible=True)  # Make the download link visible
+        else:
+            output_file.update(visible=False)  # Hide if no file
+
+    text_input.change(fn=update_output, inputs=text_input, outputs=output_file, show_progress=True)
+    export_btn.click(fn=on_export_click, inputs=[export_btn, text_input], outputs=output_file)
 
 app.launch()