Update app.py

download the video with the original subtitles

app.py

@@ -61,7 +61,7 @@ def add_subtitle(video):
                 transcriptions = list(executor.map(transcribe_audio, chunks))
 
             full_transcription = " ".join(transcriptions)
-            subtitle_storage["original"] = full_transcription  # Store the original subtitle
+            subtitle_storage["original"] = full_transcription  # Store the original subtitle (English or other)
             subtitle_storage["video_path"] = video_path  # Store the video path
 
         return f"Subtitle added: {full_transcription[:100]}..."  # Display first 100 characters
@@ -119,14 +119,14 @@ def download_word():
     except Exception as e:
         return f"Error in saving subtitles as Word: {e}"
 
-def download_video():
+def download_original_subtitled_video():
+    """Download video with original subtitles (no translation)."""
     try:
-        # Add subtitles to the video
+        # Add original subtitles to the video
         original_subtitle = subtitle_storage.get("original")
-        translated_subtitle = subtitle_storage.get("translated")
 
-        if not original_subtitle or not translated_subtitle:
-            return "No subtitles to overlay on video!"
+        if not original_subtitle:
+            return "No original subtitles available!"
 
         video_path = subtitle_storage.get("video_path")
         video = mp.VideoFileClip(video_path)
@@ -134,13 +134,13 @@ def download_video():
         # Function to generate subtitle text
         generator = lambda txt: TextClip(txt, font='Arial', fontsize=24, color='white')
 
-        # Generate subtitles (each subtitle will last for 2.5 seconds with millisecond precision)
+        # Generate original subtitles with start_time, end_time, and text
         subs = []
         subtitle_length = 2.5  # seconds each subtitle will be displayed
-        for i in range(0, len(translated_subtitle), 40):  # Reduced chunk size for better handling
+        for i in range(0, len(original_subtitle), 40):  # Adjusted chunk size
             start_time = (i // 40) * subtitle_length
             end_time = start_time + subtitle_length
-            subtitle_text = translated_subtitle[i:i + 40].strip()  # Get the next 40 characters
+            subtitle_text = original_subtitle[i:i + 40].strip()  # Get the next 40 characters
 
             if subtitle_text:
                 subs.append((start_time, end_time, subtitle_text))  # Create a tuple for start time, end time, and text
@@ -154,14 +154,14 @@ def download_video():
         # Overlay subtitles on video
         subtitled_video = mp.CompositeVideoClip([video, subtitles.set_position(('center', 'bottom'))])
 
-        output_video_path = "subtitled_video.mp4"
+        output_video_path = "original_subtitled_video.mp4"
         subtitled_video.write_videofile(output_video_path, codec='libx264')  # Use libx264 for better compatibility
 
         # Return the file path for download
-        return output_video_path  # Gradio will handle this as a downloadable file
+        return output_video_path
 
     except Exception as e:
-        return f"Error in generating subtitled video: {e}"
+        return f"Error in generating original subtitled video: {e}"
 
 # Gradio UI Interface
 with gr.Blocks() as demo:
@@ -198,12 +198,12 @@ with gr.Blocks() as demo:
 
     download_button.click(download_word, inputs=None, outputs=download_status)
 
-    # Download Subtitled Video
+    # Download Original Subtitled Video (no translation)
     with gr.Row():
-        download_video_button = gr.Button("Download Subtitled Video")
-        download_video_status = gr.File(label="Download Subtitled Video")  # File output for Video download
+        download_original_video_button = gr.Button("Download Original Subtitled Video")
+        download_original_video_status = gr.File(label="Download Original Subtitled Video")  # File output for original subtitled video
 
-    download_video_button.click(download_video, inputs=None, outputs=download_video_status)
+    download_original_video_button.click(download_original_subtitled_video, inputs=None, outputs=download_original_video_status)
 
 # Launch the Gradio app
 demo.launch()