import gradio as gr

RESPONSES = {
    "hello": "How far! Wetin dey happen?",
    "hi": "Ehen! Na so. How you dey?",
    "jollof": "Ah! Nigerian jollof na the best, no cap!",
    "lagos": "Lagos! The city wey never sleep!",
    "music": "Afrobeats don take over the world!",
    "food": "Nigerian food na the sweetest!",
}

def respond(message, history):
    msg = message.lower()
    for key, response in RESPONSES.items():
        if key in msg:
            return response
    return "Wetin else you wan yarn? Ask about Lagos, jollof, or music!"

demo = gr.ChatInterface(
    fn=respond,
    title="Sisi Lola AI",
    description="Chat with Nigeria virtual content creator!",
    examples=["Hello!", "Tell me about Lagos", "What about jollof?"],
)

if __name__ == "__main__":
    demo.launch(server_name="0.0.0.0", server_port=7860)