Update app.py

app.py

@@ -7,12 +7,12 @@ from PIL import Image
 import streamlit as st
 from huggingface_hub import AsyncInferenceClient
 import asyncio
+from moviepy.editor import ImageSequenceClip
 
 MAX_SEED = np.iinfo(np.int32).max
 client = AsyncInferenceClient()
 DATA_PATH = Path("./data")
 DATA_PATH.mkdir(exist_ok=True)
-
 PREDEFINED_SEED = random.randint(0, MAX_SEED)
 
 async def generate_image(prompt, width, height, seed, model_name):
@@ -72,6 +72,7 @@ def get_prompts():
     prompt_files = [file for file in DATA_PATH.glob("*.txt") if file.is_file()]
     return {file.stem.replace("prompt_", ""): file for file in prompt_files}
 
+# Función para borrar todas las imágenes y prompts
 def delete_all_images():
     try:
         files = [file for file in DATA_PATH.glob("*.jpg")]
@@ -90,67 +91,85 @@ def download_images_as_zip():
     with open(zip_path, "rb") as zip_file:
         st.download_button(label="Descargar imágenes en .zip", data=zip_file, file_name="images.zip", mime="application/zip")
 
+def create_video_from_images():
+    try:
+        image_files = sorted(DATA_PATH.glob("*.jpg"))
+        if not image_files:
+            st.error("No hay imágenes disponibles para crear un video.")
+            return
+        
+        image_sequence = [Image.open(image_file) for image_file in image_files]
+        frame_rate = 1  # 1 segundo por cada imagen
+        
+        clip = ImageSequenceClip([np.array(img) for img in image_sequence], fps=frame_rate)
+        video_path = DATA_PATH / "output_video.mp4"
+        clip.write_videofile(str(video_path), codec="libx264")
+
+        with open(video_path, "rb") as video_file:
+            st.download_button(label="Descargar video generado", data=video_file, file_name="output_video.mp4", mime="video/mp4")
+        
+        st.success("El video ha sido generado con éxito.")
+    except Exception as e:
+        st.error(f"Error al generar el video: {e}")
+
 def main():
     st.set_page_config(layout="wide")
     st.title("Generador de Imágenes")
-    st.warning("Este espacio contiene contenido que no es adecuado para todas las audiencias. Se recomienda discreción.")
-    agree = st.checkbox("Soy mayor de 18 años y entiendo que el contenido puede no ser apropiado.")
 
-    if agree:
-        prompt = st.sidebar.text_input("Descripción de la imagen", max_chars=500)
-        format_option = st.sidebar.selectbox("Formato", ["9:16", "16:9"])
-        model_option = st.sidebar.selectbox("Modelo", ["enhanceaiteam/Flux-uncensored", "enhanceaiteam/Flux-Uncensored-V2"])
-        
-        width, height = (720, 1280) if format_option == "9:16" else (1280, 720)
-
-        if st.sidebar.button("Generar Imagen"):
-            with st.spinner("Generando imagen..."):
-                result = asyncio.run(gen(prompt, width, height, model_option))
-                image_paths = result[0]
-                prompt_file = result[1]
-
-            if image_paths:
-                if Path(image_paths).exists():
-                    st.image(image_paths, caption="Imagen Generada")
-                else:
-                    st.error("El archivo de imagen no existe.")
-
-                if prompt_file and Path(prompt_file).exists():
-                    prompt_text = Path(prompt_file).read_text()
-                    st.write(f"Prompt utilizado: {prompt_text}")
-                else:
-                    st.write("El archivo del prompt no está disponible.")
-
-        files, usage = get_storage()
-        st.text(usage)
-        cols = st.columns(6)
-        prompts = get_prompts()
-
-        for idx, file in enumerate(files):
-            with cols[idx % 6]:
-                image = Image.open(file)
-                prompt_file = prompts.get(Path(file).stem.replace("image_", ""), None)
-                prompt_text = Path(prompt_file).read_text() if prompt_file else "No disponible"
-                
-                st.image(image, caption=f"Imagen {idx+1}")
-                st.write(f"Prompt: {prompt_text}")
-                
-                if st.button(f"Borrar Imagen {idx+1}", key=f"delete_{idx}"):
-                    try:
-                        os.remove(file)  
-                        if prompt_file:
-                            os.remove(prompt_file)
-                        st.success(f"Imagen {idx+1} y su prompt fueron borrados.")
-                    except Exception as e:
-                        st.error(f"Error al borrar la imagen o prompt: {e}")
-
-        if st.button("Borrar todas las imágenes"):
-            delete_all_images()
-
-        download_images_as_zip()
-
-    else:
-        st.error("No puedes acceder a este espacio hasta que confirmes que eres mayor de edad.")
+    prompt = st.sidebar.text_input("Descripción de la imagen", max_chars=500)
+    format_option = st.sidebar.selectbox("Formato", ["9:16", "16:9"])
+    model_option = st.sidebar.selectbox("Modelo", ["modelo_actual", "enhanceaiteam/Flux-Uncensored-V2"])
+    
+    width, height = (720, 1280) if format_option == "9:16" else (1280, 720)
+
+    if st.sidebar.button("Generar Imagen"):
+        with st.spinner("Generando imagen..."):
+            result = asyncio.run(gen(prompt, width, height, model_option))
+            image_paths = result[0]
+            prompt_file = result[1]
+
+        if image_paths:
+            if Path(image_paths).exists():
+                st.image(image_paths, caption="Imagen Generada")
+            else:
+                st.error("El archivo de imagen no existe.")
+
+            if prompt_file and Path(prompt_file).exists():
+                prompt_text = Path(prompt_file).read_text()
+                st.write(f"Prompt utilizado: {prompt_text}")
+            else:
+                st.write("El archivo del prompt no está disponible.")
+
+    files, usage = get_storage()
+    st.text(usage)
+    cols = st.columns(6)
+    prompts = get_prompts()
+
+    for idx, file in enumerate(files):
+        with cols[idx % 6]:
+            image = Image.open(file)
+            prompt_file = prompts.get(Path(file).stem.replace("image_", ""), None)
+            prompt_text = Path(prompt_file).read_text() if prompt_file else "No disponible"
+            
+            st.image(image, caption=f"Imagen {idx+1}")
+            st.write(f"Prompt: {prompt_text}")
+            
+            if st.button(f"Borrar Imagen {idx+1}", key=f"delete_{idx}"):
+                try:
+                    os.remove(file)  
+                    if prompt_file:
+                        os.remove(prompt_file)
+                    st.success(f"Imagen {idx+1} y su prompt fueron borrados.")
+                except Exception as e:
+                    st.error(f"Error al borrar la imagen o prompt: {e}")
+
+    if st.button("Borrar todas las imágenes"):
+        delete_all_images()
+
+    download_images_as_zip()
+
+    if st.button("Generar video con las imágenes"):
+        create_video_from_images()
 
 if __name__ == "__main__": 
     main()