Update app.py

app.py

@@ -3,13 +3,9 @@ from huggingface_hub import InferenceClient
 import os
 import requests
 from typing import List, Dict, Union
-import concurrent.futures
 import traceback
 
-# 환경 변수에서 토큰 가져오기
 HF_TOKEN = os.getenv("HF_TOKEN")
-
-# 추론 API 클라이언트 설정
 hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=HF_TOKEN)
 
 def get_headers():
@@ -29,52 +25,28 @@ def get_most_liked_spaces(limit: int = 100) -> Union[List[Dict], str]:
     try:
         response = requests.get(url, params=params, headers=get_headers())
         response.raise_for_status()
-        data = response.json()
-        
-        if isinstance(data, list):
-            return data
-        else:
-            return f"Unexpected API response format: {type(data)}"
+        return response.json()
     except requests.RequestException as e:
         return f"API request error: {str(e)}"
     except ValueError as e:
         return f"JSON decoding error: {str(e)}"
 
-def capture_thumbnail(space_id: str) -> bytes:
-    screenshot_url = f"https://huggingface.co/spaces/{space_id}/thumbnail.png"
-    try:
-        response = requests.get(screenshot_url, headers=get_headers())
-        if response.status_code == 200:
-            return response.content
-    except requests.RequestException:
-        pass
-    return None
-
-
-def format_spaces(spaces: Union[List[Dict], str]) -> List[Dict]:
-    if isinstance(spaces, str):
-        return [{"error": spaces}]
-    
-    valid_spaces = []
-    for space in spaces:
-        if isinstance(space, dict):
-            valid_spaces.append(space)
-        else:
-            print(f"Skipping invalid space: {space}")
+def get_hardware_requirements(space: Dict) -> str:
+    sdk = space.get('sdk', {})
+    if isinstance(sdk, dict):
+        hardware = sdk.get('hardware')
+        if hardware:
+            return hardware
     
-    formatted = []
-    for space in valid_spaces:
-        try:
-            formatted.append(format_space(space))
-        except Exception as e:
-            print(f"Error formatting space: {space.get('id', 'Unknown')}. Error: {str(e)}")
+    host_requirements = space.get('host_requirements', {})
+    if isinstance(host_requirements, dict):
+        hardware = host_requirements.get('hardware', {})
+        if isinstance(hardware, dict):
+            return 'CPU' if hardware.get('gpu') is False else 'GPU'
     
-    return formatted
+    return 'Unknown'
 
 def format_space(space: Dict) -> Dict:
-    if not isinstance(space, dict):
-        raise ValueError(f"Invalid space format: {space}")
-    
     space_id = space.get('id', 'Unknown')
     space_name = space_id.split('/')[-1] if '/' in space_id else space_id
     
@@ -85,7 +57,7 @@ def format_space(space: Dict) -> Dict:
     space_likes = space.get('likes', 'N/A')
     space_url = f"https://huggingface.co/spaces/{space_id}"
     
-    thumbnail = capture_thumbnail(space_id)
+    thumbnail_url = f"https://huggingface.co/spaces/{space_id}/resolve/main/thumbnail.png"
     hardware = get_hardware_requirements(space)
     
     return {
@@ -94,32 +66,30 @@ def format_space(space: Dict) -> Dict:
         "author": space_author,
         "likes": space_likes,
         "url": space_url,
-        "thumbnail": thumbnail,
+        "thumbnail": thumbnail_url,
         "hardware": hardware
     }
 
-def get_hardware_requirements(space: Dict) -> str:
-    if not isinstance(space, dict):
-        return "Unknown"
+def format_spaces(spaces: Union[List[Dict], str]) -> List[Dict]:
+    if isinstance(spaces, str):
+        return [{"error": spaces}]
     
-    sdk = space.get('sdk', {})
-    if not isinstance(sdk, dict):
-        return "Unknown"
+    return [format_space(space) for space in spaces if isinstance(space, dict)]
+
+def summarize_space(space: Dict) -> str:
+    system_message = "당신은 Hugging Face Space의 내용을 요약하는 AI 조수입니다. 주어진 정보를 바탕으로 간결하고 명확한 요약을 제공해주세요."
+    user_message = f"다음 Hugging Face Space를 요약해주세요: {space['name']} by {space['author']}. 좋아요 수: {space['likes']}. URL: {space['url']}"
     
-    hf = sdk.get('hf', {})
-    if not isinstance(hf, dict):
-        return "Unknown"
+    messages = [
+        {"role": "system", "content": system_message},
+        {"role": "user", "content": user_message}
+    ]
     
