import streamlit as st

from api import fetch_okrs
from config import SYSTEM_PROMPT, INPUT_TEMPLATE, PROMPT_TEMPLATE
from utils import construct_prompt, extract_llm_response

def proposer_page():
    # Streamlit App Layout
    st.title("OKR Generator")

    # Input Section and Buttons Row
    st.subheader("Enter your idea or goal:")
    user_input = st.text_area(
        "Input your idea here:",
        value=st.session_state.get("user_input", INPUT_TEMPLATE.strip()),
        height=200,
    )

    col1, col2 = st.columns([1, 1])
    with col1:
        if st.button("Reset All"):
            st.session_state.clear()
    with col2:
        generate_okrs_clicked = st.button("Generate OKRs")
        
    if generate_okrs_clicked:
        if not user_input.strip():
            st.warning("Please provide some input before generating OKRs.")
        else:
            with st.spinner("Generating OKRs..."):
                # Construct prompt and call API
                #prompt = construct_prompt(prompt_template=PROMPT_TEMPLATE, user_input=user_input)
                response = fetch_okrs(user_input=user_input)
                if response:
                    # Extract Objective and Key Results from response
                    print(response)
                    objective, key_results, hint = extract_llm_response(response)
                    
                    st.session_state["objective"] = objective
                    st.session_state["key_results"] = key_results
                    st.session_state["hint"] = hint

    #st.subheader("Hint to improve the OKR proposal")
    #st.text(hint)

    # Display Results Only if an OKR Has Been Generated
    if "objective" in st.session_state and "key_results" in st.session_state:
            
        # Display Objective Field with Responsibles Input Below It
        st.subheader("Proposal Objective:")
        objective_text = st.text_area(
            "Proposal Objective:",
            value=st.session_state.get("objective", ""),
            height=100,
        )
        
        responsible_for_objective = st.text_input(
            "Responsibles for Objective (comma-separated):",
            value="",
            placeholder="e.g., Khanh Dinh, John Doe",
            key="responsibles_objective"
        )

        # Display Key Results with Responsibles Below Each One
        st.subheader("Proposal Key Results:")
        key_result_boxes = []
        responsibles_for_key_results = []
        
        for i, kr in enumerate(st.session_state["key_results"], start=1):
            kr_text = st.text_area(f"Key Result {i}:", value=kr, key=f"kr_{i}")
            responsible_for_kr = st.text_input(
                f"Responsibles for Key Result {i} (comma-separated):",
                value="",
                placeholder="e.g., Khanh Dinh",
                key=f"responsibles_kr_{i}"
            )
            
            key_result_boxes.append(kr_text)
            responsibles_for_key_results.append(responsible_for_kr)

        # Finalize Button Center-Aligned
        #finalize_col = st.columns([3, 2, 3])[1]
        #with finalize_col:
        if st.button("Finalize"):
            finalized_objective = objective_text.strip()
            finalized_key_results = [st.session_state[f"kr_{i+1}"].strip() for i in range(len(key_result_boxes))]

            # Append initials of responsibles to Objective and Key Results
            responsibles_list_objective = [name.strip() for name in responsible_for_objective.split(",") if name.strip()]
            initials_objective = [f"[{''.join([part[0] for part in name.split()]).upper()}]" for name in responsibles_list_objective]
            initials_str_objective = ", ".join(initials_objective)

            finalized_objective = f"{initials_str_objective} {finalized_objective}" 

            finalized_key_results_with_initials = []
            for i, kr in enumerate(finalized_key_results):
                responsibles_list_kr = [name.strip() for name in responsibles_for_key_results[i].split(",") if name.strip()]
                initials_kr = [f"{''.join([part[0] for part in name.split()]).upper()}" for name in responsibles_list_kr]
                initials_str_kr = ", ".join(initials_kr)
                finalized_key_results_with_initials.append(f"KR{i+1}: [{initials_str_kr}] {kr}")

            # Display finalized data in non-editable format (full width)
            st.subheader("Finalized Objective:")
            st.code(
                body=finalized_objective,
                language=None,
                wrap_lines=True
                )

            st.subheader("Finalized Key Results:")
            for kr in finalized_key_results_with_initials:
                st.code(
                    body=kr,
                    language=None,
                    wrap_lines=True
                )