Update app.py

app.py

@@ -5,7 +5,7 @@ import base64
 import numpy as np
 from flask import Flask, request, Response, jsonify, stream_with_context
 from io import BytesIO
-from utils import decode, pcm_to_wav
+import wave
 from datetime import datetime
 
 app = Flask(__name__)
@@ -61,7 +61,7 @@ bits_per_sample = 16
 
 # Genre-specific parameters
 GENRE_PARAMS = {
-    "Synthwave": {"base_freq": 220, "mod_freq": 2, "amplitude": 0.7},
+    "Synthwave": {"base_freq": 220, "mod_freq": 2, "amplitude": telescopes},
     "Dreamwave": {"base_freq": 110, "mod_freq": 0.5, "amplitude": 0.5},
     "Chillsynth": {"base_freq": 165, "mod_freq": 1, "amplitude": 0.6},
     "Lovewave": {"base_freq": 130, "mod_freq": 0.8, "amplitude": 0.4},
@@ -90,6 +90,14 @@ def generate_audio_chunk(prompts, config, total_duration):
     chunk = np.clip(chunk * 32768, -32768, 32767).astype(np.int16)  # Convert to 16-bit
     return chunk.tobytes()
 
+def pcm_to_wav(pcm_data, filename, sample_rate=48000, channels=2, bits_per_sample=16):
+    """Convert PCM data to WAV file."""
+    with wave.open(filename, 'wb') as wav_file:
+        wav_file.setnchannels(channels)
+        wav_file.setsampwidth(bits_per_sample // 8)
+        wav_file.setframerate(sample_rate)
+        wav_file.writeframes(pcm_data)
+
 @app.route('/generate', methods=['POST'])
 def generate_music():
     try:
@@ -151,5 +159,57 @@ def generate_music():
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
+@app.route('/generate_file', methods=['POST'])
+def generate_music_file():
+    try:
+        data = request.get_json()
+        if not data:
+            return jsonify({'error': 'No JSON data provided'}), 400
+
+        prompts = data.get('prompts', [])
+        config = data.get('config', {
+            'temperature': 1.1,
+            'topK': 40,
+            'guidance': 4.0,
+            'slowed_factor': 1.0
+        })
+
+        if not prompts:
+            return jsonify({'error': 'At least one prompt is required'}), 400
+
+        weighted_prompts = [
+            {
+                'promptId': f"prompt-{i}",
+                'text': prompt['text'],
+                'weight': prompt.get('weight', 1.0),
+                'color': prompt.get('color', '#9900ff')
+            } for i, prompt in enumerate(prompts)
+        ]
+
+        # Collect all audio chunks
+        total_duration = 0
+        target_duration = 60  # 1 minute
+        audio_chunks = []
+        session = MockLiveMusicSession(model)
+        session.is_playing = True
+
+        while total_duration < target_duration and session.is_playing:
+            chunk_data = generate_audio_chunk(weighted_prompts, config, total_duration)
+            audio_chunks.append(chunk_data)
+            total_duration += 5 * config.get('slowed_factor', 1.0)
+
+        session.close()
+
+        # Combine chunks and save to WAV
+        pcm_data = b''.join(audio_chunks)
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        output_filename = f"generated_music_{timestamp}.wav"
+        pcm_to_wav(pcm_data, output_filename, sample_rate, channels, bits_per_sample)
+
+        return jsonify({'message': f'Audio file saved as {output_filename}'})
+
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)