Update app.py

app.py

@@ -1,40 +1,73 @@
 import os
 import gradio as gr
-from gradio_client import Client, handle_file
-from themes import IndonesiaTheme  # Impor tema custom dari themes.py
+import requests
+import json
 from PIL import Image
+import tempfile
+
+# Siapkan URL untuk permintaan API
+api_url = "https://kwai-kolors-kolors-virtual-try-on.hf.space/queue/join?__theme=system"
 
-# 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):
+def save_temp_image(image_array):
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
     image = Image.fromarray(image_array.astype('uint8'))
-    image.save(filename)
+    image.save(temp_file.name)
+    return temp_file.name
 
 # 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)
+    person_img_path = save_temp_image(person_img)
+    garment_img_path = save_temp_image(garment_img)
+    
+    payload = json.dumps({
+        "data": [
+            {
+                "path": person_img_path,
+                "url": f"https://deddy-toko-baju-virtual-gratis.hf.space/file={person_img_path}",
+                "size": None,
+                "orig_name": os.path.basename(person_img_path),
+                "mime_type": None,
+                "is_stream": False,
+                "meta": {
+                    "_type": "gradio.FileData"
+                }
+            },
+            {
+                "path": garment_img_path,
+                "url": f"https://deddy-toko-baju-virtual-gratis.hf.space/file={garment_img_path}",
+                "size": None,
+                "orig_name": os.path.basename(garment_img_path),
+                "mime_type": None,
+                "is_stream": False,
+                "meta": {
+                    "_type": "gradio.FileData"
+                }
+            },
+            seed,
+            randomize_seed
+        ],
+        "event_data": None,
+        "fn_index": 2,  # Indeks API yang sesuai
+        "trigger_id": 26,
+        "session_hash": "l8hba3fqoe8"  # Hash session harus diperbarui jika berubah
+    })
     
-    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"
-    )
+    headers = {
+        'content-type': 'application/json',
+        'User-Agent': 'Mozilla/5.0'
+    }
     
-    # Hapus file sementara setelah selesai
-    os.remove(person_img_path)
-    os.remove(garment_img_path)
+    # Kirim permintaan API
+    response = requests.post(api_url, headers=headers, data=payload)
     
-    return result[0], result[1], result[2]
+    if response.status_code == 200:
+        result = response.json()
+        # Mengambil hasil dari API
+        return result['data'][0], seed, "Proses selesai."
+    else:
+        return None, seed, f"Error: {response.status_code}"
 
 # Fungsi untuk memuat deskripsi dari file
 def load_description(fp):