client.py

import streamlit as st
from openai import OpenAI
import requests

# 設定値
URL = "http://localhost:7133/api/invoke/sync"
REQUIRE_ADDITIONAL_INFO = True

# タイトルを表示する。
st.title("Simple agent client")

if "Clear" not in st.session_state:
    st.session_state.Clear = False

if "messages" not in st.session_state:
    st.session_state.messages = []

if REQUIRE_ADDITIONAL_INFO & ("additional_info" not in st.session_state):
    st.session_state.additional_info = []

if "called_agent" not in st.session_state:
    st.session_state.called_agent = []

for idx, message in enumerate(st.session_state.messages):
    if message["role"] == "user":
        with st.chat_message(message["role"], avatar="😊"):
            st.markdown(message["content"])
    elif message["role"] == "assistant":
        with st.chat_message(message["role"], avatar="🤖"):
            st.markdown(f"""
                        {message["content"]}
                        
                        使用されたエージェント:{st.session_state.called_agent[idx]}
                            """)

if prompt := st.chat_input("ここに入力"):
    st.session_state.messages.append({"role": "user", "content": prompt})
    st.session_state.called_agent.append("")
    with st.chat_message("user", avatar="😊"):
        st.markdown(prompt)

    with st.chat_message("assistant", avatar="🤖"):
        message_placeholder = st.empty()
        
        # APIにPOSTリクエストを送信
        response = requests.post(URL, json={"messages": st.session_state.messages, "requireAdditionalInfo": REQUIRE_ADDITIONAL_INFO})
        response_data = response.json()
        response_content = response_data["content"]
        called_agents  = ", ".join(response_data["calledAgentNames"])
        
        message_placeholder.markdown(f"""
                                     {response_content}
                                     
                                     使用されたエージェント:{called_agents}
                                     """)
        st.session_state.messages.append({"role": "assistant", "content": response_content})
        st.session_state.called_agent.append(called_agents)
        if REQUIRE_ADDITIONAL_INFO and len(response_data["additionalInfo"]) > 0:
            st.session_state.additional_info.extend(response_data["additionalInfo"])
        st.session_state.Clear = True

if st.session_state.Clear:
    if st.button('clear chat history'):
        st.session_state.messages = []
        st.session_state.called_agent = []
        full_response = ""
        st.session_state.Clear = False 
        st.rerun()


if REQUIRE_ADDITIONAL_INFO:
    st.markdown(
        """
       <style>
       [data-testid="stSidebar"][aria-expanded="true"]{
           min-width: 550px;
           max-width: 550px;
       }
       """,
        unsafe_allow_html=True,
    )   
    with st.sidebar:
        st.write("### Additional information")
        for idx, info in enumerate(st.session_state.additional_info):
            if info["$type"] == "markdown":
                st.markdown(info["markdownText"])
            elif info["$type"] == "link":
                st.write(info["linkText"])
                st.write(info["linkUrl"])
            st.markdown("---")