Update app.py

app.py

@@ -370,8 +370,6 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
 
     with ms.Application() as app:
         with antd.ConfigProvider():
-            
-
             # Drawer 컴포넌트들
             with antd.Drawer(open=False, title="code", placement="left", width="750px") as code_drawer:
                 code_output = legacy.Markdown()
@@ -379,11 +377,30 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
             with antd.Drawer(open=False, title="history", placement="left", width="900px") as history_drawer:
                 history_output = legacy.Chatbot(show_label=False, flushing=False, height=960, elem_classes="history_chatbot")
     
-            with antd.Drawer(open=False, title="Session History", placement="left", width="900px") as session_drawer:
-                session_list = gr.Dropdown(label="Select Session", choices=[])
-                session_history = legacy.Chatbot(show_label=False, height=960)
-            
-
+            with antd.Drawer(
+                open=False, 
+                title="Session History", 
+                placement="left", 
+                width="900px",
+                className="session-drawer"
+            ) as session_drawer:
+                with antd.Flex(vertical=True, gap="middle"):
+                    gr.Markdown("### Previous Sessions")
+                    session_list = gr.Dropdown(
+                        label="Select a session to view history",
+                        choices=[],
+                        className="session-list"
+                    )
+                    session_history = legacy.Chatbot(
+                        show_label=False,
+                        height=960,
+                        elem_classes="session-chatbot"
+                    )
+                    close_btn = antd.Button(
+                        "Close",
+                        type="default",
+                        className="close-btn"
+                    )
 
             # 메인 컨텐츠를 위한 Row
             with antd.Row(gutter=[32, 12]) as layout:    
@@ -521,15 +538,6 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                             historyBtn = antd.Button("📜 history", type="default")
                             sessionBtn = antd.Button("📚 sessions", type="default")
 
-                        # 세션 버튼 이벤트 핸들러
-                        sessionBtn.click(
-                            lambda: (gr.update(open=True), update_session_list()),
-                            inputs=[],
-                            outputs=[session_drawer, session_list]
-                        )
-
-    
-                        
                         gr.HTML('<div class="render_header"><span class="header_btn"></span><span class="header_btn"></span><span class="header_btn"></span></div>')
                         with antd.Tabs(active_key="empty", render_tab_bar="() => null") as state_tab:
                             with antd.Tabs.Item(key="empty"):
@@ -558,48 +566,79 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                     print(f"Error executing code: {str(e)}")
                     return None, gr.update(active_key="empty")
 
-
-
-        # 이벤트 핸들러들
-        execute_btn.click(
-            fn=execute_code,
-            inputs=[input],
-            outputs=[sandbox, state_tab]
-        )
-
-        codeBtn.click(lambda: gr.update(open=True),
-                      inputs=[], outputs=[code_drawer])
-        code_drawer.close(lambda: gr.update(
-            open=False), inputs=[], outputs=[code_drawer])
-
-        historyBtn.click(history_render, inputs=[history], outputs=[history_drawer, history_output])
-        history_drawer.close(lambda: gr.update(
-            open=False), inputs=[], outputs=[history_drawer])
-
-        sessionBtn.click(
-            lambda: (gr.update(open=True), update_session_list()),
-            inputs=[],
-            outputs=[session_drawer, session_list]
-        )
-
-        session_list.change(
-            load_session_history, 
-            inputs=[session_list], 
-            outputs=[session_history]
-        )
-
-        btn.click(
-            demo_instance.generation_code,
-            inputs=[input, setting, history],
-            outputs=[code_output, history, sandbox, state_tab, code_drawer]
-        )
-
-        clear_btn.click(
-            demo_instance.clear_history,
-            inputs=[],
-            outputs=[history]
-        )
-
+            def show_session_history():
+                return (
+                    gr.update(open=True),  # session_drawer
+                    update_session_list(),  # session_list
+                    []  # session_history 초기화
+                )
+
+            def close_session_drawer():
+                return (
+                    gr.update(open=False),  # session_drawer
+                    gr.update(choices=[]),  # session_list
+                    []  # session_history 초기화
+                )
+
+            # 이벤트 핸들러들
+            execute_btn.click(
+                fn=execute_code,
+                inputs=[input],
+                outputs=[sandbox, state_tab]
+            )
+
+            codeBtn.click(
+                lambda: gr.update(open=True),
+                inputs=[], 
+                outputs=[code_drawer]
+            )
+            
+            code_drawer.close(
+                lambda: gr.update(open=False),
+                inputs=[], 
+                outputs=[code_drawer]
+            )
+
+            historyBtn.click(
+                history_render,
+                inputs=[history],
+                outputs=[history_drawer, history_output]
+            )
+            
+            history_drawer.close(
+                lambda: gr.update(open=False),
+                inputs=[],
+                outputs=[history_drawer]
+            )
+
+            # 세션 관련 이벤트 핸들러
+            sessionBtn.click(
+                show_session_history,
+                outputs=[session_drawer, session_list, session_history]
+            )
+
+            close_btn.click(
+                close_session_drawer,
+                outputs=[session_drawer, session_list, session_history]
+            )
+
+            session_list.change(
+                load_session_history,
+                inputs=[session_list],
+                outputs=[session_history]
+            )
+
+            btn.click(
+                demo_instance.generation_code,
+                inputs=[input, setting, history],
+                outputs=[code_output, history, sandbox, state_tab, code_drawer]
+            )
+
+            clear_btn.click(
+                demo_instance.clear_history,
+                inputs=[],
+                outputs=[history]
+            )
 
 if __name__ == "__main__":
     try: