Update app.py

app.py

@@ -2,6 +2,12 @@ import gradio as gr
 import os
 from videocr import save_subtitles_to_file
 
+# Function to list files in /data directory
+def list_files():
+    files = os.listdir('/data')
+    file_paths = [f"/data/{file}" for file in files]
+    return file_paths
+
 # Define OCR function
 def run_video_ocr(input_video, output_file_name, language_code, use_gpu, start_time, end_time, confidence_threshold, similarity_threshold, frames_to_skip, crop_x, crop_y, crop_width, crop_height):
     try:
@@ -62,13 +68,22 @@ def video_ocr_interface():
 
         output_label = gr.Textbox(label="Status", interactive=False)
 
+        # Define the file explorer component
+        file_list = gr.File(label="Download Extracted .srt Files", interactive=True)
+
+        # Refresh button to update the list of files
+        refresh_btn = gr.Button("Refresh File List")
+        
         # Define what happens when the button is clicked
         submit_btn.click(
             fn=run_video_ocr,
             inputs=[input_video, output_file_name, language_code, use_gpu, start_time, end_time, confidence_threshold, similarity_threshold, frames_to_skip, crop_x, crop_y, crop_width, crop_height],
             outputs=output_label
         )
-    
+        
+        # Refresh the file explorer when the refresh button is clicked
+        refresh_btn.click(fn=list_files, inputs=[], outputs=file_list)
+
     return demo
 
 # Launch the Gradio interface