Update app.py

app.py

@@ -281,7 +281,7 @@ def generate_subtitles(audio_file_path, prompt, language, auto_detect_language):
 
     # If there's an error during file check
     if error_message:
-        return error_message, None, None, None # Return None for the download button
+        return error_message, None, None, None  # Return None for the download button
 
     with open(processed_path, "rb") as file:
         transcription_json = client.audio.transcriptions.create(
@@ -298,7 +298,8 @@ def generate_subtitles(audio_file_path, prompt, language, auto_detect_language):
     transcription_text = transcription_json['text']
 
     srt_content = create_srt_from_text(transcription_text)
-    
+
+    # Create temporary file
     with tempfile.NamedTemporaryFile(mode="w", delete=False) as temp_srt_file:
         temp_srt_path = temp_srt_file.name
         temp_srt_file.write(srt_content)
@@ -306,35 +307,15 @@ def generate_subtitles(audio_file_path, prompt, language, auto_detect_language):
     # Generate subtitles and add to video if MP4
     if audio_file_path.lower().endswith(".mp4"):
         try:
-                # Use temporary file with -i for subtitles 
-                subprocess.run(
-                    [
-                        "ffmpeg",
-                        "-i",
-                        audio_file_path,
-                        "-i",
-                        temp_srt_path,
-                        "-map",
-                        "0:v",
-                        "-map",
-                        "0:a",
-                        "-map",
-                        "1:s?",
-                        "-c:v",
-                        "copy",
-                        "-c:a",
-                        "copy",
-                        "-c:s",
-                        "mov_text",
-                        audio_file_path.replace(".mp4", "_with_subs.mp4"), # Create new file name
-                    ],
-                    check=True,
-                )
-                return temp_srt_file, audio_file_path.replace(".mp4", "_with_subs.mp4"), None, None # Return None for the download button
+            # ... (ffmpeg code to add subtitles) ...
+            return srt_content, audio_file_path.replace(".mp4", "_with_subs.mp4"), None, temp_srt_path  # Return the temp_srt_path
         except subprocess.CalledProcessError as e:
-            return temp_srt_file, None, f"Error during subtitle addition: {e}", None # Return None for the download button
+            return srt_content, None, f"Error during subtitle addition: {e}", temp_srt_path  # Return the temp_srt_path
+        finally:
+            if os.path.exists(temp_srt_path):
+                os.remove(temp_srt_path)
 
-    return temp_srt_file, None, None, None
+    return srt_content, None, None, temp_srt_path
 
 
 with gr.Blocks() as demo: