ドキュメント横断検索

import os

import streamlit as st
from qaip import Qaip

st.set_page_config(page_title="QAIP Chatbot", page_icon="🤖", layout="wide")

st.title("🤖 QAIP Chatbot")
st.markdown("Chat with an AI assistant powered by QAIP")

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

if "qaip_client" not in st.session_state:
    api_key = os.getenv("QAIP_API_KEY", "your-api-key-here")
    st.session_state.qaip_client = Qaip(
        api_key=api_key,
        base_url="https://developer.qaip.com/api/v1",
    )

with st.sidebar:
    st.header("Settings")

    api_key_input = st.text_input(
        "API Key",
        type="password",
        value=os.getenv("QAIP_API_KEY", ""),
        help="Enter your QAIP API key",
    )

    if api_key_input:
        st.session_state.qaip_client = Qaip(
            api_key=api_key_input,
            base_url="https://developer.qaip.com/api/v1",
        )

    if st.button("Clear Chat History", type="secondary"):
        st.session_state.messages = []
        st.rerun()

for message in st.session_state.messages:
    with st.chat_message(message["role"]):
        st.markdown(message["content"])

if prompt := st.chat_input("What would you like to know?"):
    st.session_state.messages.append({"role": "user", "content": prompt})

    with st.chat_message("user"):
        st.markdown(prompt)

    with st.chat_message("assistant"):
        message_placeholder = st.empty()
        full_response = ""

        try:
            with st.spinner("Thinking..."):
                response = st.session_state.qaip_client.completion(
                    messages=st.session_state.messages,
                    citation=True,
                    tags=["documentation"],
                )

                if hasattr(response, "choices") and response.choices:
                    full_response = response.choices[0].message.content
                elif hasattr(response, "content"):
                    full_response = response.content
                elif hasattr(response, "text"):
                    full_response = response.text
                else:
                    full_response = str(response)

                message_placeholder.markdown(full_response)

        except Exception as e:
            error_msg = f"Error: {str(e)}"
            message_placeholder.error(error_msg)
            full_response = error_msg

        st.session_state.messages.append(
            {"role": "assistant", "content": full_response}
        )

if not st.session_state.messages:
    st.info("👋 Welcome! Start a conversation by typing a message below.")