Add Loading

app.py

@@ -3,50 +3,61 @@ from transformers import pipeline
 import scipy.io.wavfile
 import os
 
-# Stelle sicher, dass der static-Ordner existiert
+# Sicherstellen, dass der 'static'-Ordner existiert
 os.makedirs('static', exist_ok=True)
 
-print("Lade MusicGen Modell... (Dies kann einige Minuten dauern)")
-synthesiser = pipeline(
-    "text-to-audio", 
+# MusicGen-Pipeline initialisieren
+synthesizer = pipeline(
+    "text-to-audio",
     "facebook/musicgen-small",
     model_kwargs={"attn_implementation": "eager"}
 )
-print("Modell erfolgreich geladen!")
 
 # Funktion zur Musikgenerierung
 def generate_music(description):
     try:
-        print(f"Generiere Musik für: '{description}'")
-        music = synthesiser(description, forward_params={"do_sample": True})
-        
-        # Datei speichern
+        # Musik generieren
+        music = synthesizer(description, forward_params={"do_sample": True})
+        # Ausgabe-Pfad definieren
         output_path = "static/generated_music.wav"
+        # Musikdatei speichern
         scipy.io.wavfile.write(
-            output_path, 
-            rate=music["sampling_rate"], 
+            output_path,
+            rate=music["sampling_rate"],
             data=music["audio"]
         )
-        print("Musik erfolgreich generiert!")
-        return output_path  # Rückgabe des Dateipfads
+        return output_path  # Pfad zur generierten Musikdatei zurückgeben
     except Exception as e:
-        print(f"Fehler bei der Musikgenerierung: {str(e)}")
-        return f"Fehler: {str(e)}"
+        return str(e)  # Fehlernachricht zurückgeben
 
-# Gradio-Interface
-description = "Gib eine Beschreibung ein, um Musik zu generieren:"
-title = "MusicGen: Text-to-Audio Generator"
-examples = [["A relaxing piece of piano music for a calm evening."]]
+# Gradio-Interface erstellen
+with gr.Blocks() as demo:
+    gr.Markdown("# 🎶 MusicGen: Text-zu-Audio Generator")
+    gr.Markdown("Gib eine Beschreibung ein, um ein Musikstück zu generieren.")
+    
+    with gr.Row():
+        with gr.Column(scale=3):
+            description = gr.Textbox(
+                label="Musikbeschreibung",
+                placeholder="Z.B. Ein fröhliches Lied mit Gitarren.",
+                lines=3
+            )
+            generate_button = gr.Button("🎼 Generieren")
+        with gr.Column(scale=2):
+            output = gr.Audio(label="Generierte Musik", type="filepath", autoplay=True)
+    
+    # Beispiel-Eingaben
+    examples = [
+        ["Ein entspannendes Klavierstück für einen ruhigen Abend."],
+        ["Ein energiegeladener Rocksong mit E-Gitarren."],
+        ["Sanfte Hintergrundmusik mit Naturgeräuschen."]
+    ]
+    gr.Examples(examples=examples, inputs=description)
+    
+    # Ereignisbindung
+    generate_button.click(fn=generate_music, inputs=description, outputs=output)
 
-interface = gr.Interface(
-    fn=generate_music,
-    inputs=gr.Textbox(label="Musikbeschreibung", placeholder="Z.B. Ein fröhliches Lied mit Gitarren."),
-    outputs=gr.Audio(label="Generierte Musik", type="filepath"),
-    title=title,
-    description=description,
-    examples=examples
-)
-
-# App starten
+# Anwendung starten
 if __name__ == "__main__":
-    interface.launch()
+    demo.launch()
+# ddd