Update video_script.py

video_script.py

@@ -10,7 +10,6 @@ import subprocess
 from urllib.request import urlretrieve
 from openai import OpenAI
 from dotenv import load_dotenv
-import langdetect
 
 # Load environment variables
 load_dotenv()
@@ -158,23 +157,6 @@ def split_into_chunks(text, chunk_size=3):
     words = text.split()
     return [' '.join(words[i:i+chunk_size]) for i in range(0, len(words), chunk_size)]
 
-def detect_language(text):
-    try:
-        return langdetect.detect(text)
-    except:
-        return 'en'  # Default to English if detection fails
-
-def get_font_for_language(lang):
-    # Map of language codes to appropriate fonts
-    lang_font_map = {
-        'en': 'Impact',  # English
-        'hi': 'Noto Sans Devanagari',  # Hindi
-        'ur': 'Noto Sans Arabic',  # Urdu
-        'ar': 'Noto Sans Arabic',  # Arabic
-        # Add more mappings as needed
-    }
-    return lang_font_map.get(lang, 'Arial Unicode MS')  # Default to a Unicode-compatible font
-
 def create_ass_file(chunks, chunk_duration, output_ass, video_width, video_height):
     font_size = int(video_height * 0.05)
     impact_font_url = "https://picfy.xyz/uploads/impact.ttf"
@@ -184,25 +166,18 @@ def create_ass_file(chunks, chunk_duration, output_ass, video_width, video_heigh
     if not os.path.exists(impact_font_path):
         urlretrieve(impact_font_url, impact_font_path)
     
-    with open(output_ass, 'w', encoding='utf-8') as f:
+    with open(output_ass, 'w') as f:
         f.write("[Script Info]\nScriptType: v4.00+\nPlayResX: {}\nPlayResY: {}\n\n".format(video_width, video_height))
         f.write("[V4+ Styles]\nFormat: Name, Fontname, Fontsize, PrimaryColour, SecondaryColour, OutlineColour, BackColour, Bold, Italic, Underline, StrikeOut, ScaleX, ScaleY, Spacing, Angle, BorderStyle, Outline, Shadow, Alignment, MarginL, MarginR, MarginV, Encoding\n")
-        
-        # Define styles for different languages
-        for lang in set(detect_language(chunk) for chunk in chunks):
-            font = get_font_for_language(lang)
-            f.write(f"Style: {lang},\{font},{font_size},&H00FFFFFF,&H000000FF,&H00000000,&H00000000,0,0,0,0,100,100,0,0,1,2,0,5,10,10,10,1\n")
-        
-        f.write("\n[Events]\nFormat: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text\n")
+        f.write("Style: Default,Impact,{},&H00FFFFFF,&H000000FF,&H00000000,&H00000000,0,0,0,0,100,100,0,0,1,2,0,5,10,10,10,1\n\n".format(font_size))
+        f.write("[Events]\nFormat: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text\n")
 
         for i, chunk in enumerate(chunks):
             start_time = i * chunk_duration
             end_time = (i + 1) * chunk_duration
-            lang = detect_language(chunk)
-            f.write("Dialogue: 0,{},{},{},0,0,0,,{}\n".format(
+            f.write("Dialogue: 0,{},{},Default,,0,0,0,,{}\n".format(
                 format_time(start_time),
                 format_time(end_time),
-                lang,
                 chunk
             ))
 
@@ -225,7 +200,7 @@ def add_captions_to_video(video_file, audio_file, output_file):
     ass_file = "subtitles.ass"
     create_ass_file(chunks, chunk_duration, ass_file, int(video_width), int(video_height))
     
-    ffmpeg_cmd = f"ffmpeg -i {video_file} -i {audio_file} -vf \"subtitles={ass_file}:force_style='Fontname=Arial Unicode MS'\" -c:a aac -c:v libx264 {output_file}"
+    ffmpeg_cmd = f"ffmpeg -i {video_file} -i {audio_file} -vf \"ass={ass_file}:fontsdir=.\" -c:a aac -c:v libx264 {output_file}"
     subprocess.run(ffmpeg_cmd, shell=True, check=True)
     
     os.remove(ass_file)