import gradio as gr

def answer(user_input):
    base_response = f"""
Halo! Selamat datang di *Gozeal Training Center* 👋

Saya adalah asisten virtual yang siap bantu Anda memahami semua tentang training, program, dan kebutuhan pengembangan skill.

Jawaban untuk pertanyaan Anda:
-------------------------------------
**{user_input}**

Berikut informasi yang mungkin relevan:

"""

    user_input_low = user_input.lower()

    if "kelas" in user_input_low or "training" in user_input_low:
        return base_response + """
📚 Program Training Gozeal:
1. Training Dasar Digital Skill
2. Training Editing & Desain
3. Training Softskill & Pengembangan Diri
4. Training Bisnis & Marketing

Tanyakan: “jadwal training”.
"""
    elif "jadwal" in user_input_low:
        return base_response + """
🗓 Jadwal Training:
- Weekday: Senin – Jumat
- Weekend: Sabtu – Minggu
- Waktu fleksibel sesuai kebutuhan perusahaan

Tanyakan: “harga training”.
"""
    elif "harga" in user_input_low or "biaya" in user_input_low:
        return base_response + """
💰 Harga Training:
- Rp 300.000 – 1.500.000 per peserta
- Corporate: Rp 1.5 – 15 juta per sesi

Bisa custom sesuai kebutuhan!
"""
    elif "kontak" in user_input_low or "hubungi" in user_input_low:
        return base_response + """
📞 Kontak:
- WA: 081388003100
- IG: @gozeal
- Web: https://gozealstore.com/
"""
    else:
        return base_response + """
Kata kunci yang tersedia:
- kelas
- jadwal training
- harga
- kontak
"""

with gr.Blocks() as demo:
    gr.Markdown("# Gozeal Training Center – Virtual Assistant")
    gr.Markdown("Tanyakan apa saja tentang training atau layanan Gozeal!")

    gr.ChatInterface(
        fn=answer,
        title="Gozeal Assistant",
        description="Asisten virtual untuk informasi training & program Gozeal Training Center.",
    )

demo.launch()