"""
Aplikasi Toko Baju Virtual - Virtual Try-On

Deskripsi:
Aplikasi ini memungkinkan pengguna untuk mencoba pakaian secara virtual dengan 
mengunggah gambar diri mereka sendiri dan gambar pakaian yang ingin dicoba. 
Menggunakan teknologi **Virtual Try-On** yang dipanggil melalui API eksternal untuk 
menghasilkan gambar realistis dari pengguna yang mengenakan pakaian yang dipilih.

Oleh : 
©2024 __drat. | 🇮🇩 Untuk Indonesia Jaya!

"""

import os
import gradio as gr
from gradio_client import Client, handle_file
from gradio_client.exceptions import AppError  # Import untuk menangkap kesalahan Gradio Client
from themes import IndonesiaTheme
from PIL import Image
import random
import tempfile
import time

# Siapkan URL untuk permintaan API Virtual Try-On
url_api = os.environ['url_api']
url_api2 = os.environ['url_api2']
url_api3 = os.environ['url_api3']
url_api4 = os.environ['url_api4']
url_api5 = os.environ['url_api5']
url_api6 = os.environ['url_api6']
url_api7 = os.environ['url_api7']

# Fungsi untuk memilih URL secara acak
def get_random_api_url():
    return random.choice([url_api, url_api2, url_api3, url_api4, url_api5, url_api6, url_api7])

# Fungsi untuk menyimpan gambar sementara ke file
def save_temp_image(image_array):
    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
    image = Image.fromarray(image_array.astype('uint8'))
    image.save(temp_file.name)
    return temp_file.name

# Fungsi untuk memanggil API Virtual Try-On
def virtual_tryon(person_img, garment_img, garment_des, seed, randomize_seed, denoise_steps, is_checked, is_checked_crop):
    # Pilih URL secara acak
    selected_api_url = get_random_api_url()
    client = Client(selected_api_url)
    
    # Simpan gambar sementara
    person_img_path = save_temp_image(person_img)
    garment_img_path = save_temp_image(garment_img)
    
    try:
        # Pemanggilan API yang sudah disesuaikan dengan format baru
        result = client.predict(
            dict={
                "background": handle_file(person_img_path),
                "layers": [],
                "composite": None  # Menggunakan None sesuai dengan format yang diberikan
            },
            garm_img=handle_file(garment_img_path),
            garment_des=garment_des,  # Deskripsi pakaian dari input pengguna
            is_checked=is_checked,  # Checkbox apakah checked atau tidak
            is_checked_crop=is_checked_crop,  # Checkbox untuk cropping
            denoise_steps=denoise_steps,  # Denoise steps slider
            seed=seed,
            api_name="/tryon"
        )

        # Daftar kalimat acak yang menarik dan lucu
        compliments = [
            "Anda tampak hebat dengan baju itu!",
            "Wow, Anda terlihat sangat memukau sekarang!",
            "Siap jadi pusat perhatian dengan outfit ini!",
            "Anda terlihat seperti bintang di red carpet!",
            "Baju ini benar-benar cocok dengan gaya Anda!",
            "Luar biasa! Anda terlihat seperti seorang model profesional!",
            "Siapa yang tidak terpesona melihat Anda sekarang?",
            "Gaya Anda sangat menakjubkan dengan pakaian ini!",
            "Perfect match! Anda terlihat menawan!",
            "Siap untuk sesi foto? Anda benar-benar keren!"
        ]
        
        # Tampilkan hasil dan kalimat acak yang menarik
        return result[0], random.choice(compliments), "Proses selesai."

    except AppError as e:
        # Tampilkan pesan error di result_info jika terjadi kesalahan API
        return None, str(e), "Terjadi kesalahan selama proses."

    finally:
        # Hapus file sementara setelah selesai
        os.remove(person_img_path)
        os.remove(garment_img_path)
    
    # Tampilkan pesan peringatan dan menunggu selama 1 menit sebelum proses baru dimulai
    gr.Warning("Ruang ganti kami terbatas. Mohon menunggu antreannya.")
    time.sleep(60)


