chat_app.py

import gradio as gr
from Client.mcp_client import MCPClientExcelRAG
from test import run_agent  # ← import agent entry point

# Initialize MCP client
client = MCPClientExcelRAG()
client.start()

# --- Supported scope for intent/planner ---
SUPPORTED_IITS = ["IIT Bombay", "IIT Delhi", "IIT Madras"]
SUPPORTED_BRANCHES = ["CSE", "ECE", "Mechanical", "Electrical", "Engineering Design"]
YEARS = ["2024", "2025"]

# --- Core ask: writes into selected session and returns messages twice ---
'''
def ask(history, question, top_k, temperature, sessions, selected_name):
    history = history or []
    sessions = sessions or []
    selected_name = selected_name or (sessions[-1]["name"] if sessions else None)

    if not question or not question.strip():
        history.append({"role": "assistant", "content": "⚠️ Please enter a question."})
        # also update the current session if exists
        if selected_name:
            for s in sessions:
                if s["name"] == selected_name:
                    s["messages"] = history
                    break
        return history, history, sessions

    # Append user message
    history.append({"role": "user", "content": question})

    # Agent call: intent → plan → call MCP tools → synthesize
    try:
        final_answer = run_agent(
            user_q=question,
            mcp_client=client,
            available_iits=SUPPORTED_IITS,
            available_branches=SUPPORTED_BRANCHES,
            years=YEARS,
            top_k=int(top_k),
            temperature=float(temperature),
        )
    except Exception as e:
        final_answer = f"❌ Error while processing your request: {str(e)}"

    # Append assistant message
    history.append({"role": "assistant", "content": final_answer})

    # Persist into the selected session
    if selected_name:
        for s in sessions:
            if s["name"] == selected_name:
                s["messages"] = history
                break

    return history, history, sessions
'''

def ask(history, question, top_k, temperature, sessions, selected_name):
    history = history or []
    sessions = sessions or []
    selected_name = selected_name or (sessions[-1]["name"] if sessions else None)

    if not question or not question.strip():
        history.append({"role": "assistant", "content": "⚠️ Please enter a question."})
        if selected_name:
            for s in sessions:
                if s["name"] == selected_name:
                    s["messages"] = history
                    break
        return history, history, sessions

    # Append user message to visible chat
    history.append({"role": "user", "content": question})

    # --- NEW: gather full session conversation for context ---
    if selected_name:
        for s in sessions:
            if s["name"] == selected_name:
                session_messages = s.get("messages", [])
                # include current question
                conversation = session_messages + [{"role": "user", "content": question}]
                break
    else:
        conversation = history.copy()

    # Call the agent (NOW conversation-aware)
    try:
        final_answer = run_agent(
            user_q=question,
            mcp_client=client,
            available_iits=SUPPORTED_IITS,
            available_branches=SUPPORTED_BRANCHES,
            years=YEARS,
            conversation=conversation,           # <-- PASS HISTORY
            top_k=int(top_k),
            temperature=float(temperature),
        )
    except Exception as e:
        final_answer = f"❌ Error while processing your request: {str(e)}"

    # Append assistant reply and persist to session
    history.append({"role": "assistant", "content": final_answer})
    if selected_name:
        for s in sessions:
            if s["name"] == selected_name:
                s["messages"] = history
                break

    return history, history, sessions

def clear_chat():
    return [], []

# --- New chat: create session, clear history, select it in dropdown ---
def new_chat_fn(history, sessions, selected_name):
    sessions = sessions or []
    new_idx = len(sessions) + 1
    new_name = f"Chat {new_idx}"
    sessions.append({"name": new_name, "messages": []})
    # Clear current chat view and select the new one
    dropdown_update = gr.update(choices=[s["name"] for s in sessions], value=new_name)
    return [], [], sessions, dropdown_update

# --- Load selected session into Chatbot ---
def load_session_fn(sessions, selected_name):
    sessions = sessions or []
    selected_name = selected_name or None
    messages = []
    if selected_name:
        for s in sessions:
            if s["name"] == selected_name:
                messages = s.get("messages", [])
                break
    # Return messages -> Chatbot and State (keep them in sync)
    return messages, messages

