ae_api.py

import requests
import os

# Configuration
AUDIO_ENGINE_URL = "http://127.0.0.1:6000/generate"

def request_music(prompt, duration=10):
    """
    Call this from your GUI button.
    """
    payload = {
        "prompt": prompt,
        "duration": duration
    }

    try:
        print(f"Requesting audio: {prompt}...")
        response = requests.post(AUDIO_ENGINE_URL, json=payload, timeout=60)

        if response.status_code == 200:
            data = response.json()
            if data['status'] == 'SUCCESS':
                print(f"Audio Ready: {data['file']}")
                return data['file'] # Returns the path to the .wav
            else:
                print("Engine Error:", data.get('message'))
        else:
            print("Engine Unreachable")

    except Exception as e:
        print(f"Connection Failed. Is audio_engine.py running? {e}")
        return None

# --- TEST (Run this line when you click your GUI button) ---
# file_path = request_music("Dark cyberpunk bassline", 5)
# if file_path: play_sound(file_path)