Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 import requests
-from typing import List, Dict, Union, Tuple
+from typing import List, Dict, Union
 import concurrent.futures
 import base64
 
@@ -12,6 +12,9 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 # 추론 API 클라이언트 설정
 hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=HF_TOKEN)
 
+# 전역 변수로 선택된 space 정보 저장
+selected_space_info = None
+
 def get_headers():
     if not HF_TOKEN:
         raise ValueError("Hugging Face token not found in environment variables")
@@ -105,10 +108,10 @@ def summarize_space(space: Dict) -> str:
 def create_ui():
     spaces_list = get_most_liked_spaces()
     formatted_spaces = format_spaces(spaces_list)
+    print(f"Total spaces loaded: {len(formatted_spaces)}")  # 디버깅 출력
 
     with gr.Blocks() as demo:
         gr.Markdown("# Hugging Face Most Liked Spaces")
-        selected_space_id = gr.State(None)
         
         with gr.Row():
             with gr.Column(scale=1):
@@ -123,27 +126,28 @@ def create_ui():
                 app_py_content = gr.Code(language="python", label="app.py 내용")
 
         def on_select(evt: gr.SelectData):
+            global selected_space_info
             selected_id = evt.value[1]  # tuple의 두 번째 요소가 space_id입니다.
             selected_space = next((space for space in formatted_spaces if space['id'] == selected_id), None)
             if selected_space:
                 app_content = get_app_py_content(selected_id)
-                print(f"Selected space: {selected_space['name']}")  # 디버깅용 출력
-                return selected_id, selected_space['url'], app_content
-            print("Selected space not found")  # 디버깅용 출력
-            return None, "선택된 space를 찾을 수 없습니다.", ""
-
-        def on_summarize(space_id):
-            if space_id:
-                selected_space = next((space for space in formatted_spaces if space['id'] == space_id), None)
-                if selected_space:
-                    summary = summarize_space(selected_space)
-                    print(f"Summarizing space: {selected_space['name']}")  # 디버깅용 출력
-                    return summary
-            print("No space selected for summarization")  # 디버깅용 출력
+                selected_space_info = selected_space
+                print(f"Selected space: {selected_space['name']}")  # 디버깅 출력
+                return f"선택된 Space: {selected_space['name']} (ID: {selected_id})\nURL: {selected_space['url']}", app_content
+            print(f"Selected space not found for ID: {selected_id}")  # 디버깅 출력
+            return "선택된 space를 찾을 수 없습니다.", ""
+
+        def on_summarize():
+            global selected_space_info
+            if selected_space_info:
+                summary = summarize_space(selected_space_info)
+                print(f"Summarizing space: {selected_space_info['name']}")  # 디버깅 출력
+                return summary
+            print("No space selected for summarization")  # 디버깅 출력
             return "선택된 space가 없습니다. 먼저 리스트에서 space를 선택해주세요."
 
-        space_list.select(on_select, None, [selected_space_id, output, app_py_content])
-        summarize_btn.click(on_summarize, inputs=[selected_space_id], outputs=[output])
+        space_list.select(on_select, None, [output, app_py_content])
+        summarize_btn.click(on_summarize, None, output)
 
     return demo