Update server.py

server.py

@@ -82,19 +82,31 @@ def _predict_from_removebg_api(image_blob: bytes):
     print("✅ Sukses menggunakan remove.bg API.")
     return StreamingResponse(io.BytesIO(response.content), media_type="image/png")
 
+# GANTI SELURUH FUNGSI LAMA DENGAN YANG INI
+
 def _predict_from_gradio_space(local_path: str, space_name: str, api_name: str):
     """Fungsi yang direfaktor untuk memanggil Gradio Space manapun dengan api_name dinamis."""
     print(f"Mencoba Prioritas Fallback: Gradio Space '{space_name}' dengan API '{api_name}'...")
     client = Client(src=space_name)
-    result = client.predict(image=handle_file(local_path), api_name=api_name) # <-- Gunakan variabel api_name
+    result = client.predict(image=handle_file(local_path), api_name=api_name)
     
-    # Logika untuk mengambil hasil gambar dari output Gradio
-    out_any = result[1] if isinstance(result, (list, tuple)) and len(result) >= 2 else result
-    if isinstance(out_any, str) and os.path.exists(out_any):
-        with open(out_any, "rb") as f:
+    # --- LOGIKA BARU UNTUK MEMBACA HASIL ---
+    image_path = None
+    # Jika hasilnya adalah list atau tuple (seperti di Mystyc Space yang baru)
+    if isinstance(result, (list, tuple)) and len(result) > 0:
+        # Ambil elemen PERTAMA (indeks 0) sebagai path gambar
+        image_path = result[0]
+    # Jika hasilnya hanya string (seperti di Gradio Space yang lama)
+    elif isinstance(result, str):
+        image_path = result
+
+    # Lanjutkan proses jika path gambar valid
+    if image_path and isinstance(image_path, str) and os.path.exists(image_path):
+        with open(image_path, "rb") as f:
             print(f"✅ Sukses menggunakan Gradio Space '{space_name}'.")
             return StreamingResponse(io.BytesIO(f.read()), media_type="image/png")
     
+    # Jika format tidak dikenali sama sekali
     raise ValueError(f"Output dari Gradio Space '{space_name}' tidak dikenali: {repr(result)}")
 
 # --- ENDPOINTS ---