Create app.py

app.py

@@ -0,0 +1,87 @@
+import gradio as gr
+import torch
+import os
+import sys
+import numpy as np
+from PIL import Image
+import tempfile
+import uuid
+from datetime import datetime
+
+# Per garantire che le versioni corrispondano
+import pkg_resources
+required_packages = {
+    'torch': '>=2.4.0',
+    'transformers': '>=4.35.0',
+    'diffusers': '>=0.25.0'
+}
+
+for package, version in required_packages.items():
+    pkg_resources.require(f"{package}{version}")
+
+# Clona e installa Open-Sora se necessario
+if not os.path.exists("Open-Sora"):
+    os.system("git clone https://github.com/hpcaitech/Open-Sora")
+    os.system("cd Open-Sora && pip install -e .")
+    os.system("pip install xformers==0.0.27.post2 --index-url https://download.pytorch.org/whl/cu121")
+    os.system("pip install flash-attn --no-build-isolation")
+
+# Aggiungi il percorso al PYTHONPATH
+sys.path.append("Open-Sora")
+
+# Scarica i modelli se necessario
+if not os.path.exists("./ckpts"):
+    os.system("mkdir -p ./ckpts")
+    os.system("pip install huggingface_hub")
+    os.system("huggingface-cli download hpcai-tech/Open-Sora-v2 --local-dir ./ckpts")
+
+# Funzione per la generazione del video
+def generate_video(prompt):
+    try:
+        # Crea un ID unico per il file di output
+        output_id = str(uuid.uuid4())
+        output_path = f"outputs/{output_id}.mp4"
+        if not os.path.exists("outputs"):
+            os.makedirs("outputs")
+        
+        # Comando per generare il video (adattato dalla documentazione)
+        cmd = f"cd Open-Sora && python -m scripts.diffusion.inference \
+                configs/diffusion/inference/256px.py \
+                --cond_type t2v_head \
+                --prompt \"{prompt}\" \
+                --save-dir ../{output_path}"
+        
+        print(f"Esecuzione comando: {cmd}")
+        result = os.system(cmd)
+        
+        if result != 0:
+            return "Errore durante la generazione del video", None
+        
+        video_url = output_path
+        return "Generazione completata", video_url
+    except Exception as e:
+        return f"Errore: {str(e)}", None
+
+# Interfaccia Gradio
+with gr.Blocks() as demo:
+    gr.Markdown("# Generatore di Video con Open-Sora")
+    
+    with gr.Row():
+        with gr.Column():
+            prompt_input = gr.Textbox(
+                label="Descrivi il video che vuoi generare",
+                placeholder="Es. Un panda che mangia bambù in una foresta, stile cinematografico"
+            )
+            generate_btn = gr.Button("Genera Video")
+        
+        with gr.Column():
+            status_output = gr.Textbox(label="Stato")
+            video_output = gr.Video(label="Video Generato")
+    
+    generate_btn.click(
+        fn=generate_video,
+        inputs=[prompt_input],
+        outputs=[status_output, video_output]
+    )
+
+demo.launch()