Update app.py

app.py

@@ -122,8 +122,6 @@ def explain_usage(app_content: str):
     except Exception as e:
         return f"사용법 설명 생성 중 오류 발생: {str(e)}"
 
-
-
 def analyze_space(url: str, progress=gr.Progress()):
     try:
         space_id = url.split('spaces/')[-1]
@@ -157,16 +155,13 @@ def analyze_space(url: str, progress=gr.Progress()):
         print(traceback.format_exc())
         return f"오류가 발생했습니다: {str(e)}", "", None, "", "", "", ""
 
-        
 def respond(message: str, chat_history: List[Dict[str, str]], max_tokens: int, temperature: float, top_p: float) -> str:
     system_message = """당신은 허깅페이스에 특화된 AI 코딩 전문가입니다. 사용자의 질문에 친절하고 상세하게 답변해주세요. 
     Gradio 특성을 정확히 인식하고 Requirements.txt 누락없이 코딩과 오류를 해결해야 합니다.
     항상 정확하고 유용한 정보를 제공하도록 노력하세요."""
     
     messages = [{"role": "system", "content": system_message}]
-    for chat in chat_history:
-        messages.append({"role": "user", "content": chat["content"]})
-        messages.append({"role": "assistant", "content": chat["content"]})
+    messages.extend(chat_history)
     messages.append({"role": "user", "content": message})
     
     try:
@@ -205,6 +200,8 @@ def create_ui():
             padding: 10px 20px;
             margin: 0;
             transition: background-color 0.3s;
+            font-size: 16px;
+            font-weight: bold;
         }
         .tab-nav button:hover {
             background-color: #2980b9;
@@ -212,19 +209,6 @@ def create_ui():
         .tab-nav button.selected {
             color: #2c3e50 !important;
             background-color: #ecf0f1;
-            font-weight: bold;
-        }
-        .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;
         }
         input[type="text"], textarea {
             color: #2c3e50 !important;
@@ -232,11 +216,10 @@ def create_ui():
         }
         """
 
-
-        with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
+        with gr.Blocks(css=css, theme=gr.themes.Default()) as demo:
             gr.Markdown("# HuggingFace Space Analyzer")
             
-            with gr.Tabs(elem_classes="main-tabs") as tabs:
+            with gr.Tabs() as tabs:
                 with gr.TabItem("분석"):
                     with gr.Row():
                         with gr.Column(scale=6):  # 왼쪽 60%
@@ -254,13 +237,10 @@ def create_ui():
                             
                             with gr.Group(elem_classes="output-group scroll-lock"):
                                 tree_view_output = gr.Textbox(label="파일 구조 (Tree View)", lines=20)
-                            
-                            with gr.Group(elem_classes="output-group scroll-lock"):
-                                file_buttons = gr.HTML(label="파일 리스트")
                         
                         with gr.Column(scale=4):  # 오른쪽 40%
                             with gr.Group(elem_classes="output-group full-height"):
-                                code_tabs = gr.Tabs(elem_classes="code-tabs")
+                                code_tabs = gr.Tabs()
                                 with code_tabs:
                                     app_py_tab = gr.TabItem("app.py")
                                     with app_py_tab:
@@ -270,7 +250,7 @@ def create_ui():
                                         requirements_content = gr.Textbox(label="requirements.txt", lines=30)
 
                 with gr.TabItem("AI 코딩"):
-                    chatbot = gr.Chatbot(label="대화", type="messages")  # 여기를 'messages'로 수정
+                    chatbot = gr.Chatbot(label="대화")
                     msg = gr.Textbox(label="메시지")
                     
                     # 숨겨진 상태로 파라미터 설정
@@ -290,82 +270,27 @@ def create_ui():
                     gr.Examples(examples, inputs=msg)
 
                     def respond_wrapper(message, chat_history, max_tokens, temperature, top_p):
+
                         bot_message = respond(message, chat_history, max_tokens, temperature, top_p)
-                        chat_history.append((message, bot_message))  # 튜플 형식으로 변경
+                        chat_history.append({"role": "user", "content": message})
+                        chat_history.append({"role": "assistant", "content": bot_message})
                         return "", chat_history
 
                     msg.submit(respond_wrapper, [msg, chatbot, 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, summary_output, analysis_output, usage_output]
-            ).then(
-                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)
-            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
 
     except Exception as e:
@@ -373,7 +298,6 @@ def create_ui():
         print(traceback.format_exc())
         raise
 
-
 if __name__ == "__main__":
     try:
         print("Starting HuggingFace Space Analyzer...")
@@ -386,8 +310,8 @@ if __name__ == "__main__":
         
         print("Launching Gradio app...")
         demo.launch(
-            server_name="0.0.0.0",  # 모든 네트워크 인터페이스에서 접근 가능하도록 설정
-            server_port=7860,  # 기본 Gradio 포트
+            server_name="0.0.0.0",
+            server_port=7860,
             share=False,
             debug=True,
             show_api=False
@@ -399,4 +323,4 @@ if __name__ == "__main__":
         print(traceback.format_exc())
         raise
 
-print("Script execution completed.")        
+print("Script execution completed.")