Update app.py

app.py

@@ -16,34 +16,28 @@ def text_to_srt(text):
             continue  # Skip lines that don't match the expected format
     return srt_content
 
-def export_file(srt_content):
-    if srt_content.strip() == "":
-        return None
-    # 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):
+def convert_and_prepare_file(text_input):
     srt_content = text_to_srt(text_input)
-    file_path = export_file(srt_content)
-    return file_path
+    if not srt_content.strip():
+        return None  # No valid content to create a file
+    # Write SRT content to a temporary file
+    temp_file = NamedTemporaryFile(delete=False, suffix=".srt", mode='w+', dir="./")
+    temp_file.write(srt_content)
+    temp_file.close()
+    return temp_file.name  # Return the path for Gradio to handle
 
 with gr.Blocks() as app:
-    gr.Markdown("### Text to SRT Converter")
-    text_input = gr.TextArea(label="Enter text in the specified format")
-    export_btn = gr.Button("Export as SRT")
-    output_file = gr.File(label="Download SRT", visible=False)  # Initially hidden
+    with gr.Row():
+        text_input = gr.TextArea(label="Enter text in the specified format")
+        export_btn = gr.Button("Export as SRT")
+    output_file = gr.File(label="Download SRT", interactive=True, visible=False)
 
-    # Update to use a state variable for holding the SRT file path
-    def on_export_click(_, file_path):
+    def on_export_click(text_input):
+        file_path = convert_and_prepare_file(text_input)
         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
+            return file_path, True  # Return the path and make the file visible
+        return None, False  # If no file was created, return None and keep the file component hidden
 
-    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)
+    export_btn.click(fn=on_export_click, inputs=text_input, outputs=[output_file, output_file.visible])
 
-app.launch()
+app.launch(share=True)