Update Advisernote.py

Advisernote.py

@@ -8,6 +8,7 @@ import warnings
 import platform
 import imaplib
 import email
+import html
 from imaplib import IMAP4_SSL
 from email.header import decode_header, make_header
 from datetime import datetime
@@ -124,7 +125,8 @@ if 'refs' not in st.session_state: st.session_state['refs'] = ""
 if 'folder_path' not in st.session_state: st.session_state['folder_path'] = ""
 if 'doc_context' not in st.session_state: st.session_state['doc_context'] = ""
 if 'results' not in st.session_state: st.session_state['results'] = None
-if 'notion_client_name' not in st.session_state: st.session_state['notion_client_name'] = ""
+if 'notion_client_name' not in st.session_state: st.session_state['notion_client_name'] = "Serafi & Tusiga"
+if 'case_comments' not in st.session_state: st.session_state['case_comments'] = ""
 
 # =========================================================================
 # [보안 끝판왕] 최초 로그인 비번 세팅 & 관리자 실시간 동기화/삭제 통제 시스템
@@ -165,6 +167,25 @@ def update_user_password(email_addr, new_password):
     conn.close()
     return updated
 
+def add_or_restore_user(email_addr, initial_password=""):
+    email_addr = email_addr.strip().lower()
+    conn = sqlite3.connect(USER_DB_FILE)
+    existing = conn.execute("SELECT email FROM users WHERE email = ?", (email_addr,)).fetchone()
+    if existing:
+        conn.execute(
+            "UPDATE users SET password = ?, active = 1 WHERE email = ?",
+            (initial_password, email_addr)
+        )
+    else:
+        conn.execute(
+            "INSERT INTO users (email, password, active) VALUES (?, ?, 1)",
+            (email_addr, initial_password)
+        )
+    conn.commit()
+    updated = conn.total_changes > 0
+    conn.close()
+    return updated
+
 def revoke_user_access(email_addr):
     conn = sqlite3.connect(USER_DB_FILE)
     conn.execute("UPDATE users SET active = 0 WHERE email = ?", (email_addr,))
@@ -296,7 +317,11 @@ def fetch_notion_comments(page_id):
                 created_at = c.get("created_time", "").split("T")[0] # YYYY-MM-DD 형식만 추출
                 
                 c_text_list = c.get("rich_text", [])
-                c_text = c_text_list[0].get("text", {}).get("content", "").strip() if c_text_list else ""
+                c_text = "".join(
+                    item.get("plain_text", item.get("text", {}).get("content", ""))
+                    for item in c_text_list
+                    if isinstance(item, dict)
+                ).strip()
                 
                 if c_text:
                     # 🟢 [수정] [날짜] 코멘트 내용 형식으로 조합
@@ -489,6 +514,31 @@ if check_password():
                 })
             
             st.table(pd.DataFrame(admin_data))
+
+            st.markdown("**Add or Restore User Access**")
+            add_col_email, add_col_password = st.columns([2, 1])
+            new_user_email = add_col_email.text_input(
+                "Employee email address",
+                key="admin_add_user_email",
+                placeholder="new.user@fluxfinance.co.nz"
+            ).strip().lower()
+            new_user_password = add_col_password.text_input(
+                "Initial password",
+                type="password",
+                key="admin_add_user_password",
+                help="Leave blank to let the user set their password on first login."
+            )
+
+            if st.button("➕ Add / Restore User", use_container_width=True):
+                if not new_user_email:
+                    st.error("Please enter an email address.")
+                elif not re.match(r"^[^@\s]+@[^@\s]+\.[^@\s]+$", new_user_email):
+                    st.error("Please enter a valid email address.")
+                elif add_or_restore_user(new_user_email, new_user_password):
+                    st.success(f"✅ Access granted for {new_user_email}.")
+                    st.rerun()
+                else:
+                    st.error("⚠️ System Error: Failed to update user access.")
             
             st.markdown("**Deactivate & Revoke User Permissions**")
             target_del = st.selectbox("Select email address to revoke access", ["-"] + list(current_users.keys()))
@@ -1068,6 +1118,15 @@ if check_password():
                         status.update(label="⚠️ 매칭 데이터 없음", state="error")
                     else:
                         status.update(label=f"❌ Failed: {msg}", state="error")
+
+            case_comments = st.session_state.get('case_comments', '').strip()
+            if case_comments:
+                html_case_comments = html.escape(case_comments).replace('\n', '<br>')
+                st.markdown(f"""
+                    <div style='background-color: #F0FDF4; padding: 12px 15px; border-radius: 8px; border: 1px solid #BBF7D0; font-size: 0.85rem; margin-top: 12px; color: #166534; line-height: 1.6;'>
+                        <b>💬 Case Comments:</b><br><br>{html_case_comments}
+                    </div>
+                """, unsafe_allow_html=True)
     # TAB 2: 이메일 검색창
     with tab_email:                    
         with st.container(border=True):
@@ -1433,7 +1492,7 @@ if st.session_state['results'] is not None:
             # 🟢 [기존 기능 유지] 코멘트(Comments)가 있을 경우, 본문 아래에 별도의 말풍선 박스로 예쁘게 렌더링
             comments_data = row.get('Comments', '').strip()
             if comments_data:
-                html_comments = comments_data.replace('\n', '<br>')
+                html_comments = html.escape(comments_data).replace('\n', '<br>')
                 st.markdown(f"""
                     <div style='background-color: #FEF3C7; padding: 12px 15px; border-radius: 8px; border: 1px solid #FCD34D; font-size: 0.85rem; margin-top: 12px; color: #92400E; line-height: 1.6;'>
                         <b>💬 Notion Comments:</b><br><br>{html_comments}