Update app.py

app.py

@@ -108,26 +108,16 @@ def create_ui():
     formatted_spaces = format_spaces(spaces_list)
     print(f"Total spaces loaded: {len(formatted_spaces)}")  # 디버깅 출력
 
-    space_choices = {f"{space['name']} by {space['author']} (Likes: {space['likes']})": space['id'] for space in formatted_spaces}
+    space_choices = [f"{space['name']} by {space['author']} (Likes: {space['likes']})" for space in formatted_spaces]
 
-    with gr.Blocks(css="#space-list { max-height: 400px; overflow-y: auto; }") as demo:
+    with gr.Blocks() as demo:
         gr.Markdown("# Hugging Face Most Liked Spaces")
         
         with gr.Row():
             with gr.Column(scale=1):
-                with gr.Column(elem_id="space-list"):
-                    space_radios = []
-                    for space in formatted_spaces:
-                        with gr.Row():
-                            radio = gr.Radio(
-                                choices=[f"{space['name']} by {space['author']} (Likes: {space['likes']})"],
-                                label="",
-                                value=None,
-                                elem_id=f"radio-{space['id']}"
-                            )
-                            space_radios.append(radio)
-                            gr.Button("🔗", elem_id=f"link-{space['id']}")
+                space_dropdown = gr.Dropdown(choices=space_choices, label="Select a Space")
                 summarize_btn = gr.Button("요약")
+                link_btn = gr.Button("🔗 Open Space")
             
             with gr.Column(scale=2):
                 output = gr.Textbox(label="Space 정보", lines=10)
@@ -136,14 +126,13 @@ def create_ui():
         def on_select(choice):
             try:
                 print(f"Selected: {choice}")  # 디버깅 출력
-                selected_id = space_choices[choice[0]]  # choice is now a list with one item
-                selected_space = next((space for space in formatted_spaces if space['id'] == selected_id), None)
+                selected_space = next((space for space in formatted_spaces if f"{space['name']} by {space['author']} (Likes: {space['likes']})" == choice), None)
                 if selected_space:
-                    app_content = get_app_py_content(selected_id)
-                    print(f"Selected space: {selected_space['name']} (ID: {selected_id})")  # 디버깅 출력
-                    return f"선택된 Space: {selected_space['name']} (ID: {selected_id})\nURL: {selected_space['url']}", app_content
+                    app_content = get_app_py_content(selected_space['id'])
+                    print(f"Selected space: {selected_space['name']} (ID: {selected_space['id']})")  # 디버깅 출력
+                    return f"선택된 Space: {selected_space['name']} (ID: {selected_space['id']})\nURL: {selected_space['url']}", app_content
                 else:
-                    print(f"Space not found for ID: {selected_id}")  # 디버깅 출력
+                    print(f"Space not found for choice: {choice}")  # 디버깅 출력
                     return "선택된 space를 찾을 수 없습니다.", ""
             except Exception as e:
                 print(f"Error in on_select: {str(e)}")
@@ -153,8 +142,7 @@ def create_ui():
         def on_summarize(choice):
             try:
                 if choice:
-                    selected_id = space_choices[choice[0]]  # choice is now a list with one item
-                    selected_space = next((space for space in formatted_spaces if space['id'] == selected_id), None)
+                    selected_space = next((space for space in formatted_spaces if f"{space['name']} by {space['author']} (Likes: {space['likes']})" == choice), None)
                     if selected_space:
                         summary = summarize_space(selected_space)
                         print(f"Summarizing space: {selected_space['name']}")  # 디버깅 출력
@@ -166,28 +154,25 @@ def create_ui():
                 print(traceback.format_exc())  # 상세한 오류 정보 출력
                 return f"요약 중 오류가 발생했습니다: {str(e)}"
 
-        for radio in space_radios:
-            radio.change(on_select, radio, [output, app_py_content])
-
-        summarize_btn.click(on_summarize, inputs=[gr.State(lambda: next((radio for radio in space_radios if radio.value), None))], outputs=[output])
-
-        # JavaScript to open links in new tabs
-        link_script = """
-        function setupLinkButtons() {
-            document.querySelectorAll('button[id^="link-"]').forEach(button => {
-                button.addEventListener('click', function() {
-                    var spaceId = this.id.split('-')[1];
-                    window.open('https://huggingface.co/spaces/' + spaceId, '_blank');
-                });
-            });
-        }
-        if (document.readyState === 'complete') {
-            setupLinkButtons();
-        } else {
-            document.addEventListener('DOMContentLoaded', setupLinkButtons);
+        def open_space_link(choice):
+            if choice:
+                selected_space = next((space for space in formatted_spaces if f"{space['name']} by {space['author']} (Likes: {space['likes']})" == choice), None)
+                if selected_space:
+                    return f"<script>window.open('{selected_space['url']}', '_blank');</script>"
+            return ""
+
+        space_dropdown.change(on_select, space_dropdown, [output, app_py_content])
+        summarize_btn.click(on_summarize, inputs=[space_dropdown], outputs=[output])
+        link_btn.click(open_space_link, inputs=[space_dropdown], outputs=[gr.HTML()])
+
+        # Add JavaScript to handle link opening
+        gr.HTML("""
+        <script>
+        function openSpaceLink(url) {
+            window.open(url, '_blank');
         }
-        """
-        demo.load(None, None, None, _js=link_script)
+        </script>
+        """)
 
     return demo