import gradio as gr
from transformers import pipeline

# NotaGen modelini yükle
generator = pipeline('text-generation', model='ElectricAlexis/NotaGen-large')

# Müzik üretme fonksiyonu
def generate_music(prompt):
    output = generator(prompt, max_length=100, num_return_sequences=1)
    return output[0]['generated_text']

# Gradio arayüzü oluştur
interface = gr.Interface(
    fn=generate_music,
    inputs="text",
    outputs="text",
    title="NotaGen ile Müzik Oluşturma",
    description="Bir müzik tarzı girin (örneğin, 'Barok piyano parçası') ve notaları üretin!"
)

# Arayüzü başlat
interface.launch()