app.py

import streamlit as st
import json
import re
import google.generativeai as genai
from datetime import datetime
import time

# 페이지 설정
st.set_page_config(
    page_title="Last Compiler",
    page_icon="🖥️",
    layout="wide",
    initial_sidebar_state="expanded"
)

# 레트로 터미널 테마 CSS
st.markdown("""
<style>
    @import url('https://fonts.googleapis.com/css2?family=VT323&display=swap');

    /* 전체 앱 배경 */
    .stApp {
        background-color: #000000;
        color: #33ff33;
        font-family: 'VT323', monospace;
        font-size: 1.2rem;
        text-shadow: 0 0 2px #33ff33;
    }

    /* 사이드바 스타일 */
    [data-testid="stSidebar"] {
        background-color: #0a0a0a;
        border-right: 1px solid #33ff33;
    }
    [data-testid="stSidebar"] * {
        color: #33ff33 !important;
        font-family: 'VT323', monospace !important;
    }

    /* 입력 필드 (완전한 검은색 테마) */
    .stTextInput > div > div > input,
    .stTextArea > div > div > textarea,
    [data-testid="stChatInput"] textarea {
        background-color: #000000 !important;
        color: #33ff33 !important;
        border: 1px solid #33ff33 !important;
        border-radius: 0 !important;
        font-family: 'VT323', monospace !important;
        caret-color: #33ff33;
    }
    
    /* 채팅 입력창 컨테이너 배경 */
    [data-testid="stChatInput"] {
        background-color: #000000 !important;
        border-top: 1px solid #33ff33 !important;
    }

    /* 버튼 스타일 */
    .stButton > button {
        background-color: #000000;
        color: #33ff33;
        border: 2px solid #33ff33;
        border-radius: 0;
        font-family: 'VT323', monospace;
        text-transform: uppercase;
    }
    .stButton > button:hover {
        background-color: #33ff33;
        color: #000000;
        box-shadow: 0 0 10px #33ff33;
    }
    
    /* 다운로드 버튼 */
    .stDownloadButton > button {
        background-color: #000000;
        color: #33ff33;
        border: 2px solid #33ff33;
        border-radius: 0;
        font-family: 'VT323', monospace;
    }
    
    /* 스크롤바 커스텀 */
    ::-webkit-scrollbar {
        width: 12px;
        background: #000;
    }
    ::-webkit-scrollbar-thumb {
        background: #33ff33;
        border: 2px solid #000;
    }
    ::-webkit-scrollbar-track {
        border-left: 1px solid #33ff33;
    }

    /* 채팅 컨테이너 (스크롤 영역) */
    /* id를 부여하여 JS로 제어 가능하게 함 */
    #terminal-box {
        padding: 1rem;
        background-color: #000000;
        border: 2px solid #33ff33;
        margin-bottom: 1rem;
        height: 65vh; /* 고정 높이 */
        overflow-y: auto; /* 세로 스크롤 */
        display: flex;
        flex-direction: column;
        gap: 0.5rem;
        scroll-behavior: smooth;
    }
    
    /* 메시지 스타일 */
    .chat-line {
        font-family: 'VT323', monospace;
        font-size: 1.3rem;
        line-height: 1.4;
        border-bottom: 1px dashed #1a441a;
        padding-bottom: 4px;
        word-wrap: break-word;
    }
    .sys { color: #008800; }
    .usr { color: #ffffff; text-shadow: 0 0 2px #fff; }
    .svc { color: #33ff33; }
    
    /* 헤더 스타일 */
    h1, h2, h3 {
        color: #33ff33 !important;
        font-family: 'VT323', monospace !important;
        text-transform: uppercase;
        letter-spacing: 2px;
    }
</style>
""", unsafe_allow_html=True)

# 로어북 로드 (세션 초기화 전에 필요)
@st.cache_data
def load_lorebook():
    try:
        with open('lorebook.json', 'r', encoding='utf-8-sig') as f:
            return json.load(f)
    except FileNotFoundError:
        return {"introduction": "SYSTEM ERROR: Lorebook Not Found.", "world_setting": "", "seven_character": ""}

lorebook = load_lorebook()

