import gradio as gr
import random
import requests
import os

def generate_music(prompt):
    prompt = prompt.lower()
    music = [
        "https://www.free-stock-music.com/music/alex-productions-christmas-snow.mp3",
        "https://www.free-stock-music.com/music/alex-productions-christmas-is-coming.mp3",
        "https://www.free-stock-music.com/music/maxkomusic-christmas-is-coming.mp3"
    ]
    output_file_path = None
    if "music" in prompt:
        audio_file = requests.get(random.choice(music))
        file_name = "generated_music.mp3"
        output_file_path = os.path.join("/tmp", file_name)
        with open(output_file_path, 'wb') as f:
            f.write(audio_file.content)

    return output_file_path

# Creating Gradio interface
ai = gr.Interface(
    fn=generate_music,
    inputs=gr.Textbox(lines=2, label="Enter your prompt"),
    outputs=gr.Audio(type="filepath", label="Generated music"),
    title="KVIMusicist",
    description="An AI that provides music based on your prompt."
)

ai.launch()