Update app.py

app.py

@@ -13,6 +13,7 @@ from moviepy import VideoFileClip, TextClip, CompositeVideoClip, AudioFileClip
 import pysrt
 import instaloader
 import time
+import concurrent.futures
 api_key = "268976:66f4f58a2a905"
 
 
@@ -40,8 +41,51 @@ def download_file(url):
     except requests.exceptions.RequestException as e:
         print(f"An error occurred: {e}")
 
+def download_chunk(url, start, end, filename, index):
+    headers = {'Range': f'bytes={start}-{end}'}
+    response = requests.get(url, headers=headers, stream=True)
+    response.raise_for_status()
+    chunk_filename = f'{filename}.part{index}'
+    with open(chunk_filename, 'wb') as file:
+        for chunk in response.iter_content(chunk_size=8192):
+            if chunk:
+                file.write(chunk)
+    return chunk_filename
+
+def merge_files(filename, num_parts):
+    with open(filename, 'wb') as output_file:
+        for i in range(num_parts):
+            part_filename = f'{filename}.part{i}'
+            with open(part_filename, 'rb') as part_file:
+                output_file.write(part_file.read())
+            # Optionally, delete the part file after merging
+            # os.remove(part_filename)
+
+def download_file_in_parallel(link, size, num_threads=4):
+    url = link.split("#")[0]
+    filename = link.split("#")[1]
+    print(url+" filename: "+filename)
+    response = requests.head(url)
+    #file_size = int(response.headers['Content-Length'])
+    chunk_size = size // num_threads
+
+    ranges = [(i * chunk_size, (i + 1) * chunk_size - 1) for i in range(num_threads)]
+    ranges[-1] = (ranges[-1][0], size - 1)  # Adjust the last range to the end of the file
+
+    with concurrent.futures.ThreadPoolExecutor(max_workers=num_threads) as executor:
+        futures = [
+            executor.submit(download_chunk, url, start, end, filename, i)
+            for i, (start, end) in enumerate(ranges)
+        ]
+        for future in concurrent.futures.as_completed(futures):
+            future.result()  # Ensure all threads complete
+
+    merge_files(filename, num_threads)
+    print(f'Downloaded successfully: {filename}')
+
+
 
-def one_youtube(link):
+def one_youtube(link, api_key):
 
     # Fetch video ID
     video_id_url = f"https://one-api.ir/youtube/?token={api_key}&action=getvideoid&link={link}"
@@ -63,8 +107,15 @@ def one_youtube(link):
     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)
+    for f in formats_list:
+        if f["format_note"] == "360p":
+            download_id = f["id"]
+            video_size = f["filesize"]
+    for f in formats_list:
+        if f["format_note"] == "medium":
+            audio_id = f["id"]
+            audio_size = f["filesize"]
+    
     if not download_id or not audio_id:
         return None, None
 
@@ -81,8 +132,9 @@ def one_youtube(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)
+    print(video_size , audio_size)
+    download_file_in_parallel(vid_str, video_size)
+    download_file_in_parallel(audio_str, audio_size)
 
     return video_name, audio_name
 
@@ -312,7 +364,7 @@ def process_video(url, type):
         input_video_name = input_video.replace(".mp4", "")
         input_audio = extract_audio(input_video)
     elif type=="youtube":
-        input_video, input_audio = one_youtube(url)
+        input_video, input_audio = one_youtube(url, api_key)
         input_video_name = input_video.replace(".mp4", "")
     segments = transcribe(audio=input_audio)
     language = "fa"