Update app.py

app.py

@@ -5,9 +5,20 @@ from themes import IndonesiaTheme  # Impor tema custom dari themes.py
 from PIL import Image
 import random
 import tempfile  # Untuk membuat file sementara
+import time  # Untuk fungsi penundaan
 
-# Siapkan URL dan header untuk permintaan API
+# 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):
@@ -18,7 +29,10 @@ def save_temp_image(image_array):
 
 # 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):
-    client = Client(url_api)
+    
+    # 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)
@@ -56,11 +70,16 @@ def virtual_tryon(person_img, garment_img, garment_des, seed, randomize_seed, de
         "Gaya Anda sangat menakjubkan dengan pakaian ini!",
         "Perfect match! Anda terlihat menawan!",
         "Siap untuk sesi foto? Anda benar-benar keren!"
-        ]
+    ]
 
-    # Ubah bagian return dengan memilih kalimat secara acak
+    # Mengembalikan hasil dan pesan pujian acak
     return result[0], random.choice(compliments), "Proses selesai."
 
+# Fungsi untuk menampilkan pesan popup dan menunda eksekusi
+def show_popup_and_wait():
+    gr.alert("Ruang ganti kami terbatas. Mohon menunggu antreannya.")
+    time.sleep(60)  # Menunggu selama 1 menit
+
 # Path untuk contoh gambar
 example_path = os.path.join(os.path.dirname(__file__), 'assets')
 
@@ -72,7 +91,6 @@ garm_list_path = [os.path.join(example_path, "cloth", garm) for garm in garm_lis
 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 {
@@ -131,6 +149,7 @@ with gr.Blocks(css=css, theme=IndonesiaTheme()) as TokoBajuVirtual:
             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)
@@ -144,7 +163,13 @@ with gr.Blocks(css=css, theme=IndonesiaTheme()) as TokoBajuVirtual:
             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]  # Menghapus string "Hasil"
+                outputs=[result_img, result_info]
+            )
+            # Tambahkan penundaan setelah klik dan pesan popup
+            run_button.click(
+                fn=show_popup_and_wait,  # Menampilkan popup dan menunda eksekusi
+                inputs=[], 
+                outputs=[]
             )
     
     # Tambahkan footer di bagian bawah