Update app.py

app.py

@@ -108,6 +108,8 @@ def create_ui():
 
     with gr.Blocks() as demo:
         gr.Markdown("# Hugging Face Most Liked Spaces")
+        selected_space = gr.State(None)
+        
         with gr.Row():
             with gr.Column(scale=1):
                 space_list = gr.List(
@@ -122,22 +124,20 @@ def create_ui():
 
         def on_select(evt: gr.SelectData):
             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_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)
+            selected_space_data = next((space for space in formatted_spaces if space['id'] == selected_space_id), None)
+            if selected_space_data:
+                app_content = get_app_py_content(selected_space_data['id'])
+                return selected_space_data, selected_space_data['url'], app_content
+            return None, "선택된 space를 찾을 수 없습니다.", ""
+
+        def on_summarize(space):
+            if space:
+                summary = summarize_space(space)
                 return summary
-            return "선택된 space를 찾을 수 없습니다."
+            return "선택된 space가 없습니다. 먼저 리스트에서 space를 선택해주세요."
 
-        space_list.select(on_select, None, [output, app_py_content])
-        summarize_btn.click(on_summarize, space_list, output)
+        space_list.select(on_select, None, [selected_space, output, app_py_content])
+        summarize_btn.click(on_summarize, inputs=[selected_space], outputs=[output])
 
     return demo