Update app.py

app.py

@@ -41,11 +41,8 @@ st.title("Générateur de Diaporama Vidéo et Musique")
 # Sélectionnez les images
 uploaded_files = st.file_uploader("Sélectionnez des images (PNG, JPG, JPEG)", type=["png", "jpg", "jpeg"], accept_multiple_files=True)
 
-# Sélection de la durée d'affichage de chaque image avec une barre horizontale (en secondes)
-image_duration = st.slider("Sélectionnez la durée d'affichage de chaque image (en secondes)", 1, 10, 4)
-
 if uploaded_files:
-    # Créer un répertoire temporaire pour stocker les images
+    # Créez un répertoire temporaire pour stocker les images
     temp_dir = tempfile.mkdtemp()
     
     # Enregistrez les images téléchargées dans le répertoire temporaire
@@ -72,28 +69,6 @@ if uploaded_files:
     for i, image_path in enumerate(image_paths):
         st.image(image_path, caption=f"Description : {descriptions[i]}", use_column_width=True)
 
-    # Créer une vidéo à partir des images
-    if image_paths:
-        output_video_path = os.path.join(temp_dir, "slideshow.mp4")
-
-        # Débit d'images par seconde (calculé en fonction de la durée de chaque image)
-        frame_rate = 1 / image_duration
-
-        image_clips = [ImageSequenceClip([image_path], fps=frame_rate, durations=[image_duration]) for image_path in image_paths]
-
-        final_clip = concatenate_videoclips(image_clips, method="compose")
-
-        final_clip.write_videofile(output_video_path, codec='libx264', fps=frame_rate)
-
-        # Afficher la vidéo
-        st.video(open(output_video_path, 'rb').read())
-
-        # Supprimer le répertoire temporaire
-        for image_path in image_paths:
-            os.remove(image_path)
-        os.remove(output_video_path)
-        os.rmdir(temp_dir)
-
     # Générez de la musique à partir des descriptions
     st.header("Génération de Musique à partir des Descriptions")
 
@@ -103,24 +78,24 @@ if uploaded_files:
 
     # Configuration de la musique
     seed = st.number_input("Seed", value=45)
-    music_duration = st.slider("Durée de la musique (en secondes)", 10, 180, 60, 10)
+    duration = st.slider("Duration (seconds) de la musique", 2.5, 10.0, 5.0, 2.5)
     guidance_scale = st.slider("Guidance scale", 0.0, 4.0, 2.5, 0.5)
-    n_candidates = st.slider("Nombre de formes d'onde à générer", 1, 3, 3, 1)
+    n_candidates = st.slider("Number waveforms to generate", 1, 3, 3, 1)
 
     def score_waveforms(text, waveforms):
         inputs = processor(text=text, audios=list(waveforms), return_tensors="pt", padding=True)
         inputs = {key: inputs[key].to(device) for key in inputs}
         with torch.no_grad():
-            logits_per_text = clap_model(**inputs).logits_per_text
-            probs = logits_per_text.softmax(dim=-1)
-            most_probable = torch.argmax(probs)
+            logits_per_text = clap_model(**inputs).logits_per_text  # this is the audio-text similarity score
+            probs = logits_per_text.softmax(dim=-1)  # we can take the softmax to get the label probabilities
+            most_probable = torch.argmax(probs)  # and now select the most likely audio waveform
         waveform = waveforms[most_probable]
         return waveform
 
     if st.button("Générer de la musique"):
         waveforms = pipe(
             music_input,
-            audio_length_in_s=music_duration,
+            audio_length_in_s=duration,
             guidance_scale=guidance_scale,
             num_inference_steps=100,
             num_waveforms_per_prompt=n_candidates if n_candidates else 1,
@@ -132,17 +107,34 @@ if uploaded_files:
         else:
             waveform = waveforms[0]
 
-        # Sauvegardez la musique générée dans un fichier temporaire
-        music_temp_path = os.path.join(temp_dir, "generated_music.wav")
-        waveform.save(music_temp_path)
-
         # Afficher le lecteur audio
-        st.audio(open(music_temp_path, 'rb').read(), format="audio/wav", sample_rate=16000)
+        st.audio(waveform, format="audio/wav", sample_rate=16000)
+
+    # Créer une vidéo à partir des images
+    st.header("Génération du Diaporama Vidéo")
+
+    # Configuration du diaporama
+    image_duration = st.slider("Duration (seconds) des images dans le diaporama", 1, 10, 4)
+
+    if st.button("Créer le Diaporama Vidéo"):
+        def create_video(images, output_video_path):
+            frame_rate = 1 / image_duration
+
+            image_clips = [ImageSequenceClip([image], fps=frame_rate, durations=[image_duration]) for image in images]
+
+            final_clip = concatenate_videoclips(image_clips, method="compose")
+
+            final_clip.write_videofile(output_video_path, codec='libx264', fps=frame_rate)
+
+        output_video_path = os.path.join(temp_dir, "slideshow.mp4")
+        create_video(image_paths, output_video_path)
+
+        # Afficher la vidéo
+        st.video(open(output_video_path, 'rb').read())
 
         # Supprimer le répertoire temporaire
         for image_path in image_paths:
             os.remove(image_path)
         os.remove(output_video_path)
-        os.remove(music_temp_path)
         os.rmdir(temp_dir)