Update app.py

app.py

@@ -86,7 +86,6 @@ def summarize_code(app_content: str):
     except Exception as e:
         return f"요약 생성 중 오류 발생: {str(e)}"
 
-
 def analyze_code(app_content: str):
     system_message = """당신은 Python 코드를 분석하는 AI 조수입니다. 주어진 코드를 분석하여 다음 항목에 대해 설명해주세요:
     A. 배경 및 필요성
@@ -160,8 +159,6 @@ def respond(message: str, chat_history: List[Dict[str, str]], system_message: st
     except Exception as e:
         return f"응답 생성 중 오류 발생: {str(e)}"
 
-
-
 def create_ui():
     try:
         css = """
@@ -247,9 +244,7 @@ def create_ui():
                         top_p = gr.Slider(minimum=0, maximum=1, value=0.9, label="Top P")
                     
                     examples = [
-                        ["패션 코드 실행"],
-                        ["UHD 이미지 코드 실행"],
-                        ["MixGEN 코드 실행"],
+
                         ["상세한 사용 방법을 마치 화면을 보면서 설명하듯이 4000 토큰 이상 자세히 설명하라"],
                         ["FAQ 20건을 상세하게 작성하라. 4000토큰 이상 사용하라."],
                         ["사용 방법과 차별점, 특징, 강점을 중심으로 4000 토큰 이상 유튜브 영상 스크립트 형태로 작성하라"],
@@ -268,7 +263,87 @@ def create_ui():
 
                     msg.submit(respond_wrapper, [msg, chatbot, system_message, max_tokens, temperature, top_p], [msg, chatbot])
 
-            # ... (나머지 코드는 그대로 유지)
+            space_id_state = gr.State()
+            tree_structure_state = gr.State()
+
+            def update_file_buttons(tree_structure, space_id):
+                if tree_structure is None:
+                    return ""
+                
+                def get_files(node):
+ 
+                    files = []
+                    if node["type"] == "file":
+                        files.append(node)
+                    elif node["type"] == "directory":
+                        for child in node.get("children", []):
+                            files.extend(get_files(child))
+                    return files
+
+                files = get_files(tree_structure)
+                buttons_html = "<div style='display: flex; flex-direction: column;'>"
+                for file in files:
+                    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)
+
+            analyze_button.click(
+                analyze_space,
+                inputs=[url_input],
+                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],
+                outputs=[requirements_content]
+            )
+
+            file_path_input = gr.Textbox(visible=False)
+            space_id_input = gr.Textbox(visible=False)
+
+            file_path_input.change(
+                open_file,
+                inputs=[file_path_input, space_id_input],
+                outputs=[code_tabs, code_tabs]
+            )
+
+            # JavaScript 코드를 HTML에 직접 삽입
+            gr.HTML("""
+            <script>
+            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'));
+                }
+            }
+            </script>
+            """)
 
         return demo
 
@@ -288,4 +363,4 @@ if __name__ == "__main__":
         )
     except Exception as e:
         print(f"Error in main: {str(e)}")
-        print(traceback.format_exc())
+        print(traceback.format_exc())