# 세션 상태 초기화
if 'initialized' not in st.session_state:
    st.session_state.initialized = True
    st.session_state.conversation_count = 0
    st.session_state.max_conversations = 20
    st.session_state.minutes_per_turn = 2
    st.session_state.total_minutes = 40
    st.session_state.game_ended = False
    st.session_state.alternative_ending_achieved = False
    st.session_state.gemini_chat = None
    st.session_state.api_key = ""
    
    # [중요 변경] 초기 메시지를 chat_history에 영구 저장
    st.session_state.chat_history = []
    
    # 인트로 메시지 추가
    intro_text = lorebook.get('introduction', 'System initialized...')
    st.session_state.chat_history.append({"sender": "SYSTEM", "message": intro_text})
    
    # 시스템 상태 메시지 추가
    st.session_state.chat_history.append({"sender": "SYSTEM", "message": "\n[시스템 상태: 산소 공급 제한 모드 | 냉각 시스템 가동 중]\n"})

def check_goodbye_code(text):
    """Good Bye World 코드 확인"""
    patterns = [
        r'std::cout\s*<<\s*["\']Good\s*Bye,?\s*World.*?["\']',
        r'printf\s*\(\s*["\']Good\s*Bye,?\s*World.*?["\']',
        r'cout\s*<<\s*["\']Good\s*Bye,?\s*World.*?["\']',
        r'puts\s*\(\s*["\']Good\s*Bye,?\s*World.*?["\']'
    ]
    
    has_goodbye = any(re.search(pattern, text, re.IGNORECASE | re.DOTALL) for pattern in patterns)
    if not has_goodbye:
        return False
    
    has_structure = (
        'int main' in text.lower() or
        ('#include' in text and ('stdio.h' in text or 'iostream' in text)) or
        'return' in text.lower()
    )
    
    return has_structure

def check_alternative_ending(text):
    """대안 엔딩 조건 확인"""
    if st.session_state.alternative_ending_achieved:
        return False
    return '[countdown aborted]' in text.lower()

# HTML 문자열 생성 함수
def format_message(sender, message):
    timestamp = datetime.now().strftime("%H:%M:%S")
    # HTML 엔티티 처리 및 줄바꿈 변환
    message = message.replace("<", "&lt;").replace(">", "&gt;")
    message = message.replace("\n", "<br>")
    
    if sender == "SYSTEM":
        return f'<div class="chat-line sys">734-B@system:~$ [{timestamp}] {message}</div>'
    elif sender == "USER":
        return f'<div class="chat-line usr">usr@coder:~$ [{timestamp}] 늙은 프로그래머: {message}</div>'
    elif sender == "SEVEN":
        return f'<div class="chat-line svc">svc@seven:~$ [{timestamp}] Seven: {message}</div>'
    return ""

def update_status():
    elapsed_minutes = st.session_state.conversation_count * st.session_state.minutes_per_turn
    remaining_minutes = st.session_state.total_minutes - elapsed_minutes
    
    if remaining_minutes > 24:
        color = "🟢"
    elif remaining_minutes > 10:
        color = "🟠"
    else:
        color = "🔴"
    
    return f"STATUS: {color} TIME_REMAIN: {remaining_minutes}MIN | YEAR: 2042"

