Update app.py

app.py

@@ -10,6 +10,7 @@ import asyncio
 from gradio_client import Client
 import time
 import threading
+import json
 
 HF_TOKEN = os.getenv("HF_TOKEN")
 hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=HF_TOKEN)
@@ -90,28 +91,62 @@ def get_app_py_content(space_id: str) -> str:
             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 get_space_structure(space_id: str) -> Dict:
+    api_url = f"https://huggingface.co/api/spaces/{space_id}/tree"
+    try:
+        response = requests.get(api_url, headers=get_headers())
+        response.raise_for_status()
+        return response.json()
+    except requests.RequestException as e:
+        return {"error": f"API request error: {str(e)}"}
+
+def format_tree_structure(tree_data: Dict, indent: str = "") -> str:
+    formatted = ""
+    for item in tree_data.get("tree", []):
+        if item["type"] == "file":
+            formatted += f"{indent}├── {item['path']}\n"
+        elif item["type"] == "directory":
+            formatted += f"{indent}├── {item['path']}/\n"
+            formatted += format_tree_structure(item, indent + "│   ")
+    return formatted
+
+def format_list_structure(tree_data: Dict) -> List[str]:
+    formatted = []
+    for item in tree_data.get("tree", []):
+        if item["type"] == "file":
+            formatted.append(item["path"])
+        elif item["type"] == "directory":
+            formatted.append(f"{item['path']}/")
+            formatted.extend(format_list_structure(item))
+    return formatted
+
 def on_select(space):
     try:
         print(f"Selected space: {space['name']}")
         summary = summarize_space(space)
         app_content = get_app_py_content(space['id'])
+        tree_structure = get_space_structure(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}"
+        
+        tree_view = format_tree_structure(tree_structure)
+        list_view = "\n".join(format_list_structure(tree_structure))
+        
         print(f"Returning URL: {space['url']}")
-        return info, app_content, space['url']
+        return info, app_content, space['url'], tree_view, list_view
     except Exception as e:
         print(f"Error in on_select: {str(e)}")
         print(traceback.format_exc())
-        return f"오류가 발생했습니다: {str(e)}", "", ""
+        return f"오류가 발생했습니다: {str(e)}", "", "", "", ""
 
 def update_screenshot(url, last_url, force_update=False):
     print(f"Updating screenshot. Current URL: {url}, Last URL: {last_url}, Force update: {force_update}")
     if url and (url != last_url or force_update):
-
         screenshot = take_screenshot(url)
         print("Screenshot updated")
         return screenshot, url
@@ -123,7 +158,6 @@ def refresh_screenshot(url, last_url):
     # 항상 강제로 업데이트
     return update_screenshot(url, last_url, force_update=True)
 
-
 def take_screenshot(url):
     try:
         print(f"Taking screenshot of URL: {url}")
@@ -138,7 +172,6 @@ def take_screenshot(url):
     except Exception as e:
         print(f"Screenshot error: {str(e)}")
         return Image.new('RGB', (600, 360), color='lightgray')
-            
 
 def generate_usage_guide(app_content):
     system_message = "당신은 Python 코드를 분석하여, 화면 보듯이 이용 방법을 설명하는 AI 조수입니다. app.py 코드를 바탕으로 코드에 대한 언급은 제외하고, 이용자 관점에서 1) 기존 유사 기술 방식괴 비교해 특징, 장점에 대해 친절하고 자세하게 상세한 사용 방법을 제공해주세요."
@@ -169,7 +202,7 @@ def create_ui():
             width: 100% !important;
             margin-top: 5px !important;
         }
-        #info-output, #usage-guide {
+        #info-output, #usage-guide, #tree-view, #list-view {
             height: 400px;
             overflow-y: auto;
             padding-right: 10px;
@@ -201,35 +234,43 @@ def create_ui():
                         space_rows.append((space_row, button, space))
 
                 with gr.Column(scale=2):
