# ============================================================ # app.py — 상상파인더 UI # ============================================================ import os, re, json, hashlib, warnings from datetime import datetime import html import streamlit as st from recommend import ( load_notices_from_supabase, two_tower_recommend, classify_job_type, get_supabase, ) from api.services.search_service import hybrid_search as pinecone_hybrid_search, _profile_summary try: from dotenv import load_dotenv load_dotenv() except ImportError: pass warnings.filterwarnings("ignore") os.environ["TRANSFORMERS_VERBOSITY"] = "error" # ── 경로 설정 ───────────────────────────────────────────────── _BASE_DIR = os.path.dirname(os.path.abspath(__file__)) GEMINI_API_KEY = os.getenv("GEMINI_API_KEY") os.makedirs(os.path.join(_BASE_DIR, "data"), exist_ok=True) # ── 신버전 카테고리 ─────────────────────────────────────────── CATEGORIES = [ "취업/채용", "학사행정", "학생활동/비교과", "대외활동", "공모전/경진대회", "국제교류", "창업", "장학금", "기숙사", "ROTC" ] PINECONE_CATEGORY_MAP = { "취업/채용": ["취업/채용", "인턴십"], "학사행정": ["학사행정"], "학생활동/비교과": ["학생활동/비교과", "비교과", "교육/특강"], "대외활동": ["대외활동", "봉사/서포터즈"], "공모전/경진대회": ["공모전/경진대회"], "국제교류": ["국제교류"], "창업": ["창업"], "장학금": ["장학금", "학자금/근로장학"], "기숙사": ["기숙사", "기숙사/생활관"], "ROTC": ["ROTC"], } COLLEGE_MAP = { "크리에이티브인문예술대학": ["영미문화콘텐츠트랙", "영미언어정보트랙", "한국어교육트랙", "역사문화큐레이션트랙", "역사콘텐츠트랙", "지식정보문화트랙", "디지털인문정보학트랙", "동양화전공", "서양화전공", "한국무용전공", "현대무용전공", "발레전공"], "미래융합사회과학대학": ["국제무역트랙", "글로벌비지니스트랙", "기업ㆍ경제분석트랙", "경제금융투자트랙", "공공행정트랙", "법&정책트랙", "부동산트랙", "스마트도시ㆍ교통계획트랙", "기업경영트랙", "비지니스애널리틱스트랙", "회계ㆍ재무경영트랙"], "디자인대학": ["패션마케팅트랙", "패션디자인트랙", "패션크리에이티브디렉션트랙", "미디어디자인트랙", "시각디자인트랙", "영상ㆍ애니메이션디자인트랙", "UX/UI디자인트랙", "인테리어디자인트랙", "VMDㆍ전시디자인트랙", "게임그래픽디자인트랙", "뷰티디자인매니지먼트학과"], "IT공과대학": ["모바일소프트웨어트랙", "빅데이터트랙", "디지털콘텐츠ㆍ가상현실트랙", "웹공학트랙", "전자트랙", "시스템반도체트랙", "기계시스템디자인트랙", "AI로봇융합트랙", "산업공학트랙", "응용산업데이터공학트랙"], "창의융합대학": ["상상력인재학부", "문학문화콘텐츠학과", "AI응용학과", "융합보안학과", "미래모빌리티학과"], "글로벌인재대학": ["한국언어문화교육학과", "글로벌K비지니스학과", "영상엔터테인먼트학과", "패션뷰티크리에이션학과", "SW융합학과", "글로벌벤처창업학과"], "미래플러스대학": ["융합행정학과", "호텔외식경영학과", "뷰티디자인학과", "비지니스컨설팅학과", "ICT융합디자인학과", "AIㆍ소프트웨어학과", "뷰티매니지먼트학과", "디지털콘텐츠디자인학과", "인테리어디자인학과", "스마트제조혁신컨설팅학과"], } DEPT_URLS = { "한국언어문화교육학과":"https://www.hansung.ac.kr/global/1511/subview.do","글로벌K비지니스학과":"https://www.hansung.ac.kr/global/1516/subview.do","영상엔터테인먼트학과":"https://www.hansung.ac.kr/global/1521/subview.do","패션뷰티크리에이션학과":"https://www.hansung.ac.kr/global/1526/subview.do","SW융합학과":"https://www.hansung.ac.kr/global/1531/subview.do","글로벌벤처창업학과":"https://www.hansung.ac.kr/global/6807/subview.do","융합행정학과":"https://www.hansung.ac.kr/futureplus/731/subview.do","호텔외식경영학과":"https://www.hansung.ac.kr/futureplus/734/subview.do","뷰티디자인학과":"https://www.hansung.ac.kr/futureplus/737/subview.do","비지니스컨설팅학과":"https://www.hansung.ac.kr/futureplus/740/subview.do","ICT융합디자인학과":"https://www.hansung.ac.kr/futureplus/743/subview.do","AIㆍ소프트웨어학과":"https://www.hansung.ac.kr/futureplus/746/subview.do","뷰티매니지먼트학과":"https://www.hansung.ac.kr/futureplus/749/subview.do","디지털콘텐츠디자인학과":"https://www.hansung.ac.kr/futureplus/754/subview.do","인테리어디자인학과":"https://www.hansung.ac.kr/futureplus/759/subview.do","스마트제조혁신컨설팅학과":"https://www.hansung.ac.kr/futureplus/764/subview.do", "상상력인재학부":"https://www.hansung.ac.kr/CreCon/2761/subview.do","문학문화콘텐츠학과":"https://www.hansung.ac.kr/CreCon/2768/subview.do","AI응용학과":"https://www.hansung.ac.kr/CreCon/2777/subview.do","융합보안학과":"https://www.hansung.ac.kr/CreCon/2787/subview.do","미래모빌리티학과":"https://www.hansung.ac.kr/CreCon/2796/subview.do", "국제무역트랙":"https://www.hansung.ac.kr/SclScn/5260/subview.do","글로벌비지니스트랙":"https://www.hansung.ac.kr/SclScn/5267/subview.do","기업ㆍ경제분석트랙":"https://www.hansung.ac.kr/SclScn/5274/subview.do","경제금융투자트랙":"https://www.hansung.ac.kr/SclScn/5281/subview.do","공공행정트랙":"https://www.hansung.ac.kr/SclScn/5295/subview.do","법&정책트랙":"https://www.hansung.ac.kr/SclScn/5303/subview.do","부동산트랙":"https://www.hansung.ac.kr/SclScn/5313/subview.do","스마트도시ㆍ교통계획트랙":"https://www.hansung.ac.kr/SclScn/5321/subview.do","기업경영트랙":"https://www.hansung.ac.kr/SclScn/5328/subview.do","비지니스애널리틱스트랙":"https://www.hansung.ac.kr/SclScn/5336/subview.do","회계ㆍ재무경영트랙":"https://www.hansung.ac.kr/SclScn/5344/subview.do", "모바일소프트웨어트랙":"https://www.hansung.ac.kr/Engineering/4887/subview.do","빅데이터트랙":"https://www.hansung.ac.kr/Engineering/4894/subview.do","디지털콘텐츠ㆍ가상현실트랙":"https://www.hansung.ac.kr/Engineering/4901/subview.do","웹공학트랙":"https://www.hansung.ac.kr/Engineering/4908/subview.do","전자트랙":"https://www.hansung.ac.kr/Engineering/4915/subview.do","시스템반도체트랙":"https://www.hansung.ac.kr/Engineering/4922/subview.do","기계시스템디자인트랙":"https://www.hansung.ac.kr/Engineering/4929/subview.do","AI로봇융합트랙":"https://www.hansung.ac.kr/Engineering/4936/subview.do","산업공학트랙":"https://www.hansung.ac.kr/Engineering/4992/subview.do","응용산업데이터공학트랙":"https://www.hansung.ac.kr/Engineering/5020/subview.do", "패션마케팅트랙":"https://www.hansung.ac.kr/Design/5103/subview.do","패션디자인트랙":"https://www.hansung.ac.kr/Design/5110/subview.do","패션크리에이티브디렉션트랙":"https://www.hansung.ac.kr/Design/5117/subview.do","미디어디자인트랙":"https://www.hansung.ac.kr/Design/5124/subview.do","시각디자인트랙":"https://www.hansung.ac.kr/Design/5145/subview.do","영상ㆍ애니메이션디자인트랙":"https://www.hansung.ac.kr/Design/5131/subview.do","UX/UI디자인트랙":"https://www.hansung.ac.kr/Design/5173/subview.do","인테리어디자인트랙":"https://www.hansung.ac.kr/Design/5159/subview.do","VMDㆍ전시디자인트랙":"https://www.hansung.ac.kr/Design/5152/subview.do","게임그래픽디자인트랙":"https://www.hansung.ac.kr/Design/5166/subview.do","뷰티디자인매니지먼트학과":"https://www.hansung.ac.kr/Design/5180/subview.do", "영미문화콘텐츠트랙":"https://www.hansung.ac.kr/HmnArt/5641/subview.do","영미언어정보트랙":"https://www.hansung.ac.kr/HmnArt/5577/subview.do","한국어교육트랙":"https://www.hansung.ac.kr/HmnArt/5584/subview.do","역사문화큐레이션트랙":"https://www.hansung.ac.kr/HmnArt/5627/subview.do","역사콘텐츠트랙":"https://www.hansung.ac.kr/HmnArt/5634/subview.do","지식정보문화트랙":"https://www.hansung.ac.kr/HmnArt/5613/subview.do","디지털인문정보학트랙":"https://www.hansung.ac.kr/HmnArt/5620/subview.do","동양화전공":"https://www.hansung.ac.kr/HmnArt/5648/subview.do","서양화전공":"https://www.hansung.ac.kr/HmnArt/5655/subview.do","한국무용전공":"https://www.hansung.ac.kr/HmnArt/5662/subview.do","현대무용전공":"https://www.hansung.ac.kr/HmnArt/5669/subview.do","발레전공":"https://www.hansung.ac.kr/HmnArt/5676/subview.do", } # ============================================================ # 유틸 # ============================================================ def _load_image_b64(filename): import base64 try: with open(os.path.join(_BASE_DIR, filename), "rb") as f: return base64.b64encode(f.read()).decode() except: return "" def get_logo_base64(): return _load_image_b64("logo.png") def get_hsu_base64(): return _load_image_b64("hsu.png") # ============================================================ # 모델 로더 # ============================================================ def summarize_notice(title, body): import html as _html body = body or '' clean_body = re.sub(r'<[^>]+>', '', body) result = clean_body[:200] + '...' if len(clean_body) > 200 else clean_body return _html.escape(result) def get_gemini_model(api_key): try: import google.generativeai as genai genai.configure(api_key=api_key) return genai.GenerativeModel("gemini-2.5-flash") except Exception as e: st.error(f"[Gemini 오류] {e}"); return None def _result_content_for_llm(result: dict, body_map: dict[str, str] | None = None) -> str: content = str(result.get("content") or "").strip() if len(content) >= 80: return content if body_map: fallback = str(body_map.get(result.get("url"), "")).strip() if fallback: return fallback return content def generate_llm_reply(user_query, results, profile, is_first=False): model = get_gemini_model(GEMINI_API_KEY) if GEMINI_API_KEY else None if not model: return f"총 {len(results)}개의 관련 공지를 찾았습니다." if results else "관련 공지를 찾지 못했습니다." if not results: return "관련 공지를 찾지 못했습니다. 다른 키워드로 검색해보세요." top_results = results[:3] needs_body_fallback = any(len(str(r.get("content") or "").strip()) < 80 for r in top_results) body_map = {} if needs_body_fallback: notices_data = load_notices_from_supabase() body_map = {n["url"]: n.get("body", "") for n in notices_data} context_parts = [] for i, r in enumerate(top_results, start=1): body = _result_content_for_llm(r, body_map)[:800] context_parts.append( f"[공지 {i}]\n제목: {r.get('title', '')}\n날짜: {r.get('date', '')}\n내용: {body if body else '(본문 없음)'}" ) context = "\n\n".join(context_parts) greeting = f"{profile.get('name','')}님, 안녕하세요. " if is_first else "" route = (results[0].get("query_routing") or {}).get("route") if results else {} personalization = (route or {}).get("personalization") or {} profile_summary = personalization.get("profile_summary") or _profile_summary(profile) personalization_instruction = "" if personalization.get("enabled"): personalization_instruction = ( "\n- 사용자 프로필을 반영한 질문입니다. 각 추천 공지마다 사용자의 학과/학년/관심사/지역과 " "왜 맞는지 또는 지원 가능성과 관련된 근거를 한 줄로 덧붙이세요." ) if profile_summary: personalization_instruction += f"\n- 사용자 프로필: {profile_summary}" prompt = f"""당신은 한성대학교 공지사항 안내 도우미입니다. 아래 공지사항 본문을 바탕으로 사용자 질문에 직접적이고 구체적으로 답변하세요. - 날짜, 금액, 조건 등 구체적인 정보가 있으면 반드시 포함하세요. - "공지를 참고하세요" 같은 말은 절대 하지 마세요. - 2~3문장으로 간결하게 답변하세요. - 답변 시작: "{greeting}" {personalization_instruction} [공지 본문] {context} [질문] {user_query}""" try: return get_gemini_model(GEMINI_API_KEY).generate_content(prompt).text.strip() except Exception as e: return f"[Gemini 오류] {e}" # ============================================================ # CSS # ============================================================ GLOBAL_CSS = """ """ # ============================================================ # 알림 대상자 반환 # ============================================================ def _notification_user_id(profile: dict) -> str: phone = profile.get("phone") or "" seed = phone or profile.get("name") or json.dumps(profile, ensure_ascii=False, sort_keys=True) return hashlib.sha256(seed.encode("utf-8")).hexdigest()[:24] def sync_profile_to_supabase_user(profile: dict) -> None: """로컬 온보딩 프로필을 Supabase users 테이블에 업서트한다.""" if not profile.get("name"): return user_row = { "user_id": _notification_user_id(profile), "name": profile.get("name", ""), "phone": profile.get("phone"), "interests": profile.get("interests") or [], "track": profile.get("track", ""), "college": profile.get("college", ""), "grade": profile.get("grade", ""), "income_level": profile.get("income_level"), "gpa": profile.get("gpa"), "region": profile.get("region"), "loan": profile.get("loan"), "gender": profile.get("gender"), "dorm_interest": profile.get("dorm_interest") or [], "rotc_interest": profile.get("rotc_interest", False), } supabase_url = os.getenv("SUPABASE_URL") supabase_key = os.getenv("SUPABASE_SERVICE_ROLE_KEY") or os.getenv("SUPABASE_KEY") try: if supabase_url and supabase_key and not supabase_url.startswith("your_"): try: get_supabase().table("users").upsert(user_row, on_conflict="user_id").execute() except Exception as rest_error: if os.getenv("SUPABASE_DB_URL"): print(f"Supabase REST users 동기화 실패, DB 직접 연결로 재시도: {rest_error}") _sync_profile_to_supabase_user_via_db(user_row) else: raise else: _sync_profile_to_supabase_user_via_db(user_row) except Exception as e: print(f"Supabase users 동기화 실패: {e}") def _sync_profile_to_supabase_user_via_db(user_row: dict) -> None: """SUPABASE_DB_URL만 있는 로컬 환경에서 users 테이블 생성 후 업서트한다.""" from crawling.supabase_store import ensure_schema, _connect ensure_schema() with _connect() as conn: with conn.cursor() as cur: cur.execute( """ insert into users (user_id, name, phone, interests, track, updated_at) values (%s, %s, %s, %s, %s, now()) on conflict (user_id) do update set name = excluded.name, phone = excluded.phone, interests = excluded.interests, track = excluded.track, updated_at = now() """, ( user_row["user_id"], user_row["name"], user_row["phone"], user_row["interests"], user_row["track"], ), ) conn.commit() def get_notification_targets(notices: list[dict], users: list[dict] | None = None) -> list[dict]: """ 신규 공지 리스트를 받아 알림 대상자 정보 반환. 알림 담당자에게 전달할 데이터 구조: [ { "notice_id": "222025", "title": "공지 제목", "category": "장학금", "category_type": ["장학금"], "url": "https://...", "user_name": "홍길동", "phone": "010-1234-5678", "track": "영미문화콘텐츠트랙", "interests": ["장학금", "취업/채용"], }, ... ] """ if users is None: try: res = get_supabase().table("users").select("user_id,name,phone,interests,track").execute() users = res.data or [] except Exception as e: print(f"users 테이블 로드 오류: {e}") return [] targets = [] for notice in notices: category = notice.get('category', '') category_type = notice.get('category_type') or notice.get('job_types') or [] if isinstance(category_type, str): try: category_type = json.loads(category_type) except: category_type = [] for user in users: if not user or not user.get('phone'): continue interests = user.get('interests', []) if isinstance(interests, str): try: interests = json.loads(interests) except: interests = [interests] interest_set = set(interests) matched = category in interest_set or bool(interest_set.intersection(category_type)) if not matched: continue targets.append({ "notice_db_id": notice.get('id'), "notice_id": notice.get('notice_id', ''), "title": notice.get('title', ''), "category": category, "category_type": category_type, "url": notice.get('url', ''), "user_id": user.get('user_id') or user.get('id') or user.get('phone'), "user_name": user.get('name', ''), "phone": user.get('phone'), "track": user.get('track', ''), "interests": interests, }) return targets # ============================================================ # 필터링 함수 # ============================================================ def filter_scholarships(profile) -> tuple: """(신청 공지 리스트, 관련 공지 리스트) 반환""" try: today = datetime.now().strftime('%Y-%m-%d') income_level = profile.get('income_level') gpa = profile.get('gpa') region = profile.get('region') loan = profile.get('loan') grade = profile.get('grade', '') grade_num = None try: grade_num = int(grade.replace('학년', '').strip()) except: pass income_num = None if income_level and income_level != "모름/해당없음": try: income_num = int(income_level.replace("분위", "")) except: pass gpa_num = None if gpa and gpa != "모름/해당없음": try: gpa_num = float(gpa.split(" ")[0]) except: pass # 신청 공지 필터링 res = get_supabase().table("scholarships").select("*").eq("is_application", True).execute() filtered = [] for s in (res.data or []): target_grade = s.get('target_grade') if isinstance(target_grade, str): try: target_grade = json.loads(target_grade) except: target_grade = [] target_status = s.get('target_status') if isinstance(target_status, str): try: target_status = json.loads(target_status) except: target_status = [] if s.get('end_date_type') == '명시' and s.get('end_date'): if s['end_date'] < today: continue if grade_num and target_grade: if grade_num not in target_grade: continue if target_status and '재학' not in target_status: continue if income_num is not None and s.get('income_max') is not None: if income_num > s['income_max']: continue if gpa_num is not None and s.get('min_gpa') is not None: if gpa_num < s['min_gpa']: continue if s.get('region') is not None and region not in [None, "모름/해당없음"]: if s['region'] != region: continue if s.get('income_required') and loan != "대출 있음": continue filtered.append(s) application_notices = [] if filtered: notice_ids = [s['notice_id'] for s in filtered] notices_res = get_supabase().table("notices").select( "id,notice_id,title,url,posted_at,body,category" ).in_("notice_id", notice_ids).execute() notices_map = {n['notice_id']: n for n in (notices_res.data or [])} for s in filtered: notice = notices_map.get(s['notice_id'], {}) if notice: notice['scholarship_info'] = s application_notices.append(notice) # 관련 공지 최신순 3개 rel_res = get_supabase().table("scholarships").select("notice_id").eq("is_application", False).execute() related_notices = [] if rel_res.data: rel_ids = [r['notice_id'] for r in rel_res.data] rel_notices = get_supabase().table("notices").select( "id,notice_id,title,url,posted_at,body,category" ).in_("notice_id", rel_ids).order("posted_at", desc=True).limit(3).execute() related_notices = rel_notices.data or [] return application_notices, related_notices except Exception as e: print(f"장학금 필터링 오류: {e}") import traceback; traceback.print_exc() return [], [] def filter_dormitory(profile) -> list: try: today = datetime.now().strftime('%Y-%m-%d') gender = profile.get('gender') dorm_interest = profile.get('dorm_interest', []) res = get_supabase().table("dormitories").select("*").execute() filtered = [] for d in (res.data or []): if d.get('end_date_type') == '명시' and d.get('end_date'): if d['end_date'] < today: continue if dorm_interest: if not any(dorm in (d.get('name') or '') for dorm in dorm_interest): continue if gender == '남성' and d.get('male_quota') == 0: continue if gender == '여성' and d.get('female_quota') == 0: continue filtered.append(d) if not filtered: return [] notice_ids = [d['notice_id'] for d in filtered] notices_res = get_supabase().table("notices").select( "id,notice_id,title,url,posted_at,body,category" ).in_("notice_id", notice_ids).execute() notices_map = {n['notice_id']: n for n in (notices_res.data or [])} results = [] for d in filtered: notice = notices_map.get(d['notice_id'], {}) if notice: notice['dormitory_info'] = d results.append(notice) return results except Exception as e: print(f"기숙사 필터링 오류: {e}") import traceback; traceback.print_exc() return [] def filter_rotc() -> list: try: res = get_supabase().table("notices").select( "id,notice_id,title,url,posted_at,body,category" ).eq("category", "ROTC").order("posted_at", desc=True).limit(5).execute() return res.data or [] except Exception as e: print(f"ROTC 필터링 오류: {e}") return [] # ============================================================ # 온보딩 # ============================================================ def render_onboarding(): st.markdown(GLOBAL_CSS, unsafe_allow_html=True) st.markdown("
", unsafe_allow_html=True) col_logo, col_form = st.columns([1, 2], gap="large") with col_logo: st.markdown("
", unsafe_allow_html=True) logo_b64 = get_logo_base64() logo_img = f'' if logo_b64 else '
🔍
' st.markdown(f'
{logo_img}
상상파인더
한성대 공지를
스마트하게 검색하세요.
', unsafe_allow_html=True) with col_form: with st.container(border=True): st.markdown('
반갑습니다 👋
기본 정보를 알려주세요.
', unsafe_allow_html=True) r1c1, r1c2 = st.columns(2) with r1c1: name = st.text_input("이름", placeholder="홍길동", key="ob_name") with r1c2: college = st.selectbox("단과대", list(COLLEGE_MAP.keys()), key="ob_college") r2c1, r2c2 = st.columns(2) with r2c1: track = st.selectbox("트랙 / 학과", COLLEGE_MAP.get(college, ["기타"]), key="ob_track") with r2c2: grade = st.selectbox("학년", ["1학년","2학년","3학년","4학년"], key="ob_grade") interests = st.multiselect( "관심사", ["취업/채용","학사행정","학생활동/비교과","대외활동","공모전/경진대회","국제교류","창업","장학금","기숙사","ROTC"], placeholder="관심 카테고리를 선택하세요", key="ob_interests" ) # ── 장학금 추가 항목 ─────────────────────────────── income_level = None gpa = None region = None loan = None if "장학금" in interests: st.markdown("
📌 장학금 맞춤 필터
", unsafe_allow_html=True) sc1, sc2 = st.columns(2) with sc1: income_level = st.selectbox( "소득분위", ["모름/해당없음","1분위","2분위","3분위","4분위","5분위","6분위","7분위","8분위","9분위","10분위"], key="ob_income" ) with sc2: gpa = st.selectbox( "최근 학점", ["모름/해당없음","4.0 이상","3.5 이상","3.0 이상","2.5 이상","2.0 이상","2.0 미만"], key="ob_gpa" ) sc3, sc4 = st.columns(2) with sc3: region = st.selectbox( "거주 지역", ["서울","경기/인천","지방","해외"], key="ob_region" ) with sc4: loan = st.selectbox( "학자금 대출 여부", ["해당없음","대출 있음"], key="ob_loan" ) # ── 기숙사 추가 항목 ─────────────────────────────── gender = None dorm_interest = None if "기숙사" in interests: st.markdown("
📌 기숙사 맞춤 필터
", unsafe_allow_html=True) dc1, dc2 = st.columns(2) with dc1: gender = st.selectbox( "성별", ["선택안함","남성","여성"], key="ob_gender" ) with dc2: dorm_interest = st.multiselect( "관심 기숙사", ["상상빌리지","우촌학사","동소문행복기숙사","에피소드","임대기숙사"], key="ob_dorm" ) # ── ROTC 추가 항목 ───────────────────────────────── rotc_interest = False if "ROTC" in interests: st.markdown("
📌 ROTC
", unsafe_allow_html=True) rotc_interest = st.checkbox("ROTC 후보생 모집 공지 받기", value=True, key="ob_rotc") # ── 전화번호 (알림용) ────────────────────────────── st.markdown("
📱 공지 알림 수신 전화번호 (선택, 빈칸이면 알림 미사용)
", unsafe_allow_html=True) pc1, pc2, pc3 = st.columns(3) with pc1: phone1 = st.text_input("", placeholder="010", key="ob_phone1", max_chars=3, label_visibility="collapsed") with pc2: phone2 = st.text_input("", placeholder="0000", key="ob_phone2", max_chars=4, label_visibility="collapsed") with pc3: phone3 = st.text_input("", placeholder="0000", key="ob_phone3", max_chars=4, label_visibility="collapsed") phone1 = re.sub(r'\D', '', phone1) phone2 = re.sub(r'\D', '', phone2) phone3 = re.sub(r'\D', '', phone3) phone = f"{phone1}-{phone2}-{phone3}" if phone1 and phone2 and phone3 else None st.markdown("
", unsafe_allow_html=True) if st.button("시작하기 →", use_container_width=True): if not name.strip(): st.markdown('
✏️ 이름을 입력해 주세요.
', unsafe_allow_html=True) else: profile_data = { "name": name.strip(), "college": college, "track": track, "grade": grade, "interests": interests, "income_level": income_level, "gpa": gpa, "region": region, "loan": loan, "gender": gender, "dorm_interest": dorm_interest or [], "rotc_interest": rotc_interest, "phone": phone, } st.session_state.profile = profile_data st.session_state.onboarded = True sync_profile_to_supabase_user(profile_data) st.rerun() # ============================================================ # 사이드바 # ============================================================ def render_sidebar(profile): with st.sidebar: logo_b64 = get_logo_base64() logo_img = f'' if logo_b64 else '🔍' st.markdown(f'
{logo_img}
상상파인더
Hansung Notice Finder
', unsafe_allow_html=True) st.markdown('
내 정보
', unsafe_allow_html=True) info_html = "".join([f'
{k}{v}
' for k, v in [("이름", profile.get("name","")), ("단과대학", profile.get("college","")), ("트랙/학과", profile.get("track","")), ("학년", profile.get("grade",""))]]) st.markdown(info_html, unsafe_allow_html=True) st.markdown("
", unsafe_allow_html=True) st.markdown('
바로가기
', unsafe_allow_html=True) links = [("🏫","한성대학교","https://www.hansung.ac.kr/hansung/index.do"),("💻","한성 e-class","https://learn.hansung.ac.kr/"),("📋","종합정보시스템","https://info.hansung.ac.kr/"),("📊","스마트자기관리시스템","https://hsportal.hansung.ac.kr/"),("📚","학술정보관","https://hsel.hansung.ac.kr/")] track = profile.get("track","") if track in DEPT_URLS: links.append(("🎓", track, DEPT_URLS[track])) link_html = "".join([f'{icon}{label}↗' for icon, label, url in links]) st.markdown(link_html, unsafe_allow_html=True) st.markdown("
", unsafe_allow_html=True) _, btn_col, _ = st.columns([0.3, 2.4, 0.3]) with btn_col: if st.button("내 정보 다시 입력", use_container_width=True): st.session_state.onboarded = False; st.session_state.profile = {}; st.session_state.chat_history = [] st.rerun() # ============================================================ # 챗봇 # ============================================================ def search_rag_notices(query: str, category_filter: str = "전체", top_k: int = 5) -> list[dict]: try: pinecone_categories = ( PINECONE_CATEGORY_MAP.get(category_filter, [category_filter]) if category_filter != "전체" else None ) return pinecone_hybrid_search( query=query, top_k=top_k, category_filter=pinecone_categories, candidate_k=50, feature_rerank=True, profile=st.session_state.get("profile", {}), ) except Exception as e: st.error(f"[Pinecone 검색 오류] {e}") return [] def render_chatbot(profile): with st.container(border=True): st.markdown('
', unsafe_allow_html=True) if not st.session_state.chat_history: name = profile.get("name",""); hsu_b64 = get_hsu_base64() hsu_img = f'' if hsu_b64 else '
🔍
' st.markdown(f'
{hsu_img}
안녕하세요, {name}님!
한성대 공지를 자연어로 검색해보세요.
"장학금 신청 기간 알려줘"  ·  "취업박람회 언제야?"  ·  "비교과 프로그램 추천해줘"
', unsafe_allow_html=True) else: for msg in st.session_state.chat_history: if msg["role"] == "user": st.markdown(f'
{msg["content"]}
', unsafe_allow_html=True) else: st.markdown(f'
{msg["content"]}
', unsafe_allow_html=True) if msg.get("results"): for idx, r in enumerate(msg["results"][:3], start=1): title_safe = html.escape(str(r.get("title", ""))) cat_safe = html.escape(str(r.get("category", "기타"))) date_safe = html.escape(str(r.get("date", ""))) url_safe = html.escape(str(r.get("url", "#")), quote=True) st.markdown( '
' f'참고 공지 Top{idx} ' f'{cat_safe}' f'{date_safe}' f'
{title_safe}
' '
' f'공지 바로가기 →' '
' '
', unsafe_allow_html=True, ) with st.form("chat_form", clear_on_submit=True): c0, c1, c2 = st.columns([1.2, 4, 0.8]) with c0: cat_filter = st.selectbox("카테고리", ["전체"]+CATEGORIES, label_visibility="collapsed", key="chat_cat") with c1: user_input = st.text_input("메시지", placeholder="무엇이 궁금하세요?", label_visibility="collapsed") with c2: submitted = st.form_submit_button("전송", use_container_width=True) if submitted and user_input: st.session_state.chat_history.append({"role": "user", "content": user_input}) cat_filter = st.session_state.get("chat_cat", "전체") results = search_rag_notices(user_input, cat_filter, top_k=5) with st.spinner("답변 생성 중..."): reply = generate_llm_reply(user_input, results, st.session_state.profile, is_first=len(st.session_state.chat_history)==1) st.session_state.chat_history.append({"role": "bot", "content": reply, "results": results}) st.rerun() if st.session_state.chat_history: if st.button("대화 초기화"): st.session_state.chat_history = []; st.rerun() # ============================================================ # 추천 게시물 # ============================================================ def render_notice_card(notice): jts = classify_job_type(notice) job_str = " · ".join([t['job_type'] for t in jts]) if jts else "" summary = summarize_notice(notice.get('title',''), notice.get('body','')) title_safe = html.escape(notice.get('title', '')) cat_safe = html.escape(notice.get('category', '기타')) date_val = re.sub(r'<[^>]+>', '', str(notice.get('date', notice.get('posted_at','')))).strip()[:10] job_html = f'{html.escape(job_str)}' if job_str else "" sum_html = f'
{summary}
' if summary else "" st.markdown( '
' '
' f'{cat_safe}{job_html}' f'{date_val}' '
' f'
{title_safe}
' f'{sum_html}' '
' f'공지 바로가기 →' '
' '
', unsafe_allow_html=True ) def render_recommend(profile): st.markdown( f'
' f'
🎓
' f'
' f'{profile.get("college","")}  ·  {profile.get("track","")}  ·  {profile.get("grade","")}' f'
' f'관심사: {", ".join(profile.get("interests",[])) or "없음"}' f'
', unsafe_allow_html=True ) col_l, col_c, col_r = st.columns([2, 1.5, 2]) with col_c: btn_rec = st.button("맞춤 공지 추천받기", type="primary", use_container_width=True) if btn_rec: scholarship_results = [] scholarship_related = [] dorm_results = [] rotc_results = [] interests = profile.get('interests', []) # ── Two-Tower 추천 (필터링 카테고리 제외) ───────────── FILTER_CATS = ["장학금", "기숙사", "ROTC"] rec_interests = [i for i in interests if i not in FILTER_CATS] with st.spinner("추천 중..."): recs = two_tower_recommend( college = profile.get('college', ''), track = profile.get('track', ''), year = profile.get('grade', ''), interests = rec_interests, top_k = 10, ) if rec_interests else [] # ── 장학금 필터링 ────────────────────────────────────── if "장학금" in interests: with st.spinner("장학금 필터링 중..."): scholarship_results, scholarship_related = filter_scholarships(profile) # ── 기숙사 필터링 ────────────────────────────────────── if "기숙사" in interests: with st.spinner("기숙사 필터링 중..."): dorm_results = filter_dormitory(profile) # ── ROTC 필터링 ──────────────────────────────────────── if "ROTC" in interests and profile.get('rotc_interest', False): with st.spinner("ROTC 공지 로딩 중..."): rotc_results = filter_rotc() total = len(recs) + len(scholarship_results) + len(scholarship_related) + len(dorm_results) + len(rotc_results) if total == 0: st.info("추천 결과가 없습니다.") return # ── 맞춤 추천 출력 ───────────────────────────────────── if recs: st.markdown(f"
🎯 맞춤 추천 공지 ({len(recs)}개)
", unsafe_allow_html=True) for rec in recs: render_notice_card(rec['notice']) # ── 장학금 신청 공지 ─────────────────────────────────── if scholarship_results: st.markdown(f"
💰 신청 가능한 장학금 ({len(scholarship_results)}개)
", unsafe_allow_html=True) for notice in scholarship_results: render_notice_card(notice) # ── 장학금 관련 공지 ─────────────────────────────────── if scholarship_related: st.markdown(f"
📢 장학금 관련 공지 ({len(scholarship_related)}개)
", unsafe_allow_html=True) for notice in scholarship_related: render_notice_card(notice) # ── 기숙사 출력 ──────────────────────────────────────── if dorm_results: st.markdown(f"
🏠 기숙사 ({len(dorm_results)}개)
", unsafe_allow_html=True) for notice in dorm_results: render_notice_card(notice) # ── ROTC 출력 ────────────────────────────────────────── if rotc_results: st.markdown(f"
🎖️ ROTC ({len(rotc_results)}개)
", unsafe_allow_html=True) for notice in rotc_results: render_notice_card(notice) # ============================================================ # 엔트리포인트 # ============================================================ def main(): from PIL import Image try: hsu_icon = Image.open(os.path.join(_BASE_DIR, "hsu.png")) except: hsu_icon = "🔍" st.set_page_config(page_title="상상파인더", page_icon=hsu_icon, layout="wide", initial_sidebar_state="expanded") if "chat_history" not in st.session_state: st.session_state.chat_history = [] if "onboarded" not in st.session_state: st.session_state.profile = {} st.session_state.onboarded = False if not st.session_state.onboarded: render_onboarding(); return st.markdown(GLOBAL_CSS, unsafe_allow_html=True) profile = st.session_state.profile render_sidebar(profile) tab_chat, tab_rec = st.tabs([" 💬 챗봇 검색 ", " ✨ 추천 게시물 "]) with tab_chat: render_chatbot(profile) with tab_rec: render_recommend(profile) if __name__ == "__main__": main()