-    sdk_version = hf.get('version')
-    if sdk_version and int(sdk_version.split('.')[0]) >= 3:
-        return sdk.get('hardware', 'Unknown')
-    else:
-        host_requirements = space.get('host_requirements', {})
-        if isinstance(host_requirements, dict):
-            hardware = host_requirements.get('hardware', {})
-            if isinstance(hardware, dict):
-                return 'CPU' if hardware.get('gpu') is False else 'GPU'
-    return 'Unknown'
+    try:
+        response = hf_client.chat_completion(messages, max_tokens=150, temperature=0.7)
+        return response.choices[0].message.content
+    except Exception as e:
+        return f"요약 생성 중 오류 발생: {str(e)}"
 
 def get_app_py_content(space_id: str) -> str:
     app_py_url = f"https://huggingface.co/spaces/{space_id}/raw/main/app.py"
@@ -132,51 +102,34 @@ def get_app_py_content(space_id: str) -> str:
     except requests.RequestException:
         return f"Error fetching app.py content for space: {space_id}"
 
-def summarize_space(space: Dict) -> str:
-    system_message = "당신은 Hugging Face Space의 내용을 요약하는 AI 조수입니다. 주어진 정보를 바탕으로 간결하고 명확한 요약을 제공해주세요."
-    user_message = f"다음 Hugging Face Space를 요약해주세요: {space['name']} by {space['author']}. 좋아요 수: {space['likes']}. URL: {space['url']}"
-    
-    messages = [
-        {"role": "system", "content": system_message},
-        {"role": "user", "content": user_message}
-    ]
-    
-    response = hf_client.chat_completion(messages, max_tokens=150, temperature=0.7)
-    return response.choices[0].message.content
+def on_select(space):
+    try:
+        summary = summarize_space(space)
+        app_content = get_app_py_content(space['id'])
+        info = f"선택된 Space: {space['name']} (ID: {space['id']})\n"
+        info += f"Author: {space['author']}\n"
+        info += f"Likes: {space['likes']}\n"
+        info += f"Hardware: {space['hardware']}\n"
+        info += f"URL: {space['url']}\n\n"
+        info += f"요약:\n{summary}"
+        return info, app_content
+    except Exception as e:
+        print(f"Error in on_select: {str(e)}")
+        print(traceback.format_exc())
+        return f"오류가 발생했습니다: {str(e)}", ""
 
 def create_ui():
     spaces_list = get_most_liked_spaces()
     print(f"Type of spaces_list: {type(spaces_list)}")
-    if isinstance(spaces_list, str):
-        print(f"Error in get_most_liked_spaces: {spaces_list}")
-        return gr.Blocks().launch()
-    
     formatted_spaces = format_spaces(spaces_list)
-    print(f"Total spaces loaded: {len(formatted_spaces)}")  # 디버깅 출력
+    print(f"Total spaces loaded: {len(formatted_spaces)}")
 
     css = """
-    footer {
-        visibility: hidden;
-    }
-    .minimal-button {
-        min-width: 30px !important;
-        height: 25px !important;
-        line-height: 1 !important;
-        font-size: 12px !important;
-        padding: 2px 5px !important;
-    }
-    .space-row {
-        margin-bottom: 5px !important;
-    }
-    .thumbnail {
-        width: 100px;
-        height: 100px;
-        object-fit: cover;
-    }
-    .hardware-info {
-        font-size: 12px;
-        color: #666;
-    }
+    footer {visibility: hidden;}
+    .minimal-button {min-width: 30px !important; height: 25px !important; line-height: 1 !important; font-size: 12px !important; padding: 2px 5px !important;}
+    .space-row {margin-bottom: 5px !important;}
+    .thumbnail {width: 100px; height: 100px; object-fit: cover;}
+    .hardware-info {font-size: 12px; color: #666;}
     """
 
     with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
@@ -187,10 +140,7 @@ def create_ui():
                 space_rows = []
                 for space in formatted_spaces:
                     with gr.Row(elem_classes="space-row") as space_row:
-                        if space['thumbnail']:
-                            gr.Image(value=space['thumbnail'], width=100, height=100, show_label=False, elem_classes="thumbnail")
-                        else:
-                            gr.Image(value="https://huggingface.co/front/assets/huggingface_logo-noborder.svg", width=100, height=100, show_label=False, elem_classes="thumbnail")
+                        gr.Image(value=space['thumbnail'], width=100, height=100, show_label=False, elem_classes="thumbnail")
                         with gr.Column():
                             gr.Markdown(f"{space['name']} by {space['author']} (Likes: {space['likes']})", elem_classes="space-info")
                             gr.Markdown(f"Hardware: {space['hardware']}", elem_classes="hardware-info")