Update app.py

app.py

@@ -5,7 +5,6 @@ import requests
 from typing import List, Dict, Union
 import concurrent.futures
 import traceback
-import base64
 
 # 환경 변수에서 토큰 가져오기
 HF_TOKEN = os.getenv("HF_TOKEN")
@@ -126,13 +125,29 @@ def create_ui():
     }
     """
 
+    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)}", ""
+
     with gr.Blocks(css=css) as demo:
         gr.Markdown("# Hugging Face Most Liked Spaces")
         
         with gr.Row():
             with gr.Column(scale=1):
+                space_rows = []
                 for space in formatted_spaces:
-                    with gr.Row(elem_classes="space-row"):
+                    with gr.Row(elem_classes="space-row") as space_row:
                         if space['thumbnail']:
                             gr.Image(value=space['thumbnail'], show_label=False, elem_classes="thumbnail")
                         else:
@@ -140,30 +155,18 @@ def create_ui():
                         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")
-                            button.click(
-                                lambda s=space: on_select(s),
-                                inputs=[],
-                                outputs=[info_output, app_py_content]
-                            )
+                    space_rows.append((space_row, button, space))
             
             with gr.Column(scale=1):
                 info_output = gr.Textbox(label="Space 정보 및 요약", lines=10)
                 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 space_rows:
+            button.click(
+                lambda s=space: on_select(s),
+                inputs=[],
+                outputs=[info_output, app_py_content]
+            )
 
     return demo