Update main.py

main.py

@@ -72,14 +72,14 @@ async def add_audio_to_image(request: Request):
         image_file = await download_file(image_url, ".jpg")
         audio_file = await download_file(audio_url, ".mp3")
 
-        # Run ffmpeg command
-        ffmpeg_cmd = f"ffmpeg -loop 1 -i {image_file} -i {audio_file} -c:v libx264 -tune stillimage -c:a aac -b:a 192k -shortest -pix_fmt yuv420p {output_path}"
+        # Run ffmpeg command with improved audio detection parameters
+        ffmpeg_cmd = f"ffmpeg -loop 1 -i {image_file} -analyzeduration 10000000 -probesize 10000000 -i {audio_file} -c:v libx264 -tune stillimage -c:a aac -b:a 192k -strict experimental -shortest -pix_fmt yuv420p {output_path}"
         process = await asyncio.create_subprocess_shell(
             ffmpeg_cmd,
             stdout=asyncio.subprocess.PIPE,
             stderr=asyncio.subprocess.PIPE
         )
-        stdout, stderr = await process.communicate()
+        _, stderr = await process.communicate()
 
         if process.returncode != 0:
             print(f"FFmpeg error: {stderr.decode()}")
@@ -114,14 +114,14 @@ async def add_audio_to_video(request: Request):
         video_file = await download_file(video_url, ".mp4")
         audio_file = await download_file(audio_url, ".mp3")
 
-        # Run ffmpeg command
-        ffmpeg_cmd = f"ffmpeg -i {video_file} -i {audio_file} -c:v copy -c:a aac -shortest {output_path}"
+        # Run ffmpeg command with improved audio detection parameters
+        ffmpeg_cmd = f"ffmpeg -analyzeduration 10000000 -probesize 10000000 -i {video_file} -analyzeduration 10000000 -probesize 10000000 -i {audio_file} -c:v copy -c:a aac -strict experimental -shortest {output_path}"
         process = await asyncio.create_subprocess_shell(
             ffmpeg_cmd,
             stdout=asyncio.subprocess.PIPE,
             stderr=asyncio.subprocess.PIPE
         )
-        stdout, stderr = await process.communicate()
+        _, stderr = await process.communicate()
 
         if process.returncode != 0:
             print(f"FFmpeg error: {stderr.decode()}")
@@ -171,7 +171,7 @@ async def concatenate_videos(request: Request):
             stdout=asyncio.subprocess.PIPE,
             stderr=asyncio.subprocess.PIPE
         )
-        stdout, stderr = await process.communicate()
+        _, stderr = await process.communicate()
 
         if process.returncode != 0:
             print(f"FFmpeg error: {stderr.decode()}")
@@ -216,14 +216,14 @@ async def concatenate_audio(request: Request):
             for file in audio_files:
                 f.write(f"file '{file}'\n")
 
-        # Run ffmpeg command
-        ffmpeg_cmd = f"ffmpeg -f concat -safe 0 -i {concat_list_path} -c copy {output_path}"
+        # Run ffmpeg command with improved audio parameters
+        ffmpeg_cmd = f"ffmpeg -f concat -safe 0 -i {concat_list_path} -c:a aac -b:a 192k -strict experimental {output_path}"
         process = await asyncio.create_subprocess_shell(
             ffmpeg_cmd,
             stdout=asyncio.subprocess.PIPE,
             stderr=asyncio.subprocess.PIPE
         )
-        stdout, stderr = await process.communicate()
+        _, stderr = await process.communicate()
 
         if process.returncode != 0:
             print(f"FFmpeg error: {stderr.decode()}")
@@ -409,23 +409,25 @@ async def create_slideshow(request: Request):
             stdout=asyncio.subprocess.PIPE,
             stderr=asyncio.subprocess.PIPE
         )
-        stdout, stderr = await process.communicate()
+        _, stderr = await process.communicate()
 
         if process.returncode != 0:
             print(f"FFmpeg slideshow creation error: {stderr.decode()}")
             raise HTTPException(status_code=500, detail=f"FFmpeg slideshow creation failed: {stderr.decode()}")
 
         # Add audio to the slideshow
+        # Increase analyzeduration and probesize for better audio detection
         final_cmd = (
-            f'ffmpeg -i {intermediate_video} -i {audio_file} -map 0:v -map 1:a '
-            f'-c:v copy -c:a aac -shortest {output_path}'
+            f'ffmpeg -analyzeduration 10000000 -probesize 10000000 -i {intermediate_video} '
+            f'-analyzeduration 10000000 -probesize 10000000 -i {audio_file} '
+            f'-c:v copy -c:a aac -strict experimental -shortest {output_path}'
         )
         process = await asyncio.create_subprocess_shell(
             final_cmd,
             stdout=asyncio.subprocess.PIPE,
             stderr=asyncio.subprocess.PIPE
         )
-        stdout, stderr = await process.communicate()
+        _, stderr = await process.communicate()
 
         if process.returncode != 0:
             print(f"FFmpeg audio addition error: {stderr.decode()}")