# Path untuk contoh gambar
example_path = os.path.join(os.path.dirname(__file__), 'assets')

# Daftar gambar pakaian contoh
garm_list = os.listdir(os.path.join(example_path, "cloth"))
garm_list_path = [os.path.join(example_path, "cloth", garm) for garm in garm_list]

# Daftar gambar orang contoh
human_list = os.listdir(os.path.join(example_path, "human"))
human_list_path = [os.path.join(example_path, "human", human) for human in human_list]

# CSS untuk styling antarmuka
css = """
#col-left, #col-mid, #col-right {
    margin: 0 auto;
    max-width: 400px;
    padding: 10px;
    border-radius: 15px;
    background-color: #f9f9f9;
    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1);
}
#banner {
    width: 100%;
    text-align: center;
    margin-bottom: 20px;
}
#run-button {
    background-color: #ff4b5c;
    color: white;
    font-weight: bold;
    padding: 10px;
    border-radius: 10px;
    cursor: pointer;
    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.2);
}
#footer {
    text-align: center;
    margin-top: 20px;
    color: silver;
}
"""

# Membuat antarmuka Gradio dengan tema IndonesiaTheme
with gr.Blocks(css=css, theme=IndonesiaTheme()) as TokoBajuVirtual:
    # Tambahkan banner
    gr.HTML("""
        <div style='text-align: center;'>
            <img src='https://i.ibb.co.com/BgBLhSW/banner.jpg' alt='Banner' style='width: 100%; height: auto;'/>
        </div>
    """)

    # Ganti judul
    gr.Markdown("<h1 style='text-align: center;'>Selamat Datang Pelanggan Tercinta!❤️ Senang dapat melayani Anda.. </h1>")
    
    # Layout utama
    with gr.Row():
        with gr.Column(elem_id="col-left"):
            gr.Markdown("### Unggah Foto Anda")
            person_img = gr.Image(label="Foto Anda", type="numpy")
            gr.Markdown("### Deskripsi Pakaian")
            garment_des = gr.Textbox(label="Deskripsi Pakaian", placeholder="Optional : Berikan deskripsi pakaian...")
            gr.Examples(inputs=person_img, examples=human_list_path, examples_per_page=8)
        
        with gr.Column(elem_id="col-mid"):
            gr.Markdown("### Unggah Foto Pakaian")
            garment_img = gr.Image(label="Foto Pakaian", type="numpy")
            gr.Examples(inputs=garment_img, examples=garm_list_path, examples_per_page=14)
        
        with gr.Column(elem_id="col-right"):
            gr.Markdown("### Ruang Ganti Umum")
            result_img = gr.Image(label="Hasil", show_share_button=False)
            seed = gr.Slider(label="Token-Slider", minimum=0, maximum=999999, step=1, value=42)
            randomize_seed = gr.Checkbox(label="Random Token", value=True)
            denoise_steps = gr.Slider(label="Denoise Steps", minimum=1, maximum=50, step=1, value=30)
            is_checked = gr.Checkbox(label="Nyalakan Virtual Try-On", value=True)
            is_checked_crop = gr.Checkbox(label="Ukuran Disesuaikan", value=True)
            result_info = gr.Textbox(label="Jawaban CS Kami")
            
            # Tombol untuk memulai proses ganti pakaian virtual
            run_button = gr.Button(value="⭐ Mulai Ganti Pakaian ⭐", elem_id="run-button")
            run_button.click(
                fn=virtual_tryon, 
                inputs=[person_img, garment_img, garment_des, seed, randomize_seed, denoise_steps, is_checked, is_checked_crop], 
                outputs=[result_img, result_info]
            )
    
    # Tambahkan footer di bagian bawah
    gr.HTML("""
    <footer id="footer">
        Transfer Energi Semesta Digital © 2024 __drat. | 🇮🇩 Untuk Indonesia Jaya!
    </footer>
    """)

# Menjalankan aplikasi
if __name__ == "__main__":
    TokoBajuVirtual.queue(api_open=False).launch(show_api=False)