01_my_first_AI.py

import streamlit as st
from langchain.chat_models import ChatOpenAI
from langchain.schema import (SystemMessage, HumanMessage, AIMessage)


def main():
    llm = ChatOpenAI(temperature=0.9)
    # タブのタイトルなどを決める
    st.set_page_config(
        page_title="My Great ChatGPT",
        page_icon=""
    )
    st.header("My Great ChatGPT 🤗")

    # チャット履歴の初期化
    if "messages" not in st.session_state:
        st.session_state.messages = [
            SystemMessage(content="I am a helpful assistant.")
        ]

    #  ユーザーの入力を監視
    # if user_input := st.chat_input("聞きたいことを入力してね！"):
    #     st.session_state.messages.append(HumanMessage(content=user_input))
    #     with st.spinner("ChatGPT is typing ..."):
    #         response = llm(st.session_state.messages)
    #     st.session_state.messages.append(AIMessage(content=response.content))

    container = st.container()
    with container:
        with st.form(key='my_form', clear_on_submit=True):
            user_input = st.text_area(label='Message: ', key='input', height=100)
            submit_button = st.form_submit_button(label='Send')

    if submit_button and user_input:
        # 何か入力されて Submit ボタンが押されたら実行される
        st.session_state.messages.append(HumanMessage(content=user_input))
        # Streamlitのスピナー機能を利用し、ユーザーに対して処理中であることを伝える
        with st.spinner("ChatGPT is typing ..."):
            response = llm(st.session_state.messages)
        st.session_state.messages.append(AIMessage(content=response.content))
    # チャット履歴の表示
    messages = st.session_state.get('messages', [])
    for message in messages:
        if isinstance(message, AIMessage):
            with st.chat_message('assistant'):
                st.markdown(message.content)
        elif isinstance(message, HumanMessage):
            with st.chat_message('user'):
                st.markdown(message.content)
        else:  # isinstance(message, SystemMessage):
            st.write(f"System message: {message.content}")


if __name__ == '__main__':
    main()