Update app.py

app.py

@@ -4,9 +4,7 @@ import os
 import requests
 from typing import List, Dict, Union
 import concurrent.futures
-import base64
 import traceback
-import pandas as pd
 
 # 환경 변수에서 토큰 가져오기
 HF_TOKEN = os.getenv("HF_TOKEN")
@@ -42,27 +40,6 @@ def get_most_liked_spaces(limit: int = 100) -> Union[List[Dict], str]:
     except ValueError as e:
         return f"JSON decoding error: {str(e)}"
 
-def capture_thumbnail(space_id: str) -> str:
-    screenshot_url = f"https://huggingface.co/spaces/{space_id}/screenshot.jpg"
-    try:
-        response = requests.get(screenshot_url, headers=get_headers())
-        if response.status_code == 200:
-            return base64.b64encode(response.content).decode('utf-8')
-    except requests.RequestException:
-        pass
-    return ""
-
-def get_app_py_content(space_id: str) -> str:
-    app_py_url = f"https://huggingface.co/spaces/{space_id}/raw/main/app.py"
-    try:
-        response = requests.get(app_py_url, headers=get_headers())
-        if response.status_code == 200:
-            return response.text
-        else:
-            return f"app.py file not found or inaccessible for space: {space_id}"
-    except requests.RequestException:
-        return f"Error fetching app.py content for space: {space_id}"
-
 def format_space(space: Dict) -> Dict:
     space_id = space.get('id', 'Unknown')
     space_name = space_id.split('/')[-1] if '/' in space_id else space_id
@@ -89,6 +66,17 @@ def format_spaces(spaces: Union[List[Dict], str]) -> List[Dict]:
     with concurrent.futures.ThreadPoolExecutor(max_workers=10) as executor:
         return list(executor.map(format_space, spaces))
 
+def get_app_py_content(space_id: str) -> str:
+    app_py_url = f"https://huggingface.co/spaces/{space_id}/raw/main/app.py"
+    try:
+        response = requests.get(app_py_url, headers=get_headers())
+        if response.status_code == 200:
+            return response.text
+        else:
+            return f"app.py file not found or inaccessible for space: {space_id}"
+    except requests.RequestException:
+        return f"Error fetching app.py content for space: {space_id}"
+
 def summarize_space(space: Dict) -> str:
     system_message = "당신은 Hugging Face Space의 내용을 요약하는 AI 조수입니다. 주어진 정보를 바탕으로 간결하고 명확한 요약을 제공해주세요."
     user_message = f"다음 Hugging Face Space를 요약해주세요: {space['name']} by {space['author']}. 좋아요 수: {space['likes']}. URL: {space['url']}"
@@ -106,60 +94,57 @@ def create_ui():
     formatted_spaces = format_spaces(spaces_list)
     print(f"Total spaces loaded: {len(formatted_spaces)}")  # 디버깅 출력
 
-    df = pd.DataFrame(formatted_spaces)
-    df['Open'] = df['url'].apply(lambda x: f'<a href="{x}" target="_blank">🔗</a>')
+    def create_table_html(spaces):
+        html = "<table><tr><th>Title</th><th>Author</th><th>Likes</th><th>Action</th></tr>"
+        for i, space in enumerate(spaces):
+            html += f"""
+            <tr>
+                <td>{space['name']}</td>
+                <td>{space['author']}</td>
+                <td>{space['likes']}</td>
+                <td><button onclick="selectSpace({i})">클릭</button></td>
+            </tr>
+            """
+        html += "</table>"
+        return html
 
     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():
-            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):
+            with gr.Column(scale=1):
+                space_table = gr.HTML(create_table_html(formatted_spaces))
+            
+            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_index):
             try:
-                selected_space = next((space for space in formatted_spaces if space['id'] == space_id), None)
-                if selected_space:
-                    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 ID: {space_id}")  # 디버��� 출력
-                    return "선택된 space를 찾을 수 없습니다.", ""
+                space_index = int(space_index)
+                selected_space = formatted_spaces[space_index]
+                summary = summarize_space(selected_space)
+                app_content = get_app_py_content(selected_space['id'])
+                info = f"선택된 Space: {selected_space['name']} (ID: {selected_space['id']})\n"
+                info += f"Author: {selected_space['author']}\n"
+                info += f"Likes: {selected_space['likes']}\n"
+                info += f"URL: {selected_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())  # 상세한 오류 정보 출력
+                print(traceback.format_exc())
                 return f"오류가 발생했습니다: {str(e)}", ""
 
-        def on_summarize(space_id):
-            try:
-                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 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:
-                print(f"Error in on_summarize: {str(e)}")
-                print(traceback.format_exc())  # 상세한 오류 정보 출력
-                return f"요약 중 오류가 발생했습니다: {str(e)}"
+        demo.load(None, None, None, _js="""
+        function selectSpace(index) {
+            document.querySelectorAll('button').forEach(btn => btn.classList.remove('selected'));
+            event.target.classList.add('selected');
+            gradio('on_select', index);
+        }
+        """)
 
-        selected_space.change(on_select, selected_space, [output, app_py_content])
-        summarize_btn.click(on_summarize, inputs=[selected_space], outputs=[output])
+        demo.load(on_select, None, [info_output, app_py_content], _js="(x) => selectedIndex")
 
     return demo