Update modules/app.py

modules/app.py

@@ -2,9 +2,11 @@ from fastapi import FastAPI, HTTPException, Request, Depends
 from fastapi.staticfiles import StaticFiles
 from pytubefix import YouTube
 from pytubefix.exceptions import PytubeFixError
+from concurrent.futures import ThreadPoolExecutor
 import ffmpeg
 from pydantic import BaseModel, HttpUrl
 from typing import List
+import asyncio
 import requests
 import tempfile
 import time
@@ -40,9 +42,6 @@ def generate_hash(length=12):
 async def read_root():
     return {"message": "Saqib's API"}
 
-# API URL
-API_URL = "https://sxqib-api.hf.space"
-
 # Create a directory to store MP3 files if it doesn't exist
 AUDIO_DIR = "audio_files"
 os.makedirs(AUDIO_DIR, exist_ok=True)
@@ -64,6 +63,12 @@ class AudioImageInput(BaseModel):
 class VideosInput(BaseModel):
     video_urls: List[HttpUrl]
 
+thread_pool = ThreadPoolExecutor(max_workers=2)
+
+async def run_ffmpeg_async(ffmpeg_command):
+    loop = asyncio.get_running_loop()
+    await loop.run_in_executor(thread_pool, ffmpeg_command)
+
 async def download_file(url: str, suffix: str):
     async with aiohttp.ClientSession() as session:
         async with session.get(url) as response:
@@ -87,14 +92,17 @@ async def add_audio_to_image(input_data: AudioImageInput):
         # Use ffmpeg to combine image and audio into a video
         input_image = ffmpeg.input(temp_image_path, loop=1, t=5)  # 5 seconds duration
         input_audio = ffmpeg.input(temp_audio_path)
-        ffmpeg.concat(input_image, input_audio, v=1, a=1).output(output_path, vcodec='libx264', acodec='aac').run()
+        ffmpeg_command = ffmpeg.concat(input_image, input_audio, v=1, a=1).output(output_path, vcodec='libx264', acodec='aac')
+
+        # Run ffmpeg asynchronously
+        await run_ffmpeg_async(ffmpeg_command.run)
 
         # Clean up temporary files
         os.unlink(temp_image_path)
         os.unlink(temp_audio_path)
 
         # Return the URL path to the output file
-        return f"{API_URL}/output/{output_filename}"
+        return f"https://sxqib-api.hf.space/output/{output_filename}"
     except Exception as e:
         print(f"An error occurred: {str(e)}")
         print(traceback.format_exc())
@@ -104,7 +112,6 @@ async def add_audio_to_image(input_data: AudioImageInput):
 async def concatenate_videos(input_data: VideosInput):
     try:
         temp_video_paths = []
-
         # Download videos to temporary files
         for video_url in input_data.video_urls:
             temp_video_paths.append(await download_file(str(video_url), ".mp4"))
@@ -125,14 +132,17 @@ async def concatenate_videos(input_data: VideosInput):
         joined = ffmpeg.concat(*video_and_audio_streams, v=1, a=1).node
 
         # Merge video and audio
-        ffmpeg.output(joined[0], joined[1], output_path, vcodec='libx264', acodec='aac').run()
+        ffmpeg_command = ffmpeg.output(joined[0], joined[1], output_path, vcodec='libx264', acodec='aac')
+
+        # Run ffmpeg asynchronously
+        await run_ffmpeg_async(ffmpeg_command.run)
 
         # Clean up temporary files
         for path in temp_video_paths:
             os.remove(path)
 
         # Return the URL path to the output file
-        return f"{API_URL}/output/{output_filename}"
+        return f"https://sxqib-api.hf.space/output/{output_filename}"
     except Exception as e:
         print(f"An error occurred: {str(e)}")
         print(traceback.format_exc())
@@ -159,14 +169,14 @@ async def get_audio(url: str):
         print((f"Downloading audio to: {out_file}"))
         
         # Download the audio
-        video.download(output_path=AUDIO_DIR, filename=unique_filename)
+        video.download(output_path=AUDIO_DIR, filename=unique_filename, timeout=30)
         
         file_stats = os.stat(out_file)
         print(f'Size of audio file in Bytes: {file_stats.st_size}')
         
         if file_stats.st_size <= 30000000:
             # Construct the URL for the MP3 file
-            mp3_url = f"{API_URL}/audio/{unique_filename}"
+            mp3_url = f"https://sxqib-api.hf.space/audio/{unique_filename}"
             return mp3_url
         else:
             os.remove(out_file)