Update app.py

app.py

@@ -21,59 +21,6 @@ def get_headers():
         raise ValueError("Hugging Face token not found in environment variables")
     return {"Authorization": f"Bearer {HF_TOKEN}"}
 
-def get_space_structure(space_id: str) -> Dict:
-    try:
-        # space_id에서 owner와 repo_name을 분리
-        owner, repo_name = space_id.split('/')
-        
-        # HfApi를 사용하여 파일 목록을 가져옴
-        files = hf_api.list_repo_files(repo_id=space_id, repo_type="space")
-        
-        # 파일 목록을 트리 구조로 변환
-        tree = {"type": "directory", "path": "", "tree": []}
-        for file in files:
-            path_parts = file.split('/')
-            current = tree
-            for i, part in enumerate(path_parts):
-                if i == len(path_parts) - 1:  # 파일
-                    current["tree"].append({"type": "file", "path": part})
-                else:  # 디렉토리
-                    found = False
-                    for item in current["tree"]:
-                        if item["type"] == "directory" and item["path"] == part:
-                            current = item
-                            found = True
-                            break
-                    if not found:
-                        new_dir = {"type": "directory", "path": part, "tree": []}
-                        current["tree"].append(new_dir)
-                        current = new_dir
-        
-        return tree
-    except Exception as e:
-        print(f"Error in get_space_structure: {str(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 get_app_py_content(space_id: str) -> str:
     app_py_url = f"https://huggingface.co/spaces/{space_id}/raw/main/app.py"
     try:
@@ -85,9 +32,9 @@ def get_app_py_content(space_id: str) -> str:
     except requests.RequestException:
         return f"Error fetching app.py content for space: {space_id}"
 
-def summarize_space(space_id: str) -> str:
-    system_message = "당신은 Hugging Face Space의 내용을 요약하는 AI 조수입니다. 주어진 정보를 바탕으로 간결하고 명확한 요약을 제공해주세요."
-    user_message = f"다음 Hugging Face Space를 요약해주세요: {space_id}"
+def summarize_code(app_content: str) -> str:
+    system_message = "당신은 Python 코드를 분석하고 요약하는 AI 조수입니다. 주어진 코드를 3줄 이내로 간결하게 요약해주세요."
+    user_message = f"다음 Python 코드를 3줄 이내로 요약해주세요:\n\n{app_content}"
     
     messages = [
         {"role": "system", "content": system_message},
@@ -95,65 +42,67 @@ def summarize_space(space_id: str) -> str:
     ]
     
     try:
-        response = hf_client.chat_completion(messages, max_tokens=400, temperature=0.7)
+        response = hf_client.chat_completion(messages, max_tokens=200, temperature=0.7)
         return response.choices[0].message.content
     except Exception as e:
         return f"요약 생성 중 오류 발생: {str(e)}"
 
-def take_screenshot(url):
+def analyze_code(app_content: str) -> str:
+    system_message = """당신은 Python 코드를 분석하는 AI 조수입니다. 주어진 코드를 분석하여 다음 항목에 대해 설명해주세요:
+    A. 배경 및 필요성
+    B. 기능적 효용성 및 가치
+    C. 특장점
+    D. 적용 대상 및 타겟
+    E. 기대효과
+    기존 및 유사 프로젝트와 비교하여 분석해주세요."""
+    user_message = f"다음 Python 코드를 분석해주세요:\n\n{app_content}"
+    
+    messages = [
+        {"role": "system", "content": system_message},
+        {"role": "user", "content": user_message}
+    ]
+    
     try:
-        print(f"Taking screenshot of URL: {url}")
-        client = Client("ginipick/selenium-screenshot-gradio")
-        result = client.predict(url=url, api_name="/predict")
-        print(f"Screenshot result: {result}")
-        if isinstance(result, str) and os.path.exists(result):
-            return Image.open(result)
-        else:
-            print(f"Invalid result from API: {result}")
-            return Image.new('RGB', (600, 360), color='lightgray')
+        response = hf_client.chat_completion(messages, max_tokens=1000, temperature=0.7)
+        return response.choices[0].message.content
+    except Exception as e:
+        return f"분석 생성 중 오류 발생: {str(e)}"
+
+def explain_usage(app_content: str) -> str:
+    system_message = "당신은 Python 코드를 분석하여 사용법을 설명하는 AI 조수입니다. 주어진 코드를 바탕으로 마치 화면을 보는 것처럼 사용법을 상세히 설명해주세요."
+    user_message = f"다음 Python 코드의 사용법을 설명해주세요:\n\n{app_content}"
+    
+    messages = [
+        {"role": "system", "content": system_message},
+        {"role": "user", "content": user_message}
+    ]
+    
+    try:
+        response = hf_client.chat_completion(messages, max_tokens=800, temperature=0.7)
+        return response.choices[0].message.content
     except Exception as e:
-        print(f"Screenshot error: {str(e)}")
-        return Image.new('RGB', (600, 360), color='lightgray')
+        return f"사용법 설명 생성 중 오류 발생: {str(e)}"
 
 def analyze_space(url: str):
     try:
         # URL에서 space_id 추출
         space_id = url.split('spaces/')[-1]
         
-        summary = summarize_space(space_id)
         app_content = get_app_py_content(space_id)
-        tree_structure = get_space_structure(space_id)
-        
-        tree_view = format_tree_structure(tree_structure)
-        list_view = "\n".join(format_list_structure(tree_structure))
-        screenshot = take_screenshot(url)
+        summary = summarize_code(app_content)
+        analysis = analyze_code(app_content)
+        usage = explain_usage(app_content)
         
-        return summary, app_content, tree_view, list_view, screenshot, url
+        return summary, analysis, usage, app_content
     except Exception as e:
         print(f"Error in analyze_space: {str(e)}")
         print(traceback.format_exc())
-        return f"오류가 발생했습니다: {str(e)}", "", "", "", None, ""
+        return f"오류가 발생했습니다: {str(e)}", "", "", ""
 
 def create_ui():
     try:
         css = """
         footer {visibility: hidden;}
-        .minimal-button {min-width: 30px !important; height: 25px !important; line-height: 1 !important; font-size: 12px !important; padding: 2px 5px !important;}
-        .space-row {margin-bottom: 5px !important;}
-        #refresh-button, #manual-button, #open-space-button {
-            width: 100% !important;
-            margin-top: 5px !important;
-        }
-        #info-output, #usage-guide, #tree-view, #list-view {
-            height: 400px !important;
-            overflow-y: auto !important;
-            padding-right: 10px !important;
-        }
-        #app-py-content {
-            height: auto !important;
-            max-height: none !important;
-            overflow-y: visible !important;
-        }
         .output-group {
             border: 1px solid #ddd;
             border-radius: 5px;
@@ -164,63 +113,33 @@ def create_ui():
             overflow: auto !important;
             max-height: 400px !important;
         }
-        .full-height {
-            height: auto !important;
-            max-height: none !important;
-        }
         """
 
         with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
             gr.Markdown("# HuggingFace Space Analyzer")
             
             with gr.Row():
-                url_input = gr.Textbox(label="HuggingFace Space URL")
-                analyze_button = gr.Button("분석")
-            
-            with gr.Row():
-                with gr.Column(scale=2):
-                    with gr.Tabs():
-                        with gr.TabItem("기본 정보"):
-                            with gr.Group(elem_classes="output-group scroll-lock"):
-                                info_output = gr.Textbox(label="Space 정보 및 요약", elem_id="info-output", lines=20, max_lines=30)
-                            
-                            screenshot_output = gr.Image(type="pil", label="Live 화면", height=360, width=600)
-                            refresh_button = gr.Button("🔄 서비스 화면", elem_id="refresh-button")
-                            
-                            with gr.Group(elem_classes="output-group full-height"):
-                                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 scroll-lock"):
-                                tree_view = gr.Textbox(label="트리 구조", elem_id="tree-view", lines=30, max_lines=50)
-                            with gr.Group(elem_classes="output-group scroll-lock"):
-                                list_view = gr.Textbox(label="파일 리스트", elem_id="list-view", lines=30, max_lines=50)
-
-            def open_space_in_browser(url):
-                if url:
-                    import webbrowser
-                    webbrowser.open(url)
-                    return f"'{url}' 주소가 새 탭에서 열렸습니다."
-                return "선택된 Space가 없습니다."
+                with gr.Column(scale=6):  # 왼쪽 60%
+                    url_input = gr.Textbox(label="HuggingFace Space URL")
+                    analyze_button = gr.Button("분석")
+                    
+                    with gr.Group(elem_classes="output-group scroll-lock"):
+                        summary_output = gr.Textbox(label="요약 (3줄 이내)", lines=3)
+                    
+                    with gr.Group(elem_classes="output-group scroll-lock"):
+                        analysis_output = gr.Textbox(label="분석", lines=15)
+                    
+                    with gr.Group(elem_classes="output-group scroll-lock"):
+                        usage_output = gr.Textbox(label="사용법", lines=10)
+                
+                with gr.Column(scale=4):  # 오른쪽 40%
+                    with gr.Group(elem_classes="output-group scroll-lock"):
+                        app_py_content = gr.Code(language="python", label="메인 소스코드", lines=30)
 
             analyze_button.click(
                 analyze_space,
                 inputs=[url_input],
-                outputs=[info_output, app_py_content, tree_view, list_view, screenshot_output, url_input]
-            )
-
-            open_space_button.click(
-                open_space_in_browser,
-                inputs=[url_input],
-                outputs=[gr.Textbox(label="상태 메시지")]
-            )
-
-            refresh_button.click(
-                take_screenshot,
-                inputs=[url_input],
-                outputs=[screenshot_output]
+                outputs=[summary_output, analysis_output, usage_output, app_py_content]
             )
 
         return demo
@@ -236,4 +155,4 @@ if __name__ == "__main__":
         demo.launch()
     except Exception as e:
         print(f"Error in main: {str(e)}")
-        print(traceback.format_exc())           
+        print(traceback.format_exc())