Update app.py

app.py

@@ -2,6 +2,9 @@ import gradio as gr
 import os
 from videocr import save_subtitles_to_file
 
+# Define path for demo video
+DEMO_VIDEO_PATH = "/data/demo.mp4"
+
 # Function to list files in /data directory
 def list_files():
     files = os.listdir('/data')
@@ -9,19 +12,22 @@ def list_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):
+def run_video_ocr(use_demo_video, 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:
         # Ensure the output directory exists
         persistent_dir = '/data'
         if not os.path.exists(persistent_dir):
             os.makedirs(persistent_dir)
-        
+
+        # Check whether to use demo video or uploaded video
+        video_path = DEMO_VIDEO_PATH if use_demo_video else input_video
+
         # Define full path for the output file
         output_path = os.path.join(persistent_dir, output_file_name)
 
         # Save the subtitles to file
         save_subtitles_to_file(
-            input_video,       # The uploaded video file path
+            video_path,       # Use demo video or uploaded video path
             output_path,       # Output .srt file path
             lang=language_code,
             use_gpu=use_gpu,
@@ -44,7 +50,10 @@ def run_video_ocr(input_video, output_file_name, language_code, use_gpu, start_t
 def video_ocr_interface():
     with gr.Blocks() as demo:
         with gr.Row():
-            input_video = gr.File(label="Upload Video", type="filepath")  # Changed to 'filepath'
+            # Radio button for video selection
+            use_demo_video = gr.Radio(choices=["Upload Video", "Use Demo Video"], label="Choose Video", value="Upload Video")
+
+            input_video = gr.File(label="Upload Video", type="filepath", visible=True)
             output_file_name = gr.Textbox(label="Output File Name (.srt)", value="subtitle.srt")
             language_code = gr.Textbox(label="Language Code", value="ch")
             use_gpu = gr.Checkbox(label="Use GPU", value=True)
@@ -74,10 +83,16 @@ def video_ocr_interface():
         # Refresh button to update the list of files
         refresh_btn = gr.Button("Refresh File List")
         
+        # Define the visibility toggle for upload based on the radio button
+        def toggle_video_input(selected_option):
+            return {"visible": selected_option == "Upload Video"}
+
+        use_demo_video.change(fn=toggle_video_input, inputs=use_demo_video, outputs=input_video)
+
         # 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],
+            inputs=[use_demo_video, 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
         )