Update app.py

app.py

@@ -6,6 +6,7 @@ from typing import List, Dict, Union
 import concurrent.futures
 import base64
 import traceback
+import pandas as pd
 
 # 환경 변수에서 토큰 가져오기
 HF_TOKEN = os.getenv("HF_TOKEN")
@@ -73,15 +74,12 @@ def format_space(space: Dict) -> Dict:
     space_likes = space.get('likes', 'N/A')
     space_url = f"https://huggingface.co/spaces/{space_id}"
     
-    thumbnail = capture_thumbnail(space_id)
-    
     return {
         "id": space_id,
         "name": space_name,
         "author": space_author,
         "likes": space_likes,
-        "url": space_url,
-        "thumbnail": thumbnail
+        "url": space_url
     }
 
 def format_spaces(spaces: Union[List[Dict], str]) -> List[Dict]:
@@ -108,45 +106,51 @@ 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']})" for space in formatted_spaces]
+    df = pd.DataFrame(formatted_spaces)
+    df['Open'] = df['url'].apply(lambda x: f'<a href="{x}" target="_blank">🔗</a>')
 
     with gr.Blocks() as demo:
         gr.Markdown("# Hugging Face Most Liked Spaces")
         
+        space_table = gr.Dataframe(
+            value=df[['name', 'author', 'likes', 'Open']],
+            headers=['Name', 'Author', 'Likes', 'Open'],
+            row_count=(len(formatted_spaces), "fixed"),
+            col_count=(4, "fixed"),
+            interactive=False,
+            wrap=True
+        )
+        
         with gr.Row():
-            with gr.Column(scale=1):
-                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)
-                app_py_content = gr.Code(language="python", label="app.py 내용")
-
-        def on_select(choice):
+            selected_space = gr.Dropdown(choices=[space['id'] for space in formatted_spaces], label="Select a Space for Summary")
+            summarize_btn = gr.Button("요약")
+        
+        output = gr.Textbox(label="Space 정보 및 요약", lines=10)
+        app_py_content = gr.Code(language="python", label="app.py 내용")
+
+        def on_select(space_id):
             try:
-                print(f"Selected: {choice}")  # 디버깅 출력
-                selected_space = next((space for space in formatted_spaces if f"{space['name']} by {space['author']} (Likes: {space['likes']})" == choice), None)
+                selected_space = next((space for space in formatted_spaces if space['id'] == space_id), None)
                 if selected_space:
-                    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
+                    app_content = get_app_py_content(space_id)
+                    print(f"Selected space: {selected_space['name']} (ID: {space_id})")  # 디버깅 출력
+                    return f"선택된 Space: {selected_space['name']} (ID: {space_id})\nURL: {selected_space['url']}", app_content
                 else:
-                    print(f"Space not found for choice: {choice}")  # 디버깅 출력
+                    print(f"Space not found for ID: {space_id}")  # 디버깅 출력
                     return "선택된 space를 찾을 수 없습니다.", ""
             except Exception as e:
                 print(f"Error in on_select: {str(e)}")
                 print(traceback.format_exc())  # 상세한 오류 정보 출력
                 return f"오류가 발생했습니다: {str(e)}", ""
 
-        def on_summarize(choice):
+        def on_summarize(space_id):
             try:
-                if choice:
-                    selected_space = next((space for space in formatted_spaces if f"{space['name']} by {space['author']} (Likes: {space['likes']})" == choice), None)
+                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
+                        return f"Space: {selected_space['name']} by {selected_space['author']}\nLikes: {selected_space['likes']}\nURL: {selected_space['url']}\n\n요약:\n{summary}"
                 print("No space selected for summarization")  # 디버깅 출력
                 return "선택된 space가 없습니다. 먼저 리스트에서 space를 선택해주세요."
             except Exception as e:
@@ -154,25 +158,8 @@ def create_ui():
                 print(traceback.format_exc())  # 상세한 오류 정보 출력
                 return f"요약 중 오류가 발생했습니다: {str(e)}"
 
-        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');
-        }
-        </script>
-        """)
+        selected_space.change(on_select, selected_space, [output, app_py_content])
+        summarize_btn.click(on_summarize, inputs=[selected_space], outputs=[output])
 
     return demo