Update demo to get actor_id

app.py

@@ -1,55 +1,70 @@
 import gradio as gr
 import requests
-import os  # 환경 변수 사용을 위해 필요
+import os
+import json
 
-# Secrets에서 Private Space URL과 Access Token 읽기
 PRIVATE_SPACE_URL = os.getenv("PRIVATE_SPACE_URL")
 HF_TOKEN = os.getenv("HF_TOKEN")
 
 def fetch_actor_data():
     if not PRIVATE_SPACE_URL:
-        return "Error: Private Space URL not found. Please check the Secrets configuration."
+        return "Error: PRIVATE_SPACE_URL not found. Check Secrets."
     if not HF_TOKEN:
-        return "Error: Access token not found. Please check the Secrets configuration."
+        return "Error: HF_TOKEN not found. Check Secrets."
 
     headers = {
         "Authorization": f"Bearer {HF_TOKEN}"
     }
 
     try:
-        # Private Space의 /api/actor 엔드포인트로 요청
-        url = f"{PRIVATE_SPACE_URL}/api/actor"
-        response = requests.get(url, headers=headers)
+        response = requests.get(f"{PRIVATE_SPACE_URL}/api/actor", headers=headers)
 
         if response.status_code == 200:
-            # 성공적으로 데이터를 가져온 경우
+            # 응답 JSON 파싱
             data = response.json()
-
-            # actor_id 추출
+            
+            # 기대 구조: data["data"]["result"] => list of actor items
             actor_ids = []
-
-            # JSON 구조를 순회하며 actor_id 수집
-            if "results" in data:
-                for item in data["results"]:
+            if (
+                "data" in data
+                and isinstance(data["data"], dict)
+                and "result" in data["data"]
+                and isinstance(data["data"]["result"], list)
+            ):
+                for item in data["data"]["result"]:
                     if "actor_id" in item:
                         actor_ids.append(item["actor_id"])
+                    if len(actor_ids) >= 10:
+                        break
 
-            return f"Actor ID: {actor_ids}"
+            if actor_ids:
+                return f"Found actor_ids: {actor_ids}"
+            else:
+                # actor_id가 없을 경우 구조 파악용 반환
+                top_level_keys = list(data["data"].keys()) if "data" in data and isinstance(data["data"], dict) else []
+                first_item = data["data"]["result"][0] if "data" in data and "result" in data["data"] and len(data["data"]["result"]) > 0 else {}
+                return (
+                    "No actor_id found.\n\n"
+                    f"Keys in data['data']: {top_level_keys}\n\n"
+                    "First item in data['data']['result']:\n" +
+                    json.dumps(first_item, indent=2, ensure_ascii=False)
+                )
         else:
-            # 실패한 경우 상태 코드와 에러 메시지 반환
-            return f"Error {response.status_code}: {response.text}"
+            return f"Error {response.status_code}\nResponse Text:\n{response.text}"
     except Exception as e:
         return f"Request failed: {str(e)}"
 
-# Gradio 인터페이스 정의
 iface = gr.Interface(
     fn=fetch_actor_data,
-    inputs=None,  # 입력값 필요 없음
-    outputs="text",  # 텍스트 결과 출력
+    inputs=None,
+    outputs="text",
     title="Fetch Actor Data",
-    description="Fetches actor data from the private Hugging Face Space using the /api/actor endpoint."
+    description=(
+        "Fetches actor data from the private Hugging Face Space.\n"
+        "Assumes structure: {status_code:..., data:{count:..., result:[{actor_id:...}]}}\n"
+        "If no actor_id found, shows partial keys for debugging."
+    )
 )
 
-# 앱 실행
 if __name__ == "__main__":
     iface.launch()