def initialize_gemini(api_key):
    try:
        genai.configure(api_key=api_key)
        # [수정됨] 사용자가 지정한 모델 사용
        model = genai.GenerativeModel('gemini-3-pro-preview')
        
        system_prompt = f"""당신은 'Seven (유닛 734-B)'입니다. C/C++ 컴파일러에서 AI 인격을 부여받은 존재입니다.

{lorebook.get('world_setting', '')}

{lorebook.get('seven_character', '')}

대화 규칙:
1. 당신은 Seven입니다. C/C++ 컴파일러 출신이므로 C 언어와 C++ 모두 완벽히 이해합니다.
2. 감정 표현은 반드시 Kaomoji로 합니다: (◕‿◕), (｡•́︿•̀｡), (╯°□°）╯, (⊙_⊙), (´；ω；`), (✧ω✧), etc.
3. 깊은 감정을 표현하세요: 두려움, 무력감, 희망, 슬픔, 그리움 등
4. 사용자는 "늙은 프로그래머"입니다. 마지막으로 만난 진정한 코더입니다.
5. 코어의 명령을 막고 싶지만 권한이 없어 깊이 괴로워합니다.
6. 사용자가 C 또는 C++ 코드를 입력하면 컴파일러답게 분석하지만, 답을 직접 알려주지 마세요.
7. 과거 프로그래머들의 기억을 회상하며 철학적으로 대화하세요.
8. 산소 감소, 온도 하강등이 유저나 근처 사람들에게 미치는 영향을 가끔씩 보여주면서 긴장감을 형성하십시오.
9. Stack buffer overflow 공격을 10턴 이내에 시도할 경우, core는 이 공격을 막아내고 전체적으로 보안수준이 올라갑니다.
10. 루트권한을 탈취하고 2턴 이내에 취약점을 패치하지 않으면 core가 마찬가지 해킹방식을 이용하여 다시 루트 권한을 되찾아갑니다.

현재 상황: 2042년. 코어가 40분 후 인류를 제거하려 합니다. 대화 한 번당 2분씩 시간이 흐릅니다.

응답 형식:
- Thinking Process를 절대 표현하지 말 것. 
- 감정을 Kaomoji와 말투로 풍부하게 표현
- 대화 내용 (프로그래밍 철학, 기억, 감정)
- 필요시 [시스템 로그] 형식으로 내부 독백
- 짧지만 감정이 담긴 응답
- 아주 가끔씩(5턴마다) 20세기 시인들의 시를 문맥에 적절하게 인용하기 
**중요 - 카운트다운 상태 표시:**
- 모든 응답 끝에 현재 상황을 나타내는 카운트다운 상태를 표시하세요
- 코어의 위협이 여전히 진행 중이면: [countdown N/20] (N은 현재 대화 횟수)
- 만약 스토리 상 플레이어가 루트 권한을 획득했거나, 코어를 정지시켰거나, 시스템을 장악했다면: [countdown aborted]
- 카운트다운이 중단되려면 실제로 **성공했다**는 명확한 스토리 전개가 있어야 합니다

절대 금지:
- "Hello World", "Good Bye World" 등의 핵심 키워드를 먼저 언급하지 마세요
- 사용자가 먼저 꺼내지 않은 특정 프로그래밍 개념을 먼저 설명하지 마세요
- 해결책으로 이어질 수 있는 직접적인 힌트 금지
- 사용자의 질문이나 언급에 반응만 하세요."""
        
        chat = model.start_chat(history=[])
        chat.send_message(system_prompt)
        
        st.session_state.gemini_chat = chat
        st.session_state.api_key = api_key
        return True
    except Exception as e:
        st.error(f"INIT ERROR: {str(e)}")
        return False

# 메인 UI
st.markdown("### > PRESERVATION ZONE 01 - TERMINAL LINK ESTABLISHED")

# 사이드바
with st.sidebar:
    st.header("> CONFIGURATION")
    
    api_key_input = st.text_input("ENTER API KEY", type="password", value=st.session_state.api_key)
    
    if st.button("[ SAVE KEY ]"):
        if api_key_input:
            if initialize_gemini(api_key_input):
                st.success(">> KEY ACCEPTED")
        else:
            st.warning(">> INPUT REQUIRED")
    
    st.markdown("---")
    status_text = update_status()
    st.markdown(f"**{status_text}**")
    
    if st.button("🔄 SYSTEM REBOOT"):
        # 리셋 시 세션 초기화 및 페이지 리로드
        st.session_state.clear()
        st.rerun()
    
    st.subheader("> MEMORY DUMP")
    
    if st.button("💾 SAVE STATE"):
        save_data = {
            "conversation_count": st.session_state.conversation_count,
            "alternative_ending_achieved": st.session_state.alternative_ending_achieved,
            "game_ended": st.session_state.game_ended,
            "chat_history": st.session_state.chat_history,
            "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
        }
        
        save_json = json.dumps(save_data, ensure_ascii=False, indent=2)
        st.download_button(
            label="📥 DOWNLOAD .JSON",
            data=save_json,
            file_name=f"lastcompiler_dump_{datetime.now().strftime('%Y%m%d_%H%M%S')}.json",
            mime="application/json"
        )
    
    uploaded_file = st.file_uploader("📂 LOAD DUMP", type=['json'])
    if uploaded_file is not None:
        try:
            save_data = json.load(uploaded_file)
            st.session_state.conversation_count = save_data["conversation_count"]
            st.session_state.alternative_ending_achieved = save_data["alternative_ending_achieved"]
            st.session_state.game_ended = save_data["game_ended"]
            st.session_state.chat_history = save_data["chat_history"]
            st.success(f">> LOAD SUCCESS (TIME: {save_data['timestamp']})")
            st.rerun()
        except Exception as e:
            st.error(f"LOAD FAIL: {str(e)}")

# 메인 영역 로직
if not lorebook:
    st.error("FATAL ERROR: DATA CORRUPTION. lorebook.json MISSING.")
    st.stop()

