Update app.py

app.py

@@ -346,19 +346,38 @@ def clear_expired_sessions():
     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]
 
+def update_session_list():
+    try:
+        conn = sqlite3.connect('chat_history.db')
+        c = conn.cursor()
+        c.execute("SELECT session_id, created_at FROM sessions ORDER BY created_at DESC")
+        sessions = c.fetchall()
+        conn.close()
+        # 날짜와 함께 보기 좋게 포맷팅
+        formatted_sessions = [f"{sid} ({datetime.strptime(ctime, '%Y-%m-%d %H:%M:%S.%f').strftime('%Y-%m-%d %H:%M:%S')})" 
+                            for sid, ctime in sessions]
+        return gr.update(choices=formatted_sessions)
+    except Exception as e:
+        print(f"Error updating session list: {e}")
+        return gr.update(choices=[])
 
+def load_session_history(selected_session):
+    try:
+        if not selected_session:
+            return []
+        # 세션 ID 추출 (날짜 부분 제거)
+        session_id = selected_session.split(" (")[0]
+        history = get_session_history(session_id)
+        # 채팅 형식으로 변환
+        chat_history = []
+        for prompt, response, timestamp in history:
+            chat_history.append([prompt, response])
+        return chat_history
+    except Exception as e:
+        print(f"Error loading session history: {e}")
+        return []
 # Demo 인스턴스 먼저 생성
 demo_instance = Demo()
 
@@ -380,7 +399,7 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
             with antd.Drawer(
                 open=False, 
                 title="Session History", 
-                placement="left", 
+                placement="right", 
                 width="900px",
                 elem_classes="session-drawer"
             ) as session_drawer:
@@ -582,6 +601,9 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
             history_drawer.close(lambda: gr.update(
                 open=False), inputs=[], outputs=[history_drawer])
 
+
+
+            
             # 세션 관련 이벤트 핸들러들
             sessionBtn.click(
                 lambda: (gr.update(open=True), update_session_list()),
@@ -600,6 +622,8 @@ with gr.Blocks(css_paths="app.css",theme=theme) as demo:
                 outputs=[session_drawer, session_list, session_history]
             )
 
+
+
             session_list.change(
                 load_session_history, 
                 inputs=[session_list],