Update app.py

app.py

@@ -284,8 +284,9 @@ def read_srt_file(file_path):
         print(f"An error occurred: {e}")
         
 def write_srt(subtitle_text, output_file="edited_srt.srt"):
-    with open(output_file, 'w', encoding="utf-8") as file:
+    with open(output_file+".srt", 'w', encoding="utf-8") as file:
         file.write(subtitle_text)
+    return output_file+".srt"
         
 def clean_text(text):
     # Remove 'srt ' from the start of each line
@@ -386,7 +387,7 @@ def write_google(google_translate):
     google = "google_translate.srt"
     with open(google, 'a', encoding="utf-8") as f:
         f.write(google_translate)
-
+    
 def time_to_seconds(time_obj):
     return time_obj.hours * 3600 + time_obj.minutes * 60 + time_obj.seconds + time_obj.milliseconds / 1000
     
@@ -446,8 +447,9 @@ def process_video(video, url, type):
     return srt, video, input_audio
 
 def video_edit(srt, input_video, input_audio= 'audio.mp3'):
-    write_srt(srt)
-    return input_video, 'edited_srt.srt'
+    input_video_name = input_video.replace(''.mp4', '')
+    srt_name=write_srt(srt, input_video_name)
+    return input_video, srt_name
     """    input_video_name = input_video.replace(''.mp4', '')
     video = VideoFileClip(input_video)
     audio = AudioFileClip(input_audio)