Spaces:
Runtime error
Runtime error
| import logging | |
| import time | |
| import streamlit as st | |
| from db import ( | |
| debate_ip_column_available, | |
| ensure_db_login, | |
| fetch_debate_status, | |
| fetch_live_messages, | |
| fetch_opinion_change, | |
| fetch_pending_teacher_accounts, | |
| fetch_topic_data, | |
| init_db, | |
| opinion_changes_available, | |
| submit_opinion, | |
| update_topic, | |
| using_service_role_key, | |
| ) | |
| from config import APP_CSS, MAX_STUDENT_NAME_LEN | |
| from utils import anonymize_ip, get_client_ip, get_kst_now_str, log_audit | |
| from validators import validate_opinion_content, validate_student_name | |
| from views.home import render_home_page | |
| from views.lobby import render_lobby_page | |
| from components.admin_panel import render_admin_page | |
| from components.sidebar import render_sidebar | |
| logger = logging.getLogger("talk_trace_ai") | |
| if not logger.handlers: | |
| logging.basicConfig( | |
| level=logging.INFO, | |
| format="%(asctime)s | %(levelname)s | %(name)s | %(message)s" | |
| ) | |
| supabase = init_db() | |
| if not using_service_role_key(): | |
| ensure_db_login(supabase) | |
| st.set_page_config(page_title="말자취(Talk-Trace) AI", layout="wide") | |
| st.markdown(APP_CSS, unsafe_allow_html=True) | |
| if 'reset_key' not in st.session_state: st.session_state['reset_key'] = 0 | |
| if 'ai_hint_text' not in st.session_state: st.session_state['ai_hint_text'] = "" | |
| if 'ai_report_text' not in st.session_state: st.session_state['ai_report_text'] = "" | |
| if 'page' not in st.session_state: st.session_state['page'] = "home" | |
| if 'current_room' not in st.session_state: st.session_state['current_room'] = "" | |
| if 'joined' not in st.session_state: st.session_state['joined'] = False | |
| if 'teacher_auth' not in st.session_state: st.session_state['teacher_auth'] = False | |
| if 'admin_auth' not in st.session_state: st.session_state['admin_auth'] = False | |
| if 'teacher_id' not in st.session_state: st.session_state['teacher_id'] = "" | |
| if 'is_working' not in st.session_state: st.session_state['is_working'] = False | |
| if 'ai_hint_manual_mode' not in st.session_state: st.session_state['ai_hint_manual_mode'] = False | |
| if '_last_debate_status' not in st.session_state: st.session_state['_last_debate_status'] = None | |
| if st.session_state['page'] != "home": | |
| col_home_btn, _ = st.columns([1, 7]) | |
| with col_home_btn: | |
| if st.button("🏠 홈", use_container_width=True): | |
| st.session_state['page'] = "home" | |
| st.session_state['joined'] = False | |
| st.session_state['teacher_auth'] = False | |
| st.rerun() | |
| if st.session_state['page'] == "home": | |
| render_home_page() | |
| # plotly / google.generativeai 임포트는 홈 화면에서 불필요 — st.stop() 이후에 로드 | |
| from components.chat_board import render_chat_board | |
| from components.teacher_dashboard import render_teacher_dashboard | |
| from components.opinion_change import render_pre_opinion_form, render_post_opinion_section | |
| sidebar_ctx = render_sidebar(supabase) | |
| user_role = sidebar_ctx['user_role'] | |
| room_name = sidebar_ctx['room_name'] | |
| teacher_auth = sidebar_ctx['teacher_auth'] | |
| admin_auth = sidebar_ctx['admin_auth'] | |
| student_name = sidebar_ctx['student_name'] | |
| student_number = sidebar_ctx['student_number'] | |
| # admin 첫 접속 시 pending 여부에 따라 첫 화면 결정 (최초 1회) | |
| if admin_auth and not st.session_state.get('_admin_redirected'): | |
| _pending = fetch_pending_teacher_accounts(supabase) | |
| st.session_state['page'] = "admin_approval" if _pending else "lobby" | |
| st.session_state['joined'] = False | |
| st.session_state['_admin_redirected'] = True | |
| st.rerun() | |
| if st.session_state['page'] == "admin_approval": | |
| render_admin_page(supabase, user_role, teacher_auth, admin_auth) | |
| if not st.session_state['joined']: | |
| render_lobby_page(supabase, user_role, teacher_auth, room_name, student_number) | |
| # === Joined room view === | |
| topic_data = fetch_topic_data(supabase, room_name) | |
| current_topic = topic_data.get('title', "자유 주제로 대화해 봅시다.") | |
| current_mode = topic_data.get('mode', "⚔️ 찬반 토론") | |
| act_type = "토론" if "토론" in current_mode else "토의" | |
| if admin_auth: | |
| col_title, col_btn1, col_btn2 = st.columns([4, 1, 1]) | |
| with col_title: | |
| st.title(f"🎙️ 말자취(Talk-Trace) AI [{room_name}]") | |
| with col_btn1: | |
| if st.button("📝 ID 요청 수락", use_container_width=True): | |
| st.session_state['page'] = "admin_approval" | |
| st.rerun() | |
| with col_btn2: | |
| if st.button("🚪 말자취 AI 대기실", use_container_width=True): | |
| st.session_state['page'] = "lobby" | |
| st.rerun() | |
| else: | |
| st.title(f"🎙️ 말자취(Talk-Trace) AI [{room_name}]") | |
| st.info(f"**현재 주제:** {current_topic} ({current_mode})") | |
| if user_role == "교사" and teacher_auth: | |
| with st.expander("✏️ 주제 수정", expanded=False): | |
| _edit_title = st.text_input("새 주제", value=current_topic, max_chars=120, key="edit_topic_title") | |
| _edit_mode = st.radio( | |
| "진행 방식", ["⚔️ 찬반 토론", "💡 자유 토의"], | |
| index=0 if "토론" in current_mode else 1, | |
| horizontal=True, key="edit_topic_mode", | |
| ) | |
| if st.button("✅ 주제 저장", type="primary", use_container_width=True, key="edit_topic_save"): | |
| if not _edit_title.strip(): | |
| st.warning("주제를 입력해 주세요.") | |
| else: | |
| if update_topic(supabase, room_name, _edit_title.strip(), _edit_mode) is not None: | |
| st.toast("✅ 주제가 수정되었습니다.", icon="✏️") | |
| st.rerun() | |
| def _poll_debate_status(room_name): | |
| """학생 화면에서 5초마다 토론 상태를 확인하고 변경 시 전체 rerun.""" | |
| current = fetch_debate_status(supabase, room_name) | |
| if current != st.session_state.get("_last_debate_status"): | |
| st.session_state["_last_debate_status"] = current | |
| st.rerun(scope="app") | |
| def _render_opinion_input(supabase, room_name, user_role, student_name, student_number, current_mode): | |
| st.subheader("🗣️ 내 의견 작성") | |
| col_input, col_stt = st.columns([4, 1]) | |
| with col_input: | |
| user_input = st.text_area( | |
| "의견을 입력하세요", | |
| key=f"input_{st.session_state['reset_key']}", | |
| height=80, | |
| label_visibility="collapsed", | |
| ) | |
| opts = ["🔵 찬성", "🔴 반대"] if current_mode == "⚔️ 찬반 토론" else ["💡 아이디어", "➕ 보충", "❓ 질문"] | |
| sentiment = st.radio("의견 성격", opts, horizontal=True) | |
| with col_stt: | |
| st.components.v1.html( | |
| """ | |
| <button id="stt-btn" style="width:100%; height:80px; font-weight:bold; border-radius:10px; background-color:#e8f0fe; border:1px solid #1a73e8; color:#1a73e8; cursor:pointer;">🎤 음성 입력 시작</button> | |
| <p id="status" style="font-size:11px; color:gray; text-align:center; margin-top:5px;">대기 중...</p> | |
| <script> | |
| const btn = document.getElementById('stt-btn'); | |
| const status = document.getElementById('status'); | |
| const recognition = new (window.SpeechRecognition || window.webkitSpeechRecognition)(); | |
| recognition.lang = 'ko-KR'; | |
| let isRecognizing = false; | |
| btn.onclick = () => { if (!isRecognizing) recognition.start(); else recognition.stop(); }; | |
| recognition.onstart = () => { isRecognizing = true; status.innerText = "듣는 중..."; btn.style.backgroundColor = "#ff4b4b"; btn.style.color = "white"; btn.innerHTML = "🛑 음성 입력 중지"; }; | |
| recognition.onresult = (event) => { | |
| const text = event.results[0][0].transcript; | |
| const textArea = window.parent.document.querySelector('textarea[aria-label="의견을 입력하세요"]'); | |
| if (textArea) { | |
| const nativeInputValueSetter = Object.getOwnPropertyDescriptor(window.HTMLTextAreaElement.prototype, "value").set; | |
| nativeInputValueSetter.call(textArea, textArea.value + " " + text); | |
| textArea.dispatchEvent(new Event('input', { bubbles: true })); | |
| status.innerText = "입력 완료!"; | |
| } | |
| }; | |
| recognition.onend = () => { isRecognizing = false; setTimeout(() => { status.innerText = "대기 중..."; btn.style.backgroundColor = "#e8f0fe"; btn.style.color = "#1a73e8"; btn.innerHTML = "🎤 음성 입력 시작"; }, 1500); }; | |
| </script> | |
| """, | |
| height=120, | |
| ) | |
| _elapsed = time.time() - st.session_state.get('last_submit_ts', 0) | |
| _submit_cooldown = _elapsed < 3 | |
| _submit_disabled = _submit_cooldown or st.session_state.get('is_working', False) | |
| if st.button("의견 제출", use_container_width=True, type="primary", disabled=_submit_disabled): | |
| if st.session_state.get('is_working', False): | |
| st.stop() | |
| st.session_state['is_working'] = True | |
| input_ok, safe_input, input_error_code, input_error_message = validate_opinion_content(user_input, max_len=700) | |
| student_ok, safe_student_name, student_error_code, student_error_message = validate_student_name(student_name, max_len=MAX_STUDENT_NAME_LEN) | |
| student_number_ok, safe_student_number, _, student_number_error_message = validate_student_name(student_number, max_len=20) | |
| if not student_number_ok and user_role == "학생": | |
| st.session_state['is_working'] = False | |
| st.error(f"❌ {student_number_error_message}") | |
| st.stop() | |
| if user_role == "학생" and (not safe_student_name or safe_student_name == "익명"): | |
| safe_student_name = safe_student_number or "학번미입력" | |
| if input_ok and safe_input: | |
| now = get_kst_now_str() | |
| author_role_for_submit = "교사" if user_role == "교사" else "학생" | |
| client_ip = get_client_ip() | |
| insert_payload = { | |
| "room_name": room_name, "timestamp": now, "student_name": safe_student_name, | |
| "content": safe_input, "sentiment": sentiment, "author_role": author_role_for_submit, | |
| } | |
| if debate_ip_column_available() and client_ip: | |
| anonymized_ip = anonymize_ip(client_ip) | |
| if anonymized_ip: | |
| insert_payload["ip_address"] = anonymized_ip | |
| try: | |
| res = submit_opinion(supabase, insert_payload) | |
| if res is None: | |
| st.session_state['is_working'] = False | |
| st.stop() | |
| st.session_state['last_submit_ts'] = time.time() | |
| st.session_state['is_working'] = False | |
| fetch_live_messages.clear() | |
| log_audit( | |
| "opinion_submitted", | |
| room_name=room_name, actor_name=safe_student_name, | |
| role=author_role_for_submit, sentiment=sentiment, | |
| client_ip=client_ip if client_ip else "N/A", | |
| ) | |
| st.session_state['reset_key'] += 1 | |
| st.rerun(scope="app") | |
| except Exception as e: | |
| st.session_state['is_working'] = False | |
| st.error(f"저장 실패: {e}") | |
| else: | |
| st.session_state['is_working'] = False | |
| st.warning(f"{input_error_message}") | |
| if _submit_cooldown: | |
| st.caption("⏳ 제출 성공 후 3초간 의견을 제출할 수 없습니다.") | |
| time.sleep(0.5) | |
| st.rerun() | |
| if user_role == "학생" and opinion_changes_available(): | |
| _poll_debate_status(room_name) | |
| debate_status = fetch_debate_status(supabase, room_name) | |
| # 학생에게 토론 진행 상태를 항상 명시적으로 표시 | |
| if debate_status == "ended": | |
| st.warning(f"🔴 **{act_type} 종료** — 아래에서 {act_type} 후 생각 변화를 기록해주세요.") | |
| else: | |
| st.success(f"🟢 **{act_type} 진행 중** — 자유롭게 의견을 나눠보세요.") | |
| row = fetch_opinion_change(supabase, room_name, student_name) | |
| has_pre_opinion = bool((row or {}).get("pre_opinion")) | |
| if debate_status == "ended": | |
| if has_pre_opinion: | |
| render_post_opinion_section(supabase, room_name, student_name, act_type, current_topic) | |
| else: | |
| st.warning(f"{'토론' if act_type == '토론' else '토의'}이 종료되었습니다. 토론 전 생각을 미리 기록하지 않아 참여할 수 없습니다.") | |
| else: | |
| if not has_pre_opinion: | |
| render_pre_opinion_form(supabase, room_name, student_name, current_topic, act_type) | |
| st.caption("💡 위에서 토론 전 생각을 제출하면 의견 작성이 활성화됩니다.") | |
| else: | |
| _render_opinion_input(supabase, room_name, user_role, student_name, student_number, current_mode) | |
| else: | |
| _render_opinion_input(supabase, room_name, user_role, student_name, student_number, current_mode) | |
| st.divider() | |
| render_chat_board(supabase, room_name, user_role, teacher_auth, student_name, current_mode, act_type) | |
| if user_role == "교사" and teacher_auth: | |
| render_teacher_dashboard(supabase, room_name, user_role, student_name, current_topic, current_mode, act_type) | |