Update app.py

Add alert and info

app.py

@@ -12,16 +12,14 @@ Oleh :
 
 """
 
-
 import os
 import gradio as gr
 from gradio_client import Client, handle_file
-from themes import IndonesiaTheme  # Impor tema custom dari themes.py
+from themes import IndonesiaTheme
 from PIL import Image
 import random
-import tempfile  # Untuk membuat file sementara
-import time  # Untuk fungsi penundaan
-from gradio_client.exceptions import AppError  # Import untuk menangkap kesalahan Gradio Client
+import tempfile
+import time
 
 # Siapkan URL untuk permintaan API Virtual Try-On
 url_api = os.environ['url_api']
@@ -43,12 +41,11 @@ def save_temp_image(image_array):
     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)
+    # Pilih API secara acak antara dua URL
+    url_api = random.choice(["Nymbo/Virtual-Try-On", "yisol/IDM-VTON"])
+    client = Client(url_api)
     
     # Simpan gambar sementara
     person_img_path = save_temp_image(person_img)
@@ -70,11 +67,7 @@ def virtual_tryon(person_img, garment_img, garment_des, seed, randomize_seed, de
             seed=seed,
             api_name="/tryon"
         )
-        
-        # Hapus file sementara setelah selesai
-        os.remove(person_img_path)
-        os.remove(garment_img_path)
-        
+
         # Daftar kalimat acak yang menarik dan lucu
         compliments = [
             "Anda tampak hebat dengan baju itu!",
@@ -88,18 +81,22 @@ def virtual_tryon(person_img, garment_img, garment_des, seed, randomize_seed, de
             "Perfect match! Anda terlihat menawan!",
             "Siap untuk sesi foto? Anda benar-benar keren!"
         ]
-
-        # Ubah bagian return dengan memilih kalimat secara acak
+        
+        # Tampilkan hasil dan kalimat acak yang menarik
         return result[0], random.choice(compliments), "Proses selesai."
 
-    except AppError as e:
-        # Tangkap AppError dan tampilkan pesan kesalahan kepada pengguna
-        error_message = str(e)
-        return None, error_message  # Menampilkan pesan kesalahan di `result_info`
-    except Exception as e:
-        # Tangkap kesalahan umum dan tampilkan pesan kesalahan kepada pengguna
-        error_message = f"Terjadi kesalahan: {str(e)}"
-        return None, error_message
+    except gradio_client.exceptions.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)
 
 
 # Fungsi untuk menampilkan pesan popup dan menunda eksekusi