streamlit_chat.py

# Created by 'sandeep.mishra@digital.ai' at 9:21 pm 12/05/25 using PyCharm
import streamlit as st
import requests
import uuid

import os
from dotenv import load_dotenv
load_dotenv()


# API endpoint
API_URL = os.getenv("API_URL", "http://localhost:8003/query")

st.set_page_config(page_title="Chat with Agent", layout="centered")

# Session ID management
if "session_id" not in st.session_state:
    st.session_state.session_id = str(uuid.uuid4())

st.title("MCP Client: How can I help you?")

# Display the conversation history at the top
if "conversation" not in st.session_state:
    st.session_state.conversation = []

for msg in st.session_state.conversation:
    role, content = msg
    with st.chat_message(role):
        st.markdown(content)

# Input: chat-style, auto-submits on Enter
user_input = st.chat_input("Type your message here...")

if user_input:
    # Display user message immediately
    st.session_state.conversation.append(("user", f"**You**: {user_input}"))

    res = requests.post("http://localhost:8003/query", json={
        "message": user_input,
        "session_id": st.session_state.session_id
    })

    data = res.json()
    st.session_state.session_id = data.get("session_id")

    # Append assistant response
    response = data.get("response")
    if isinstance(response, dict) and "output" in response:
        content = response["output"]
    else:
        content = str(response)

    st.session_state.conversation.append(("assistant", f"**Assistant**: {content}"))

    # Rerun to update UI
    st.rerun()

# Scroll to the latest message
st.markdown('<script>window.scrollTo(0, document.body.scrollHeight);</script>', unsafe_allow_html=True)