import gradio as gr
from transformers import pipeline

pipe = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta")

def generate_lyrics(prompt):
    # Prompt'u düzenle
    full_prompt = f"<|system|>\nYou are a helpful assistant for generating song lyrics.\n<|user|>\n{prompt}\n<|assistant|>\n"
    try:
        response = pipe(full_prompt, max_new_tokens=150, do_sample=True, temperature=0.7)[0]['generated_text']
        if "<|assistant|>" in response:
            response = response.split("<|assistant|>")[1].strip()
        return response
    except Exception as e:
        return f"Error: {str(e)}"

# API endpoint
gr.Interface(
    fn=generate_lyrics,
    inputs="text",
    outputs="text",
    title="Lyrics AI",
    description="Enter a prompt to generate lyrics"
).launch()