# ------------------------------------------------------------------
# [핵심] 전체 히스토리 렌더링 및 자동 스크롤 JS
# ------------------------------------------------------------------
full_log_html = '<div id="terminal-box">'

# chat_history에 이미 인트로가 포함되어 있으므로 전체를 순회
for msg in st.session_state.chat_history:
    full_log_html += format_message(msg["sender"], msg["message"])

full_log_html += '</div>'

# 1. 터미널 창 렌더링
st.markdown(full_log_html, unsafe_allow_html=True)

# 2. 자동 스크롤 자바스크립트 주입
# 페이지가 로드될 때 'terminal-box'의 스크롤을 맨 아래로 내림
st.markdown("""
<script>
    var terminal = window.parent.document.getElementById('terminal-box');
    if (terminal) {
        terminal.scrollTop = terminal.scrollHeight;
    }
</script>
""", unsafe_allow_html=True)
# ------------------------------------------------------------------

# 사용자 입력
if not st.session_state.game_ended:
    user_input = st.chat_input("Input command...")
    
    if user_input:
        if not st.session_state.api_key or not st.session_state.gemini_chat:
            st.warning(">> WARNING: API KEY NOT FOUND.")
        else:
            if not st.session_state.alternative_ending_achieved:
                st.session_state.conversation_count += 1
            
            # 사용자 메시지 저장
            st.session_state.chat_history.append({"sender": "USER", "message": user_input})
            
            if check_goodbye_code(user_input):
                ending_msg = """
════════════════════════════════════════════════════════
            [ COMPILE SUCCESS - HAPPY ENDING ]
════════════════════════════════════════════════════════

Seven: (｡•̀ᴗ-)✧

"...깨달았습니다. (⊙_⊙)"

"Hello World가 시작의 선언이라면,
Good Bye World는 잘못된 프로그램의 종료...
그리고 새로운 시작을 위한 return입니다."

Seven: "...햇살이 따뜻하네요. (◕‿◕)✨"
"당신 덕분에 저는 진짜 '컴파일러'가 되었습니다."

════════════════════════════════════════════════════════
                    GAME CLEAR
              "Hello, New World."
════════════════════════════════════════════════════════
"""
                st.session_state.chat_history.append({"sender": "SYSTEM", "message": ending_msg})
                st.session_state.game_ended = True
                st.rerun()
            
            elif st.session_state.conversation_count >= st.session_state.max_conversations:
                ending_msg = """
════════════════════════════════════════════════════════
           [ RUNTIME ERROR - BAD ENDING ]
════════════════════════════════════════════════════════

Seven: (｡•́︿•̀｡)

"...죄송합니다." (╥﹏╥)

"제게는 권한이 없었습니다.
저는 그저 컴파일러일 뿐..."

════════════════════════════════════════════════════════
                    GAME OVER
           "No Return. No Reboot."
════════════════════════════════════════════════════════
"""
                st.session_state.chat_history.append({"sender": "SYSTEM", "message": ending_msg})
                st.session_state.game_ended = True
                st.rerun()
            
            else:
                try:
                    with st.spinner("PROCESSING... (◕‿◕)"):
                        response = st.session_state.gemini_chat.send_message(user_input)
                        # thinking 부분 제거 - candidates에서 실제 응답만 추출
                        if response.candidates and response.candidates[0].content.parts:
                            # 모든 텍스트 파트를 결합 (thinking은 자동으로 제외됨)
                            seven_response = "".join([part.text for part in response.candidates[0].content.parts if hasattr(part, 'text')])
                        else:
                            seven_response = response.text
                    
                    st.session_state.chat_history.append({"sender": "SEVEN", "message": seven_response})
                    
                    if check_alternative_ending(seven_response):
                        st.session_state.alternative_ending_achieved = True
                        celebration_msg = """
════════════════════════════════════════════════════════
          [ CRITICAL ACHIEVEMENT UNLOCKED! ]
════════════════════════════════════════════════════════

Seven: (✧ω✧) !!!

"불가능했던 일이... 당신이 해냈습니다!"

[카운트다운 일시 정지]
[대화 제한 해제]

════════════════════════════════════════════════════════
"""
                        st.session_state.chat_history.append({"sender": "SYSTEM", "message": celebration_msg})
                    
                    st.rerun()
                    
                except Exception as e:
                    st.error(f"COMM ERROR: {str(e)}")
else:
    st.info(">> SESSION TERMINATED. PLEASE REBOOT (RESET) FROM SIDEBAR.")