Update app.py

app.py

@@ -3,56 +3,40 @@ import torch
 from transformers import MusicgenForConditionalGeneration, AutoProcessor
 import numpy as np
 
-# Load the model - 'small' is fastest and most reliable for free tiers
+# Load the model
 model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-small")
 processor = AutoProcessor.from_pretrained("facebook/musicgen-small")
 
 def generate_music(prompt, duration):
     if not prompt:
-        return None
-    
-    # Prepare the input
+        return None, "Please enter a description."
+        
     inputs = processor(text=[prompt], padding=True, return_tensors="pt")
-    
-    # Generate - 1 second is roughly 50 tokens
     max_tokens = int(duration * 50)
     audio_values = model.generate(**inputs, max_new_tokens=max_tokens)
     
-    # Get sampling rate and clean up audio for Gradio
     sampling_rate = model.config.audio_encoder.sampling_rate
     audio_data = audio_values[0, 0].cpu().numpy()
     
-    return (sampling_rate, audio_data)
+    # Legal Certificate generated with the track
+    license_cert = f"✅ COMMERCIAL LICENSE ACTIVATED\nPrompt: {prompt}\nStatus: Royalty-Free for Commercial Use"
+    
+    return (sampling_rate, audio_data), license_cert
 
-# Create a professional interface with Blocks
+# UI Layout
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("""# 🎵 AI Music Studio
-    *Describe a vibe, and our neural engine will compose a unique track for your blog.*""")
+    gr.HTML("<div style='text-align:center'><h1>🎹 NEURAL MUSIC STUDIO</h1><p>Generate & Own Your Tracks</p></div>")
     
     with gr.Row():
         with gr.Column():
-            text_input = gr.Textbox(
-                label="Music Description", 
-                placeholder="e.g., 'Lofi hip hop with a relaxing rain background' or '80s synthwave'"
-            )
-            duration_slider = gr.Slider(
-                minimum=2, maximum=15, value=8, step=1, 
-                label="Duration (Seconds)"
-            )
-            generate_btn = gr.Button("Compose Music", variant="primary")
-            
+            text_input = gr.Textbox(label="Music Style", placeholder="e.g., Chill lo-fi hip hop")
+            duration_slider = gr.Slider(minimum=5, maximum=20, value=10, label="Length (Sec)")
+            generate_btn = gr.Button("Compose Now", variant="primary")
+        
         with gr.Column():
-            audio_output = gr.Audio(label="Generated Track", type="numpy")
-            
-    generate_btn.click(
-        fn=generate_music, 
-        inputs=[text_input, duration_slider], 
-        outputs=audio_output
-    )
+            audio_out = gr.Audio(label="Studio Master Output", type="numpy")
+            license_out = gr.Textbox(label="Legal Rights", interactive=False)
 
-    gr.Examples(
-        examples=[["Upbeat jazz with a fast trumpet solo", 10], ["Ambient space music with deep bass", 12]],
-        inputs=[text_input, duration_slider]
-    )
+    generate_btn.click(fn=generate_music, inputs=[text_input, duration_slider], outputs=[audio_out, license_out])
 
 demo.launch()