Update app.py

app.py

@@ -324,6 +324,18 @@ def clear_expired_sessions():
         conn.commit()
     finally:
         conn.close()
+
+def update_session_list():
+    conn = sqlite3.connect('chat_history.db')
+    c = conn.cursor()
+    c.execute("SELECT session_id FROM sessions ORDER BY created_at DESC")
+    sessions = [row[0] for row in c.fetchall()]
+    conn.close()
+    return gr.update(choices=sessions)
+
+def load_session_history(session_id):
+    history = get_session_history(session_id)
+    return [[p, r] for p, r, _ in history]
         
 with gr.Blocks(css_paths="app.css",theme=theme) as demo:
     history = gr.State([])
@@ -331,12 +343,23 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
         "system": SystemPrompt,
     })
 
-
-
     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()
+    
+            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)
+            
+
+
             # 메인 컨텐츠를 위한 Row
-            with antd.Row(gutter=[32, 12]) as layout:
+            with antd.Row(gutter=[32, 12]) as layout:    
                 # 좌측 패널
                 with antd.Col(span=24, md=8):
                     with antd.Flex(vertical=True, gap="middle", wrap=True):
@@ -463,27 +486,22 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                             outputs=[input]
                         )     
                                            
-                    # Drawer 컴포넌트들
-                    with antd.Drawer(open=False, title="code", placement="left", width="750px") as code_drawer:
-                        code_output = legacy.Markdown()
-
-                    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.Col(span=24, md=16):
                     with ms.Div(elem_classes="right_panel"):
                         with antd.Flex(gap="small", elem_classes="setting-buttons"):
                             codeBtn = antd.Button("🧑‍💻 view code", type="default")
                             historyBtn = antd.Button("📜 history", type="default")
-                            sessionBtn = antd.Button("📚 sessions", 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:
@@ -513,58 +531,49 @@ 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])
+        # 이벤트 핸들러들
+        execute_btn.click(
+            fn=execute_code,
+            inputs=[input],
+            outputs=[sandbox, state_tab]
+        )
 
-            
-            btn.click(
-                demo_instance.generation_code,  # Demo 인스턴스의 메서드 사용
-                inputs=[input, setting, history],
-                outputs=[code_output, history, sandbox, state_tab, code_drawer]
-            )
+        codeBtn.click(lambda: gr.update(open=True),
+                      inputs=[], outputs=[code_drawer])
+        code_drawer.close(lambda: gr.update(
+            open=False), inputs=[], outputs=[code_drawer])
 
-            clear_btn.click(
-                demo_instance.clear_history,  # Demo 인스턴스의 메서드 사용
-                inputs=[],
-                outputs=[history]
-            )
+        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]
+        )
 
 
-    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)
-        
-    def update_session_list():
-        conn = sqlite3.connect('chat_history.db')
-        c = conn.cursor()
-        c.execute("SELECT session_id FROM sessions ORDER BY created_at DESC")
-        sessions = [row[0] for row in c.fetchall()]
-        conn.close()
-        return gr.update(choices=sessions)
-        
-    def load_session_history(session_id):
-        history = get_session_history(session_id)
-        return [[p, r] for p, r, _ in history]
-        
-    session_list.change(load_session_history, 
-                       inputs=[session_list], 
-                       outputs=[session_history])
-    
 if __name__ == "__main__":
     init_db()
     clear_expired_sessions()  # 만료된 세션 정리