updated

app.py

@@ -49,14 +49,14 @@ def cleanup_temp_files():
     import time
     temp_dir = tempfile.gettempdir()
     cutoff_time = time.time() - 3600
-    
+
     # Clean old generated music files
     patterns = [
         os.path.join(temp_dir, "tmp*.wav"),
         os.path.join(temp_dir, "generated_music_*.wav"),
         os.path.join(temp_dir, "musicgen_variant_*.wav")
     ]
-    
+
     for pattern in patterns:
         for temp_file in glob.glob(pattern):
             try:
@@ -312,7 +312,7 @@ def generate_music_batch(text_prompt, duration=10, guidance_scale=3.0, model_mod
 
         batch_size = MODEL_CONFIG['batch_size'] if MODEL_CONFIG['batch_mode'] else 1
         prompts = [text_prompt] * batch_size
-        
+
         start_time = time.time()
         outputs = generator(
             prompts,
@@ -325,21 +325,21 @@ def generate_music_batch(text_prompt, duration=10, guidance_scale=3.0, model_mod
 
         audio_variants = []
         sample_rate = outputs[0]['sampling_rate']
-        
+
         # Create unique timestamp for this generation batch
         batch_timestamp = int(time.time() * 1000)
-        
+
         for i, output in enumerate(outputs):
             audio_data = output['audio']
-            
+
             print(f"[GENERATION] Processing variant {i + 1} audio shape: {audio_data.shape}")
-            
+
             if hasattr(audio_data, 'cpu'):
                 audio_data = audio_data.cpu().numpy()
 
             if len(audio_data.shape) == 3:
                 audio_data = audio_data[0]
-                
+
             if len(audio_data.shape) == 2:
                 if audio_data.shape[0] < audio_data.shape[1]:
                     audio_data = audio_data.T
@@ -347,36 +347,34 @@ def generate_music_batch(text_prompt, duration=10, guidance_scale=3.0, model_mod
                     audio_data = audio_data[:, 0]
                 else:
                     audio_data = audio_data.flatten()
-            
+
             audio_data = audio_data.flatten()
-            
+
             max_val = np.max(np.abs(audio_data))
             if max_val > 0:
                 audio_data = audio_data / max_val * 0.95
-            
+
             audio_data = (audio_data * 32767).astype(np.int16)
-            
+
             # Save each variant to a unique temporary file
-            temp_filename = f"musicgen_variant_{i+1}_{batch_timestamp}.wav"
+            temp_filename = f"musicgen_variant_{i + 1}_{batch_timestamp}.wav"
             temp_path = os.path.join(tempfile.gettempdir(), temp_filename)
-            
+
             sf.write(temp_path, audio_data, sample_rate)
-            
+
             print(f"[GENERATION] Variant {i + 1} saved to: {temp_path}")
             print(f"[GENERATION] Variant {i + 1} file size: {os.path.getsize(temp_path)} bytes")
-            
+
             audio_variants.append(temp_path)
-            
+
             print(f"[GENERATION] Variant {i + 1} final shape: {audio_data.shape}")
 
         while len(audio_variants) < 6:
             audio_variants.append(None)
 
-        savings_message = get_fixed_savings_message()
-        
         variants_text = "audio"
-        generation_info = f"✅ Generated {variants_text} in {generation_time:.2f}s\n{savings_message}"
-        
+        generation_info = f"✅ Generated {variants_text} in {generation_time:.2f}s\n"
+
         return audio_variants[0], audio_variants[1], audio_variants[2], audio_variants[3], audio_variants[4], \
                audio_variants[5], generation_info
 
@@ -434,12 +432,12 @@ with gr.Blocks(title="MusicGen Large - Music Generation") as demo:
             audio_output1 = gr.Audio(label="Variant 1", type="filepath", visible=actual_outputs >= 1)
             audio_output2 = gr.Audio(label="Variant 2", type="filepath", visible=actual_outputs >= 2)
             audio_outputs.extend([audio_output1, audio_output2])
-        
+
         with gr.Row():
             audio_output3 = gr.Audio(label="Variant 3", type="filepath", visible=actual_outputs >= 3)
             audio_output4 = gr.Audio(label="Variant 4", type="filepath", visible=actual_outputs >= 4)
             audio_outputs.extend([audio_output3, audio_output4])
-        
+
         with gr.Row():
             audio_output5 = gr.Audio(label="Variant 5", type="filepath", visible=actual_outputs >= 5)
             audio_output6 = gr.Audio(label="Variant 6", type="filepath", visible=actual_outputs >= 6)