Update app.py

app.py

@@ -7,6 +7,10 @@ import torch
 from diffusers import AudioLDMPipeline
 from transformers import AutoProcessor, ClapModel, BlipProcessor, BlipForConditionalGeneration
 
+# Charger le modèle et le processeur Blip pour la description d'images
+image_caption_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
+image_caption_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
+
 # make Space compatible with CPU duplicates
 if torch.cuda.is_available():
     device = "cuda"
@@ -26,10 +30,6 @@ processor = AutoProcessor.from_pretrained("sanchit-gandhi/clap-htsat-unfused-m-f
 
 generator = torch.Generator(device)
 
-# Charger le modèle et le processeur Blip pour la description d'images
-image_caption_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
-image_caption_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
-
 # Streamlit app setup
 st.set_page_config(
     page_title="Text to Media",
@@ -55,7 +55,7 @@ if uploaded_files:
             f.write(uploaded_file.read())
         image_paths.append(image_path)
         
-        # Générez la légende pour chaque image
+        # Générer la légende pour chaque image
         try:
             image = Image.open(image_path).convert("RGB")
             inputs = image_caption_processor(image, return_tensors="pt")
@@ -65,7 +65,7 @@ if uploaded_files:
         except Exception as e:
             descriptions.append("Erreur lors de la génération de la légende")
 
-    # Affichez les images avec leurs descriptions
+    # Afficher les images avec leurs descriptions
     for i, image_path in enumerate(image_paths):
         st.image(image_path, caption=f"Description : {descriptions[i]}", use_column_width=True)
 
@@ -78,7 +78,7 @@ if uploaded_files:
 
     # Configuration de la musique
     seed = st.number_input("Seed", value=45)
-    duration = st.slider("Duration (seconds)", 2.5, 10.0, 5.0, 2.5)
+    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("Number waveforms to generate", 1, 3, 3, 1)
 
@@ -107,23 +107,27 @@ if uploaded_files:
         else:
             waveform = waveforms[0]
 
-    # Afficher le lecteur audio
+        # Afficher le lecteur audio
         st.audio(waveform, format="audio/wav", sample_rate=16000)
-    
+
     # Créer une vidéo à partir des images
-    if image_paths:
-        st.header("Génération du Diaporama Vidéo")
-        
-        output_video_path = os.path.join(temp_dir, "slideshow.mp4")
+    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)
 
-        # Débit d'images par seconde (calculé en fonction de la durée de chaque image)
-        frame_rate = 1 / image_duration
+    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_path], fps=frame_rate, durations=[image_duration]) for image_path in image_paths]
+            image_clips = [ImageSequenceClip([image], fps=frame_rate, durations=[image_duration]) for image in images]
 
-        final_clip = concatenate_videoclips(image_clips, method="compose")
+            final_clip = concatenate_videoclips(image_clips, method="compose")
 
-        final_clip.write_videofile(output_video_path, codec='libx264', fps=frame_rate)
+            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())
@@ -133,5 +137,3 @@ if uploaded_files:
             os.remove(image_path)
         os.remove(output_video_path)
         os.rmdir(temp_dir)
-
-