Update app.py

app.py

@@ -67,56 +67,11 @@ def format_tree_structure(tree_data: Dict, indent: str = "") -> str:
             formatted += format_tree_structure(child, indent + "  ")
     return formatted
 
-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},
-        {"role": "user", "content": user_message}
-    ]
-    
-    try:
-        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 analyze_code(app_content: str) -> str:
-    system_message = """당신은 Python 코드를 분석하는 AI 조수입니다. 주어진 코드를 분석하여 다음 항목에 대해 설명해주세요:
-    A. 배경 및 필요성
-    B. 기능적 효용성 및 가치
-    C. 특장점
-    D. 적용 대상 및 타겟
-    E. 기대효과
-    기존 및 유사 프로젝트와 비교하여 분석해주세요. Markdown 형식으로 출력하세요."""
-    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=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 조수입니다. 주어진 코드를 바탕으로 마치 화면을 보는 것처럼 사용법을 상세히 설명해주세요. Markdown 형식으로 출력하세요."
-    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:
-        return f"사용법 설명 생성 중 오류 발생: {str(e)}"
+
+
+
 
 def analyze_space(url: str, progress=gr.Progress()):
     try:
@@ -177,6 +132,75 @@ def respond(
 
 
 
+def summarize_code(app_content: str):
+    system_message = "당신은 Python 코드를 분석하고 요약하는 AI 조수입니다. 주어진 코드를 3줄 이내로 간결하게 요약해주세요."
+    user_message = f"다음 Python 코드를 3줄 이내로 요약해주세요:\n\n{app_content}"
+    
+    messages = [
+        {"role": "system", "content": system_message},
+        {"role": "user", "content": user_message}
+    ]
+    
+    try:
+        for response in hf_client.chat_completion_stream(messages, max_tokens=200, temperature=0.7):
+            yield response.choices[0].delta.content
+    except Exception as e:
+        yield f"요약 생성 중 오류 발생: {str(e)}"
+
+def analyze_code(app_content: str):
+    system_message = """당신은 Python 코드를 분석하는 AI 조수입니다. 주어진 코드를 분석하여 다음 항목에 대해 설명해주세요:
+    A. 배경 및 필요성
+    B. 기능적 효용성 및 가치
+    C. 특장점
+    D. 적용 대상 및 타겟
+    E. 기대효과
+    기존 및 유사 프로젝트와 비교하여 분석해주세요. Markdown 형식으로 출력하세요."""
+    user_message = f"다음 Python 코드를 분석해주세요:\n\n{app_content}"
+    
+    messages = [
+        {"role": "system", "content": system_message},
+        {"role": "user", "content": user_message}
+    ]
+    
+    try:
+        for response in hf_client.chat_completion_stream(messages, max_tokens=1000, temperature=0.7):
+            yield response.choices[0].delta.content
+    except Exception as e:
+        yield f"분석 생성 중 오류 발생: {str(e)}"
+
+def explain_usage(app_content: str):
+    system_message = "당신은 Python 코드를 분석하여 사용법을 설명하는 AI 조수입니다. 주어진 코드를 바탕으로 마치 화면을 보는 것처럼 사용법을 상세히 설명해주세요. Markdown 형식으로 출력하세요."
+    user_message = f"다음 Python 코드의 사용법을 설명해주세요:\n\n{app_content}"
+    
+    messages = [
+        {"role": "system", "content": system_message},
+        {"role": "user", "content": user_message}
+    ]
+    
+    try:
+        for response in hf_client.chat_completion_stream(messages, max_tokens=800, temperature=0.7):
+            yield response.choices[0].delta.content
+    except Exception as e:
+        yield f"사용법 설명 생성 중 오류 발생: {str(e)}"
+
+def analyze_space(url: str, progress=gr.Progress()):
+    try:
+        space_id = url.split('spaces/')[-1]
+        
+        progress(0.1, desc="파일 구조 분석 중...")
+        tree_structure = get_space_structure(space_id)
+        tree_view = format_tree_structure(tree_structure)
+        
+        progress(0.3, desc="app.py 내용 가져오는 중...")
+        app_content = get_file_content(space_id, "app.py")
+        
+        progress(1.0, desc="완료")
+        return app_content, tree_view, tree_structure, space_id
+    except Exception as e:
+        print(f"Error in analyze_space: {str(e)}")
+        print(traceback.format_exc())
+        return f"오류가 발생했습니다: {str(e)}", "", None, ""
+
 def create_ui():
     try:
         css = """
@@ -196,26 +220,31 @@ def create_ui():
             overflow-y: auto !important;
         }
         .tabs-style {
-            background-color: #ffff00;
+            background-color: #ffff00 !important;
+            font-weight: bold !important;
         }
-        """
-
-        js = """
-        function openFile(path, spaceId) {
-            const filePathInput = document.querySelector('input[data-testid="file_path_input"]');
-            const spaceIdInput = document.querySelector('input[data-testid="space_id_input"]');
-            if (filePathInput && spaceIdInput) {
-                filePathInput.value = path;
-                spaceIdInput.value = spaceId;
-                filePathInput.dispatchEvent(new Event('change'));
-            }
+        .main-tabs .tabitem[id^="tab_"] {
+            background-color: #ffff00 !important;
+            font-weight: bold !important;
+        }
+        .file-button {
+            background-color: #f0f0f0;
+            border: 1px solid #ddd;
+            padding: 5px 10px;
+            margin: 2px 0;
+            cursor: pointer;
+            text-align: left;
+            width: 100%;
+        }
+        .file-button:hover {
+            background-color: #e0e0e0;
         }
         """
 
         with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
             gr.Markdown("# HuggingFace Space Analyzer")
             
-            with gr.Tabs() as tabs:
+            with gr.Tabs(elem_classes="main-tabs") as tabs:
                 with gr.TabItem("분석"):
                     with gr.Row():
                         with gr.Column(scale=6):  # 왼쪽 60%
@@ -246,7 +275,7 @@ def create_ui():
                                         app_py_content = gr.Code(language="python", label="app.py", lines=30)
                                     requirements_tab = gr.TabItem("requirements.txt")
                                     with requirements_tab:
-                                        requirements_content = gr.Textbox(label="requirements.txt", lines=30)
+                                        requirements_content = gr.Code(language="text", label="requirements.txt", lines=30)
 
                 with gr.TabItem("AI 코딩"):
                     chatbot = gr.Chatbot(label="대화", type="messages")
@@ -296,26 +325,36 @@ def create_ui():
                 files = get_files(tree_structure)
                 buttons_html = "<div style='display: flex; flex-direction: column;'>"
                 for file in files:
-                    buttons_html += f"<button onclick=\"openFile('{file['path']}', '{space_id}')\" style='margin: 5px 0; padding: 5px; cursor: pointer;'>{file['path']}</button>"
+                    buttons_html += f"<button class='file-button' onclick='openFile(\"{file['path']}\", \"{space_id}\")'>{file['path']}</button>"
                 buttons_html += "</div>"
                 return buttons_html
 
             def open_file(file_path: str, space_id: str):
                 content = get_file_content(space_id, file_path)
                 file_name = file_path.split('/')[-1]
-                if file_name == "requirements.txt":
-                    return gr.Tabs.update(selected="requirements.txt"), gr.Textbox.update(value=content, label=file_name)
-                else:
-                    return gr.Tabs.update(selected=file_name), gr.Code.update(value=content, language="python" if file_name.endswith('.py') else "plaintext", label=file_name)
+                language = "python" if file_name.endswith('.py') else "text"
+                return gr.Tabs.update(selected=file_name), gr.Code.update(value=content, language=language, label=file_name)
 
             analyze_button.click(
                 analyze_space,
                 inputs=[url_input],
-                outputs=[summary_output, analysis_output, usage_output, app_py_content, tree_view_output, tree_structure_state, space_id_state]
+                outputs=[app_py_content, tree_view_output, tree_structure_state, space_id_state]
             ).then(
                 update_file_buttons,
                 inputs=[tree_structure_state, space_id_state],
                 outputs=[file_buttons]
+            ).then(
+                summarize_code,
+                inputs=[app_py_content],
+                outputs=[summary_output]
+            ).then(
+                analyze_code,
+                inputs=[app_py_content],
+                outputs=[analysis_output]
+            ).then(
+                explain_usage,
+                inputs=[app_py_content],
+                outputs=[usage_output]
             ).then(
                 lambda space_id: get_file_content(space_id, "requirements.txt"),
                 inputs=[space_id_state],
@@ -325,18 +364,12 @@ def create_ui():
             file_path_input = gr.Textbox(visible=False)
             space_id_input = gr.Textbox(visible=False)
 
-            def handle_file_open(file_path, space_id):
-                return file_path, space_id
-
             file_path_input.change(
                 open_file,
                 inputs=[file_path_input, space_id_input],
                 outputs=[code_tabs, code_tabs]
             )
 
-            # JavaScript 코드를 HTML에 직접 삽입
-            gr.HTML(f"<script>{js}</script>")
-
         return demo
 
     except Exception as e: