Update app-backup4.py

app-backup4.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():
@@ -17,7 +13,7 @@ def get_headers():
         raise ValueError("Hugging Face token not found in environment variables")
     return {"Authorization": f"Bearer {HF_TOKEN}"}
 
-def get_most_liked_spaces(limit: int = 100) -> Union[List[Dict], str]:
+def get_most_liked_spaces(limit: int = 300) -> Union[List[Dict], str]:
     url = "https://huggingface.co/api/spaces"
     params = {
         "sort": "likes",
@@ -29,12 +25,7 @@ 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:
@@ -56,15 +47,29 @@ def format_space(space: Dict) -> Dict:
         "name": space_name,
         "author": space_author,
         "likes": space_likes,
-        "url": space_url
+        "url": space_url,
     }
 
 def format_spaces(spaces: Union[List[Dict], str]) -> List[Dict]:
     if isinstance(spaces, str):
         return [{"error": spaces}]
     
-    with concurrent.futures.ThreadPoolExecutor(max_workers=10) as executor:
-        return list(executor.map(format_space, spaces))
+    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']}"
+    
+    messages = [
+        {"role": "system", "content": system_message},
+        {"role": "user", "content": user_message}
+    ]
+    
+    try:
+        response = hf_client.chat_completion(messages, max_tokens=400, 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"
@@ -77,56 +82,56 @@ 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"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)}")
     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;}
+    """
 
-    with gr.Blocks() as demo:
+    with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
         gr.Markdown("# Hugging Face Most Liked Spaces")
         
         with gr.Row():
             with gr.Column(scale=1):
-                space_buttons = []
+                space_rows = []
                 for space in formatted_spaces:
-                    with gr.Row():
-                        gr.Markdown(f"{space['name']} by {space['author']} (Likes: {space['likes']})")
-                        button = gr.Button("클릭", elem_id=f"btn-{space['id']}")
-                        space_buttons.append(button)
+                    with gr.Row(elem_classes="space-row") as space_row:
+                        with gr.Column():
+                            gr.Markdown(f"{space['name']} by {space['author']} (Likes: {space['likes']})", elem_classes="space-info")
+                            button = gr.Button("클릭", elem_classes="minimal-button")
+                    space_rows.append((space_row, button, space))
             
             with gr.Column(scale=1):
-                info_output = gr.Textbox(label="Space 정보 및 요약", lines=10)
+                info_output = gr.Textbox(label="Space 정보 및 요약", lines=16)
                 app_py_content = gr.Code(language="python", label="app.py 내용")
 
-        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"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)}", ""
-
-        for button, space in zip(space_buttons, formatted_spaces):
-            button.click(on_select, inputs=[gr.State(space)], outputs=[info_output, app_py_content])
+        for _, button, space in space_rows:
+            button.click(
+                lambda s=space: on_select(s),
+                inputs=[],
+                outputs=[info_output, app_py_content]
+            )
 
     return demo