Update app.py

app.py

@@ -1,3 +1,18 @@
+"""
+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
@@ -6,6 +21,7 @@ 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
 
 # Siapkan URL untuk permintaan API Virtual Try-On
 url_api = os.environ['url_api']
@@ -27,9 +43,9 @@ 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)
@@ -38,42 +54,53 @@ def virtual_tryon(person_img, garment_img, garment_des, seed, randomize_seed, de
     person_img_path = save_temp_image(person_img)
     garment_img_path = save_temp_image(garment_img)
     
-    # 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"
-    )
-    
-    # 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!",
-        "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!"
-    ]
-
-    # Mengembalikan hasil dan pesan pujian acak
-    return result[0], random.choice(compliments), "Proses selesai."
+    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"
+        )
+        
+        # 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!",
+            "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!"
+        ]
+
+        # Ubah bagian return dengan memilih kalimat secara acak
+        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
+
 
 # Fungsi untuk menampilkan pesan popup dan menunda eksekusi
 def show_popup_and_wait():