Upload 43 files

.gitattributes

@@ -33,3 +33,9 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+assets/examples/garment3.png filter=lfs diff=lfs merge=lfs -text
+assets/examples/model1.png filter=lfs diff=lfs merge=lfs -text
+assets/examples/model2.png filter=lfs diff=lfs merge=lfs -text
+assets/examples/model3.png filter=lfs diff=lfs merge=lfs -text
+assets/examples/result2.png filter=lfs diff=lfs merge=lfs -text
+assets/examples/result3.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,130 @@
+import os
+import gradio as gr
+from gradio_client import Client, handle_file
+from themes import IndonesiaTheme  # Impor tema custom dari themes.py
+from PIL import Image
+
+# Siapkan URL dan header untuk permintaan API
+url_api = os.environ['url_api']
+    
+# Fungsi untuk menyimpan gambar sementara ke file
+def save_temp_image(image_array, filename):
+    image = Image.fromarray(image_array.astype('uint8'))
+    image.save(filename)
+
+# Fungsi untuk memanggil API Virtual Try-On
+def virtual_tryon(person_img, garment_img, seed, randomize_seed):
+    client = Client(url_api)
+    
+    # Simpan gambar sementara
+    person_img_path = "temp_person_img.png"
+    garment_img_path = "temp_garment_img.png"
+    save_temp_image(person_img, person_img_path)
+    save_temp_image(garment_img, garment_img_path)
+    
+    result = client.predict(
+        person_img=handle_file(person_img_path),
+        garment_img=handle_file(garment_img_path),
+        seed=seed,
+        randomize_seed=randomize_seed,
+        api_name="/tryon"
+    )
+    
+    # Hapus file sementara setelah selesai
+    os.remove(person_img_path)
+    os.remove(garment_img_path)
+    
+    return result[0], result[1], result[2]
+
+# Fungsi untuk memuat deskripsi dari file
+def load_description(fp):
+    with open(fp, 'r', encoding='utf-8') as f:
+        content = f.read()
+    return content
+
+# 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 app:
+    # 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 Orang", type="numpy")
+            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=8)
+        
+        with gr.Column(elem_id="col-right"):
+            result_img = gr.Image(label="Hasil", show_share_button=False)
+            seed = gr.Slider(label="Token-Slider", minimum=0, maximum=999999, step=1, value=0)
+            randomize_seed = gr.Checkbox(label="Random Token", value=True)
+            seed_used = gr.Number(label="Token Terpakai")
+            result_info = gr.Textbox(label="Jawaban CS Kami")
+            run_button = gr.Button(value="⭐ Mulai Ganti Pakaian ⭐", elem_id="run-button")
+            run_button.click(fn=virtual_tryon, inputs=[person_img, garment_img, seed, randomize_seed], outputs=[result_img, seed_used, 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__":
+    app.launch()

assets/.gitattributes

@@ -0,0 +1,24 @@
+cloth/09163_00.jpg filter=lfs diff=lfs merge=lfs -text
+cloth/09236_00.jpg filter=lfs diff=lfs merge=lfs -text
+cloth/09263_00.jpg filter=lfs diff=lfs merge=lfs -text
+cloth/09290_00.jpg filter=lfs diff=lfs merge=lfs -text
+cloth/14627_00.jpg filter=lfs diff=lfs merge=lfs -text
+cloth/14673_00.jpg filter=lfs diff=lfs merge=lfs -text
+cloth/04743_00.jpg filter=lfs diff=lfs merge=lfs -text
+cloth/09133_00.jpg filter=lfs diff=lfs merge=lfs -text
+cloth/09166_00.jpg filter=lfs diff=lfs merge=lfs -text
+cloth/dress_1.png filter=lfs diff=lfs merge=lfs -text
+cloth/one-piece_1.png filter=lfs diff=lfs merge=lfs -text
+cloth/upper_cloth_1.png filter=lfs diff=lfs merge=lfs -text
+human/aigc_lady_1.png filter=lfs diff=lfs merge=lfs -text
+human/aigc_lady_2.png filter=lfs diff=lfs merge=lfs -text
+human/aigc_lady_3.png filter=lfs diff=lfs merge=lfs -text
+human/aigc_model_1.png filter=lfs diff=lfs merge=lfs -text
+human/aigc_model_2.png filter=lfs diff=lfs merge=lfs -text
+human/fcee4463c7348028303fa02e18a6efc8a99b2dee351f6b2aef4eca1b.jpg filter=lfs diff=lfs merge=lfs -text
+human/0b70601293997b52b6a44870b29ec5c1.png filter=lfs diff=lfs merge=lfs -text
+human/Jensen.jpg filter=lfs diff=lfs merge=lfs -text
+human/ea37a515d5ff0b240c58251b6042986a.png filter=lfs diff=lfs merge=lfs -text
+human/lekun.png filter=lfs diff=lfs merge=lfs -text
+human/musk.png filter=lfs diff=lfs merge=lfs -text
+human/sam.jpg filter=lfs diff=lfs merge=lfs -text

assets/title.md

@@ -0,0 +1,10 @@
+<div style="display: flex; justify-content: center; align-items: center; text-align: center; font-size: 40px;">
+  <div>
+    <b>Kolors Virtual Try-On in the Wild</b>
+    <br>
+    <div style="display: flex; justify-content: center; align-items: center; text-align: center;">
+        <a href="https://github.com/Kwai-Kolors/Kolors/blob/master/imgs/Kolors_paper.pdf"><img src="https://img.shields.io/static/v1?label=Tech Report&message=Kolors&color=red"></a> &ensp;
+        <a href="https://klingai.kuaishou.com/"><img src="https://img.shields.io/static/v1?label=Official Website&message=Page&color=green"></a>
+    </div>
+  </div>
+</div>

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+gradio_client
+Pillow

themes.py

@@ -0,0 +1,55 @@
+from __future__ import annotations
+from typing import Iterable
+from gradio.themes.base import Base
+from gradio.themes.utils import colors, fonts, sizes
+
+class IndonesiaTheme(Base):
+    def __init__(
+        self,
+        *,
+        primary_hue: colors.Color | str = colors.red,
+        secondary_hue: colors.Color | str = colors.gray,
+        neutral_hue: colors.Color | str = colors.gray,
+        spacing_size: sizes.Size | str = sizes.spacing_md,
+        radius_size: sizes.Size | str = sizes.radius_md,
+        text_size: sizes.Size | str = sizes.text_lg,
+        font: fonts.Font
+        | str
+        | Iterable[fonts.Font | str] = (
+            fonts.GoogleFont("Quicksand"),
+            "ui-sans-serif",
+            "sans-serif",
+        ),
+        font_mono: fonts.Font
+        | str
+        | Iterable[fonts.Font | str] = (
+            fonts.GoogleFont("IBM Plex Mono"),
+            "ui-monospace",
+            "monospace",
+        ),
+    ):
+        super().__init__(
+            primary_hue=primary_hue,
+            secondary_hue=secondary_hue,
+            neutral_hue=neutral_hue,
+            spacing_size=spacing_size,
+            radius_size=radius_size,
+            text_size=text_size,
+            font=font,
+            font_mono=font_mono,
+        )
+        super().set(
+            body_background_fill="linear-gradient(to bottom, #e0e0e0, #7d7d7d)",  # Gradasi abu-abu muda ke abu-abu tua
+            body_background_fill_dark="linear-gradient(to bottom, #7d7d7d, #4a4a4a)",  # Gradasi abu-abu tua ke lebih gelap untuk dark mode
+            button_primary_background_fill="linear-gradient(90deg, #d84a4a, #b33030)",  # Merah ke merah tua
+            button_primary_background_fill_hover="linear-gradient(90deg, #e85b5b, #cc4b4b)",  # Merah lebih terang untuk hover
+            button_primary_text_color="white",
+            button_primary_background_fill_dark="linear-gradient(90deg, #b33030, #8f1f1f)",  # Merah tua untuk dark mode
+            slider_color="*secondary_300",
+            slider_color_dark="*secondary_600",
+            block_title_text_weight="600",
+            block_border_width="3px",
+            block_shadow="*shadow_drop_lg",
+            button_shadow="*shadow_drop_lg",
+            button_large_padding="32px",
+        )