# cycles_chat_app.py
import os, math, numpy as np, matplotlib.pyplot as plt, gradio as gr
import openai

### 0. OpenAI 키 설정 ---------------------------------------------------------
if "OPENAI_API_KEY" not in os.environ:
    os.environ["OPENAI_API_KEY"] = input("🔑 Enter your OpenAI API key: ").strip()

openai.api_key = os.environ["OPENAI_API_KEY"]

### 1. 주기 차트 함수 ----------------------------------------------------------
CYCLES = {
    "테크 사이클 (50 년)": 50,
    "재무 사이클 (80 년)": 80,
    "패권 사이클 (250 년)": 250,
}
CENTER = 2025  # 기준점

def make_cycle_points(period, start, end):
    n_min = math.floor((start - CENTER) / period)
    n_max = math.ceil((end   - CENTER) / period)
    return [CENTER + n * period for n in range(n_min, n_max + 1)]

def plot_cycles(range_start, range_end):
    plt.figure(figsize=(10, 4))
    y_positions = np.arange(len(CYCLES))
    align_years = None

    for i, (name, period) in enumerate(CYCLES.items()):
        years = make_cycle_points(period, range_start, range_end)
        plt.scatter(years, [y_positions[i]] * len(years), label=name, s=60)
        for y in years:
            plt.text(y, y_positions[i] + 0.1, str(y), ha="center", fontsize=7)
        align_years = set(years) if align_years is None else align_years & set(years)

    if align_years:
        for y in sorted(align_years):
            plt.axvline(y, color="black", linestyle="--")
            plt.text(y, y_positions[-1] + 0.35, f"공동 정렬 → {y}",
                     ha="center", va="bottom", fontsize=9, fontweight="bold")

    plt.yticks(y_positions, list(CYCLES.keys()))
    plt.ylim(-0.5, len(CYCLES) - 0.2)
    plt.grid(axis="x", alpha=0.3)
    plt.xlabel("Year")
    plt.title("주기별 발생 연도와 공동 정렬 지점")
    plt.tight_layout()
    return plt.gcf()

### 2. GPT 채팅 함수 -----------------------------------------------------------
SYSTEM_PROMPT = (
    "당신은 친절하고 분석적인 어시스턴트입니다. "
    "사용자의 질문에 한국어로 간결하고 정확하게 답변하세요."
)

def chat_with_gpt(history, user_message):
    messages = [{"role": "system", "content": SYSTEM_PROMPT}]
    for u, a in history:
        messages.append({"role": "user", "content": u})
        messages.append({"role": "assistant", "content": a})
    messages.append({"role": "user", "content": user_message})

    response = openai.chat.completions.create(
        model="gpt-3.5-turbo",
        messages=messages,
        max_tokens=600,
        temperature=0.7,
    )
    assistant_reply = response.choices[0].message.content.strip()
    return assistant_reply

### 3. Gradio 인터페이스 -------------------------------------------------------
with gr.Blocks(theme=gr.themes.Soft()) as demo:
    gr.Markdown("## 📊 3-Cycle Timeline & 💬 GPT Chat")

    with gr.Tabs():
        ## Tab 1 — Chart
        with gr.TabItem("🗓️ 주기 차트"):
            start_year = gr.Number(label="범위 시작(Year)", value=1500)
            end_year   = gr.Number(label="범위 끝(Year)",   value=2500)
            plot_out   = gr.Plot()

            def update_chart(s, e):
                return plot_cycles(int(s), int(e))

            start_year.change(update_chart, [start_year, end_year], plot_out)
            end_year.change(update_chart,   [start_year, end_year], plot_out)
            plot_out.render()  # 초기 표시

        ## Tab 2 — GPT Chat
        with gr.TabItem("💬 GPT 채팅"):
            chatbot = gr.Chatbot(label="GPT Assistant")
            user_in = gr.Textbox(placeholder="메시지를 입력하세요...", lines=3)
            send_btn = gr.Button("Send", variant="primary")

            def respond(chat_history, user_msg):
                assistant_msg = chat_with_gpt(chat_history, user_msg)
                chat_history.append((user_msg, assistant_msg))
                return chat_history, gr.Textbox(value="", interactive=True)

            send_btn.click(
                respond,
                inputs=[chatbot, user_in],
                outputs=[chatbot, user_in],
            )
            user_in.submit(
                respond,
                inputs=[chatbot, user_in],
                outputs=[chatbot, user_in],
            )

if __name__ == "__main__":
    demo.launch()