Update app.py

app.py

@@ -192,17 +192,15 @@ def create_ui():
             height: calc(100vh - 200px) !important;
             overflow-y: auto !important;
         }
+        .tabs-style {
+            background-color: #ffff00;
+        }
         """
 
         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'));
-            }
+            const event = new CustomEvent('open-file', { detail: { path, spaceId } });
+            window.dispatchEvent(event);
         }
         """
 
@@ -233,11 +231,14 @@ def create_ui():
                         
                         with gr.Column(scale=4):  # 오른쪽 40%
                             with gr.Group(elem_classes="output-group full-height"):
-                                code_tabs = gr.Tabs()
+                                code_tabs = gr.Tabs(elem_classes="tabs-style")
                                 with code_tabs:
                                     app_py_tab = gr.TabItem("app.py")
                                     with app_py_tab:
                                         app_py_content = gr.Code(language="python", label="app.py", lines=30)
+                                    requirements_tab = gr.TabItem("requirements.txt")
+                                    with requirements_tab:
+                                        requirements_content = gr.Code(language="text", label="requirements.txt", lines=30)
 
                 with gr.TabItem("AI 코딩"):
                     chatbot = gr.Chatbot()
@@ -249,6 +250,9 @@ 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 토큰 이상 유튜브 영상 스크립트 형태로 작성하라"],
@@ -285,13 +289,14 @@ 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;'>{file['path']}</button>"
+                    buttons_html += f"<button onclick=\"openFile('{file['path']}', '{space_id}')\" style='margin: 5px 0; padding: 5px; cursor: pointer;'>{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)
-                return gr.Tabs.update(selected=file_path), gr.Code(value=content, language="python", label=file_path, lines=30)
+                file_name = file_path.split('/')[-1]
+                return gr.Tabs.update(selected=file_name), gr.Code(value=content, language="python" if file_name.endswith('.py') else "text", label=file_name, lines=30)
 
             analyze_button.click(
                 analyze_space,
@@ -301,6 +306,10 @@ def create_ui():
                 update_file_buttons,
                 inputs=[tree_structure_state, space_id_state],
                 outputs=[file_buttons]
+            ).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)
@@ -318,6 +327,23 @@ def create_ui():
             # JavaScript 코드를 HTML에 직접 삽입
             gr.HTML(f"<script>{js}</script>")
 
+            # 파일 열기 이벤트 리스너 추가
+            demo.load(None, None, None, _js="""
+            () => {
+                function handleOpenFile(event) {
+                    const { path, spaceId } = event.detail;
+                    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'));
+                    }
+                }
+                window.addEventListener('open-file', handleOpenFile);
+            }
+            """)
+
         return demo
 
     except Exception as e:
@@ -332,4 +358,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())