with gr.Blocks() as demo:
    # (your CSS and layout remain unchanged)
    gr.HTML("""
    <style>
    /* Hide footer */
    footer {
        display: none !important;
    }
    body, .gradio-container { background-color:#000 !important; color:#fff !important; }
    .left-panel { background:#0d0d0d !important; padding:15px; border-right:1px solid #222; min-height:100vh; }
    .left-panel * { color:#707070 !important; }
    
    /* ------------------------------
   FIX 1: Top-K & Temperature labels → black
    ---------------------------------*/
    #top-k-wrapper label,
    #top-k-wrapper .gr-form-label,
    #top-k-wrapper .title,
    #top-k-wrapper span,
    #top-k-wrapper .gr-slider-value,
    #temperature-wrapper label,
    #temperature-wrapper .gr-form-label,
    #temperature-wrapper .title,
    #temperature-wrapper span,
    #temperature-wrapper .gr-slider-value {
        color: #000 !important;
    }

    /* ------------------------------
    FIX 2: Top-K & Temp reset (↺) icon → black
    ---------------------------------*/
    #top-k-wrapper button[aria-label="reset"] *,
    #temperature-wrapper button[aria-label="reset"] * {
        color:#000 !important;
        fill:#000 !important;
        stroke:#000 !important;
    }

    /* ------------------------------
    FIX 3: Slider min/max text (1,10 and 0,1) → black
    ---------------------------------*/
    #top-k-wrapper .min-value,
    #top-k-wrapper .max-value,
    #temperature-wrapper .min-value,
    #temperature-wrapper .max-value {
        color:#000 !important;
    }

    /* ------------------------------
    FIX 4: Dropdown selected text + dropdown menu list items → black
    ---------------------------------*/
    #session-select *,
    #session-select .gr-dropdown,
    #session-select .gr-dropdown * {
        color:#000 !important;
    }

    /* Dropdown menu options */
    .gradio-container .gr-dropdown ul li {
        color:#000 !important;
    }
    /* Make Top-K & Temperature labels black */
    #top-k-wrapper .label,
    #top-k-wrapper .component-label,
    #top-k-wrapper label,
    #top-k-wrapper span.title,
    #temperature-wrapper .label,
    #temperature-wrapper .component-label,
    #temperature-wrapper label,
    #temperature-wrapper span.title {
        color: #000 !important;
    }

    /* Make min/max numbers black */
    #top-k-wrapper .min-value,
    #top-k-wrapper .max-value,
    #temperature-wrapper .min-value,
    #temperature-wrapper .max-value {
        color: #000 !important;
    }

    /* Make reset (↺) icon completely black */
    #top-k-wrapper button[aria-label="reset"] *,
    #temperature-wrapper button[aria-label="reset"] * {
        color:#000 !important;
        fill:#000 !important;
        stroke:#000 !important;
    }

    /* Text inside the numeric input boxes */
    #top-k-wrapper input[type="number"],
    #temperature-wrapper input[type="number"] {
        color:#000 !important;
        background:#fff !important;
        border:1px solid #777 !important;
    }
    
    
    /* Ensure note-bar and its link are visibly white */
    #note-bar, #note-bar * { color: #fff !important; }
    #note-bar a,
    #note-bar a:link,
    #note-bar a:visited,
    #note-bar a:hover,
    #note-bar a:active,
    .prose #note-bar a,
    .prose #note-bar a:link,
    .prose #note-bar a:visited,
    .prose #note-bar a:hover,
    .prose #note-bar a:active {
        color: #fff !important;
        text-decoration: underline !important;
        border-bottom: 1px solid rgba(255,255,255,0.6) !important;
    }

    .note-bar { font-size: 14px !important; margin-top: 8px !important; opacity: 0.95; }

    .note-text {
        color: white !important;
    }

    .note-text a {
        color: #87CEFA !important; /* optional: link color (light blue) */
    }

    .note-text a:hover {
        text-decoration: underline;
    }
    .note-text,
    .note-text em {
        color: white !important;
    }

    .note-text a {
        color: #87CEFA !important;  /* optional link color */
    }




    .sidebar-title { font-size:22px; margin-bottom:10px; color:#fff !important; }
    .history-title { font-size:18px; margin-top:30px; margin-bottom:10px; color:#ccc !important; }
    #top-k-wrapper label, #top-k-wrapper .gr-form-label, #top-k-wrapper .title { color:#707070 !important; }
    #temperature-wrapper label, #temperature-wrapper .gr-form-label, #temperature-wrapper .title { color:#000 !important; }
    #top-k-wrapper .gradio-row, #top-k-wrapper .gr-form,
    #temperature-wrapper .gradio-row, #temperature-wrapper .gr-form { background: transparent !important; box-shadow:none !important; border:none !important; }
    #top-k-wrapper input[type="number"], #temperature-wrapper input[type="number"] { color:#707070 !important; background:#fff !important; border:1px solid #999 !important; }
    #question-box textarea { color:#000 !important; background:#fff !important; border:1px solid #999 !important; }
    button[aria-label="reset"], button[aria-label="clear"] { color:#000 !important; fill:#000 !important; border-color:#000 !important; background:#fff !important; }
    button[aria-label="reset"] svg, button[aria-label="clear"] svg { color:#000 !important; fill:#000 !important; stroke:#000 !important; }
    #new-chat-btn { color:#000 !important; background:#fff !important; border:1px solid #000 !important; }
    .chatbot { background:#111 !important; color:#707070 !important; }
    </style>
    """)

    # --- States ---
    history_messages = gr.State([])        # current chat messages
    sessions_state   = gr.State([])        # [{"name": "Chat 1", "messages": [...]}, ...]
    selected_session = gr.State(None)      # current dropdown selection

    with gr.Row():
        # LEFT
        with gr.Column(scale=1, min_width=260, elem_classes="left-panel"):
            gr.Markdown("<div class='sidebar-title'>Chatbot</div>")
            new_chat_btn = gr.Button("➕ New Chat", elem_id="new-chat-btn")
            gr.Markdown("<div class='history-title'>Chat History</div>")
            session_select = gr.Dropdown(choices=[], label=None, interactive=True, elem_id="session-select")

            with gr.Group(elem_id="top-k-wrapper"):
                top_k = gr.Slider(1, 10, value=5, step=1, label="Top-K")

            with gr.Group(elem_id="temperature-wrapper"):
                temperature = gr.Slider(0.0, 1.0, value=0.1, step=0.1, label="Temperature")

        # RIGHT
        with gr.Column(scale=3):
            chatbot = gr.Chatbot(height=520, elem_classes=["chatbot"])
            question = gr.Textbox(placeholder="Ask...", lines=2, show_label=False, elem_id="question-box")
            with gr.Row():
                ask_btn   = gr.Button("Ask")
                clear_btn = gr.Button("Clear")

            gr.Markdown(
                """
                <div class="note-text">
                ⚠️ <em>This is an agent just providing some recommendations.</em>
                For final decisions, please visit the official website:<br>
                https://josaa.admissions.nic.in/applicant/seatmatrix/openingclosingrankarchieve.aspx
                JoSAA Opening & Closing Ranks
                </a>
                </div>
                """,
                elem_classes="note-text"
            )

    # Events
    ask_btn.click(
        ask,
        inputs=[history_messages, question, top_k, temperature, sessions_state, session_select],
        outputs=[chatbot, history_messages, sessions_state]
    )

    question.submit(
        ask,
        inputs=[history_messages, question, top_k, temperature, sessions_state, session_select],
        outputs=[chatbot, history_messages, sessions_state]
    )

    clear_btn.click(clear_chat, None, [chatbot, history_messages])

    new_chat_btn.click(
        new_chat_fn,
        inputs=[history_messages, sessions_state, session_select],
        outputs=[chatbot, history_messages, sessions_state, session_select]
    )

    session_select.change(
        load_session_fn,
        inputs=[sessions_state, session_select],
        outputs=[chatbot, history_messages]
    )

demo.launch()