Update app.py

app.py

@@ -106,13 +106,37 @@ def check_job_status(job_id):
         time.sleep(10)
     return None
 
+def get_media_duration(file_path):
+    # Fetch media duration using ffprobe
+    cmd = ['ffprobe', '-v', 'error', '-show_entries', 'format=duration', '-of', 'default=noprint_wrappers=1:nokey=1', file_path]
+    result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    return float(result.stdout.strip())
+
 def combine_audio_video(video_path, audio_path, output_path):
-    cmd = [
-        'ffmpeg', '-i', video_path, '-i', audio_path,
-        '-map', '0:v', '-map', '1:a',
-        '-c:v', 'copy', '-c:a', 'aac',
-        '-shortest', '-y', output_path
-    ]
+    # Get durations of both video and audio
+    video_duration = get_media_duration(video_path)
+    audio_duration = get_media_duration(audio_path)
+
+    if video_duration > audio_duration:
+        # Trim video to match the audio length
+        cmd = [
+            'ffmpeg', '-i', video_path, '-i', audio_path,
+            '-t', str(audio_duration),  # Trim video to audio duration
+            '-map', '0:v', '-map', '1:a',
+            '-c:v', 'copy', '-c:a', 'aac',
+            '-y', output_path
+        ]
+    else:
+        # Loop video if it's shorter than audio
+        loop_count = int(audio_duration // video_duration) + 1  # Calculate how many times to loop
+        cmd = [
+            'ffmpeg', '-stream_loop', str(loop_count), '-i', video_path, '-i', audio_path,
+            '-t', str(audio_duration),  # Match the duration of the final video with the audio
+            '-map', '0:v', '-map', '1:a',
+            '-c:v', 'copy', '-c:a', 'aac',
+            '-shortest', '-y', output_path
+        ]
+
     subprocess.run(cmd, check=True)
 
 def process_video(voice, model, text, progress=gr.Progress()):
@@ -201,4 +225,4 @@ def create_interface():
 
 if __name__ == "__main__":
     app = create_interface()
-    app.launch()
+    app.launch()