Update app.py

app.py

@@ -287,7 +287,27 @@ def write_srt(subtitle_text, output_file="edited_srt.srt"):
     with open(output_file+".srt", 'w', encoding="utf-8") as file:
         file.write(subtitle_text)
     return output_file+".srt"
-        
+
+def generate_translated_subtitle(language, segments, input_video_name):
+    subtitle_file = f"sub-{input_video_name}-{language}.srt"
+    text = ""
+    lines = segments.split('\n')
+    new_list = [item for item in lines if item != '']
+    segment_number = 1
+
+    for index, segment in enumerate(new_list):
+        if (index+1) % 3 == 1 or (index+1)==1:
+            text += f"{segment}\n"
+            segment_number += 1
+        if (index+1) % 3 == 2 or (index+1)==2:
+            text += segment + "\n"
+        if (index+1) % 3 == 0:
+            text += f"\u200F{segment}\n\n"
+
+    with open(subtitle_file, "w", encoding='utf8') as f:
+        f.write(text)
+    return subtitle_file
+
 def clean_text(text):
     # Remove 'srt ' from the start of each line
     # Remove ''' from the start and end
@@ -448,7 +468,7 @@ def process_video(video, url, type):
 
 def video_edit(srt, input_video, input_audio= 'audio.mp3'):
     input_video_name = input_video.replace(".mp4", "")
-    srt_name=write_srt(srt, input_video_name)
+    srt_name=generate_translated_subtitle("fa", srt, input_video_name)
     return input_video, srt_name
     """    input_video_name = input_video.replace(''.mp4', '')
     video = VideoFileClip(input_video)