-                    with gr.Group(elem_classes="output-group"):
-                        info_output = gr.Textbox(label="Space 정보 및 요약", elem_id="info-output", lines=20, max_lines=30)
-                    url_state = gr.State("")
-                    last_url_state = gr.State("")
-                    
-                    screenshot_output = gr.Image(type="pil", label="Live 화면", height=360, width=600)
-                    refresh_button = gr.Button("🔄 서비스 화면", elem_id="refresh-button")
-                    manual_button = gr.Button("선택 서비스 특징 및 사용법", elem_id="manual-button")
-                    
-                    with gr.Group(elem_classes="output-group"):
-                        usage_guide = gr.Textbox(label="선택 서비스 특징 및 사용법", elem_id="usage-guide", visible=False, lines=20, max_lines=30)
-                    
-                    with gr.Group(elem_classes="output-group"):
-                        app_py_content = gr.Code(language="python", label="메인 소스코드", elem_id="app-py-content", lines=None, max_lines=None)
-                    
-                    open_space_button = gr.Button("선택한 Space 열기", elem_id="open-space-button")
+                    with gr.Tabs():
+                        with gr.TabItem("기본 정보"):
+                            with gr.Group(elem_classes="output-group"):
+                                info_output = gr.Textbox(label="Space 정보 및 요약", elem_id="info-output", lines=20, max_lines=30)
+                            url_state = gr.State("")
+                            last_url_state = gr.State("")
+                            
+                            screenshot_output = gr.Image(type="pil", label="Live 화면", height=360, width=600)
+                            refresh_button = gr.Button("🔄 서비스 화면", elem_id="refresh-button")
+                            manual_button = gr.Button("선택 서비스 특징 및 사용법", elem_id="manual-button")
+                            
+                            with gr.Group(elem_classes="output-group"):
+                                usage_guide = gr.Textbox(label="선택 서비스 특징 및 사용법", elem_id="usage-guide", visible=False, lines=20, max_lines=30)
+                            
+                            with gr.Group(elem_classes="output-group"):
+                                app_py_content = gr.Code(language="python", label="메인 소스코드", elem_id="app-py-content", lines=None, max_lines=None)
+                            
+                            open_space_button = gr.Button("선택한 Space 열기", elem_id="open-space-button")
+                        
+                        with gr.TabItem("코드 구조 분석"):
+                            with gr.Group(elem_classes="output-group"):
+                                tree_view = gr.Textbox(label="트리 구조", elem_id="tree-view", lines=30, max_lines=50)
+                            with gr.Group(elem_classes="output-group"):
+                                list_view = gr.Textbox(label="리스트 구조", elem_id="list-view", lines=30, max_lines=50)
                     
                     update_trigger = gr.Button("Update Screenshot", visible=False)
 
                 def on_select_with_link(space):
-                    info, app_content, url = on_select(space)
+                    info, app_content, url, tree, list_structure = on_select(space)
                     info += f"\n\n선택한 Space URL: {url}"
-                    return info, app_content, url
+                    return info, app_content, url, tree, list_structure
 
                 for _, button, space in space_rows:
                     button.click(
                         lambda s=space: on_select_with_link(s),
                         inputs=[],
-                        outputs=[info_output, app_py_content, url_state]
+                        outputs=[info_output, app_py_content, url_state, tree_view, list_view]
                     ).then(
                         update_screenshot,
                         inputs=[url_state, last_url_state],
@@ -238,6 +279,7 @@ def create_ui():
 
                 def open_space_in_browser(url):
                     if url:
+                        import webbrowser
                         webbrowser.open(url)
                         return f"'{url}' 주소가 새 탭에서 열렸습니다."
                     return "선택된 Space가 없습니다."
@@ -248,10 +290,6 @@ def create_ui():
                     outputs=[gr.Textbox(label="상태 메시지")]
                 )
 
-                def refresh_screenshot(url, last_url):
-                    print(f"Refresh button clicked. URL: {url}, Last URL: {last_url}")
-                    return update_screenshot(url, last_url)
-
                 refresh_button.click(
                     refresh_screenshot,
                     inputs=[url_state, last_url_state],
@@ -295,4 +333,6 @@ if __name__ == "__main__":
         demo.launch()
     except Exception as e:
         print(f"Error in main: {str(e)}")
-        print(traceback.format_exc())
+        print(traceback.format_exc())
+
+