Update app.py

app.py

@@ -87,7 +87,7 @@ if uploaded_files:
 
     # Configuration de la musique
     seed = st.number_input("Seed", value=45)
-    music_duration = image_duration * len(image_clips)  # Assurez-vous que la durée de la musique correspond à la vidéo
+    music_duration = final_clip.duration  # Assurez-vous que la durée de la musique correspond à la vidéo
 
     if st.button("Générer de la musique"):
         waveforms = pipe(
@@ -100,10 +100,8 @@ if uploaded_files:
         )["audios"]
 
         # Assurez-vous que la musique générée a la même durée que la vidéo
-        if len(waveforms[0]) > len(final_clip):
-            waveforms = waveforms[:, :len(final_clip)]
-        else:
-            final_clip = final_clip.set_duration(len(waveforms[0]))
+        if waveforms[0].shape[0] > final_clip.fps * final_clip.duration:
+            waveforms[0] = waveforms[0][: int(final_clip.fps * final_clip.duration)]
 
         # Intégrez la musique dans la vidéo
         final_clip = final_clip.set_audio(waveforms[0].numpy())