Update app.py

app.py

@@ -9,13 +9,82 @@ import requests
 import json
 import arabic_reshaper # pip install arabic-reshaper
 from bidi.algorithm import get_display # pip install python-bidi
-from moviepy import VideoFileClip, TextClip, CompositeVideoClip
+from moviepy.editor import VideoFileClip, TextClip, CompositeVideoClip, AudioFileClip
 import pysrt
 api_key = "268976:66f4f58a2a905"
 
 
 
 
+def fetch_data(url):
+    try:
+        response = requests.get(url)
+        response.raise_for_status()
+        return response.json()
+    except requests.exceptions.RequestException as e:
+        print(f"An error occurred: {e}")
+        return None
+
+def download_file(url):
+    try:
+        response = requests.get(url.split("#")[0], stream=True)
+        response.raise_for_status()
+        print(url.split("#")[1])
+        with open(url.split("#")[1], 'wb') as file:
+            for chunk in response.iter_content(chunk_size=8192):
+                if chunk:
+                    file.write(chunk)
+        print(f"Downloaded successfully: {url.split('#')[1]}")
+    except requests.exceptions.RequestException as e:
+        print(f"An error occurred: {e}")
+
+
+def one_youtube(link):
+
+    # Fetch video ID
+    video_id_url = f"https://one-api.ir/youtube/?token={api_key}&action=getvideoid&link={link}"
+    video_data = fetch_data(video_id_url)
+    if not video_data:
+        return None, None
+
+    video_id = video_data["result"]
+
+    # Fetch video data
+    filter_option = ""  # Replace with your filter option
+    video_data_url = f"https://youtube.one-api.ir/?token={api_key}&action=fullvideo&id={video_id}&filter={filter_option}"
+    video_data_2 = fetch_data(video_data_url)
+    if not video_data_2:
+        return None, None
+
+    formats_list = video_data_2["result"]["formats"]
+    file_name = video_data_2["result"]["title"]
+    video_name = f'{file_name}.mp4'
+    audio_name = f'{file_name}.mp3'
+
+    download_id = next((f["id"] for f in formats_list if f["format_note"] == "360p"), None)
+    audio_id = next((f["id"] for f in formats_list if f["format_note"] == "medium"), None)
+    if not download_id or not audio_id:
+        return None, None
+
+    # Fetch video and audio links
+    video_link_url = f"https://youtube.one-api.ir/?token={api_key}&action=download&id={download_id}"
+    audio_link_url = f"https://youtube.one-api.ir/?token={api_key}&action=download&id={audio_id}"
+    video_link_data = fetch_data(video_link_url)
+    audio_link_data = fetch_data(audio_link_url)
+    if not video_link_data or not audio_link_data:
+        return None, None
+
+    video_link = video_link_data["result"]["link"]
+    audio_link = audio_link_data["result"]["link"]
+    vid_str=video_link+"#"+video_name
+    audio_str=audio_link+"#"+audio_name
+    # Download video and audio files
+    download_file(vid_str)
+    download_file(audio_str)
+
+    return video_name, audio_name
+
+
 # Define your functions here
 def yt_download(url):
     yt = YouTube(url)
@@ -108,7 +177,7 @@ def write_google(google_translate):
 def time_to_seconds(time_obj):
     return time_obj.hours * 3600 + time_obj.minutes * 60 + time_obj.seconds + time_obj.milliseconds / 1000
 
-def create_subtitle_clips(subtitles, videosize, fontsize=24, font='/content/arial-unicode-ms.ttf', color='yellow', debug=False):
+def create_subtitle_clips(subtitles, videosize, fontsize=32, font='/content/Arial Unicode MS.ttf', color='yellow', debug=False):
     subtitle_clips = []
     for subtitle in subtitles:
         start_time = time_to_seconds(subtitle.start)
@@ -117,18 +186,18 @@ def create_subtitle_clips(subtitles, videosize, fontsize=24, font='/content/aria
         video_width, video_height = videosize
         reshaped_text = arabic_reshaper.reshape(subtitle.text)
         bidi_text = get_display(reshaped_text)
-        text_clip = TextClip(bidi_text, fontsize=fontsize, font=font, color=color, bg_color='black', size=(video_width*3/4, None), method='label', align='West').set_start(start_time).set_duration(duration)
+        text_clip = TextClip(bidi_text, fontsize=fontsize, font=font, color=color, bg_color='black', size=(video_width*3/4, None), method='label', align='ٍEast').set_start(start_time).set_duration(duration)
         subtitle_x_position = 'center'
         subtitle_y_position = video_height * 4 / 5
         text_position = (subtitle_x_position, subtitle_y_position)
         subtitle_clips.append(text_clip.set_position(text_position))
     return subtitle_clips
 
-def process_video(url, api_key):
-    input_video, title = yt_download(url)
+def process_video(url):
+
+    input_video, input_audio = one_youtube(url)
     input_video_name = input_video.replace(".mp4", "")
-    extracted_audio = extract_audio(input_video_name)
-    segments = transcribe(audio=extracted_audio)
+    segments = transcribe(audio=input_audio)
     language = "fa"
     subtitle_file = generate_subtitle_file(language=language, segments=segments, input_video_name=input_video_name)
     source_language = "en"
@@ -137,16 +206,19 @@ def process_video(url, api_key):
     google_translate = translate_text(api_key, source_language, target_language, srt_string)
     write_google(google_translate)
     video = VideoFileClip(input_video)
-    subtitles = pysrt.open("/content/google_translate.srt", encoding="utf-8")
+    audio = AudioFileClip(input_audio)
+    video = video.set_audio(audio)
+    print(video)
+    subtitles = pysrt.open("google_translate.srt", encoding="utf-8")
     output_video_file = input_video_name + '_subtitled' + ".mp4"
     subtitle_clips = create_subtitle_clips(subtitles, video.size)
     final_video = CompositeVideoClip([video] + subtitle_clips)
     final_video.write_videofile(output_video_file, codec="libx264", audio_codec="aac")
-    return output_video_file
+    return google_translate, output_video_file
 
-def download_file(file_path):
-    return gr.File.update(file_path)
+#def download_file(file_path):
+ #   return gr.File.update(file_path)
 
-iface = gr.Interface(fn=process_video, inputs=["text", "text"], outputs="file")
+iface = gr.Interface(fn=process_video, inputs=["text"], outputs="file")
 
-iface.launch()
+iface.launch(debug=True)