Update app.py

app.py

@@ -4,6 +4,7 @@ import os
 import requests
 from typing import List, Dict, Union, Tuple
 import concurrent.futures
+import base64
 
 # 환경 변수에서 토큰 가져오기
 HF_TOKEN = os.getenv("HF_TOKEN")
@@ -110,7 +111,7 @@ def create_ui():
         with gr.Row():
             with gr.Column(scale=1):
                 space_list = gr.List(
-                    [f"{space['name']} by {space['author']} (Likes: {space['likes']})" for space in formatted_spaces],
+                    [(f"{space['name']} by {space['author']} (Likes: {space['likes']})", space['id']) for space in formatted_spaces],
                     label="Most Liked Spaces"
                 )
                 summarize_btn = gr.Button("요약")
@@ -120,17 +121,23 @@ def create_ui():
                 app_py_content = gr.Code(language="python", label="app.py 내용")
 
         def on_select(evt: gr.SelectData):
-            selected_space = formatted_spaces[evt.index]
-            app_content = get_app_py_content(selected_space['id'])
-            return selected_space['url'], app_content
+            selected_space_id = evt.value[1]  # tuple의 두 번째 요소가 space_id입니다.
+            selected_space = next((space for space in formatted_spaces if space['id'] == selected_space_id), None)
+            if selected_space:
+                app_content = get_app_py_content(selected_space['id'])
+                return selected_space['url'], app_content
+            return "선택된 space를 찾을 수 없습니다.", ""
 
         def on_summarize(evt: gr.SelectData):
-            selected_space = formatted_spaces[evt.index]
-            summary = summarize_space(selected_space)
-            return summary
+            selected_space_id = evt.value[1]
+            selected_space = next((space for space in formatted_spaces if space['id'] == selected_space_id), None)
+            if selected_space:
+                summary = summarize_space(selected_space)
+                return summary
+            return "선택된 space를 찾을 수 없습니다."
 
         space_list.select(on_select, None, [output, app_py_content])
-        summarize_btn.click(on_summarize, None, output)
+        summarize_btn.click(on_summarize, space_list, output)
 
     return demo