Youtube 무작정 따라하기 - Chatbot PDF 문서를 AI로

Youtube 동영상에 Python streamlit을 이용하여 PDF 문서의 테이블을 검색하여 나만의 chatbot을 만들 수 있게 도와 주는 동영상 입니다.

 

추천 드립니다.

https://www.youtube.com/watch?v=3_UYupYCeSc

 

위의 동영상의 코드를 따라해 봤습니다.

자세한 내용은 위의 동영상을 보시면 저처럼 따라 할 수 있게 자세히 설명해 주고 계시네요. 

 

[소스코드]

# OpenAI Assistants API + Streamlit 문서를 읽고 검색하는 인공지능 챗봇 웹앱 만들기
# https://www.youtube.com/watch?v=3_UYupYCeSc
import streamlit as st
from openai import OpenAI
import time


assistant_id = ""

with st.sidebar:
    openai_api_key = st.text_input("OpenAI API Key", key="chatbot_api_key", type="password", value="")
    "[Get an OpenAI API key](https://platform.openai.com/account/api-keys)"

    client = OpenAI(api_key=openai_api_key)

    thread_id = st.text_input("Thread ID")

    thread_btn = st.button("Create a new thread")

    if thread_btn:
        thread = client.beta.threads.create()
        thread_id = thread.id

        st.subheader(f"{thread_id}", divider="rainbow")
        st.info("스레드가 생성되었습니다.")

st.title("💬 2023년 한국 개봉작 Chatbot")

if "messages" not in st.session_state:
    st.session_state["messages"] = [{"role": "assistant", "content": "How can I help you?"}]

for msg in st.session_state.messages:
    st.chat_message(msg["role"]).write(msg["content"])

if prompt := st.chat_input():
    if not openai_api_key:
        st.info("Please add your OpenAI API key to continue.")
        st.stop()

    if not thread_id:
        st.info("Please add your thread Id key to continue.")
        st.stop()


    st.session_state.messages.append({"role": "user", "content": prompt})
    st.chat_message("user").write(prompt)

    response = client.beta.threads.messages.create(
        thread_id,
        role="user",
        content=prompt,
    )
    print(response)

    run = client.beta.threads.runs.create(
        thread_id=thread_id,
        assistant_id=assistant_id
    )
    print(run)

    run_id = run.id

    while True:
        run = client.beta.threads.runs.retrieve(
            thread_id=thread_id,
            run_id=run_id
        )
        if run.status == "completed":
            break
        else:
            time.sleep(2)
        print(run)

    thread_messages = client.beta.threads.messages.list(thread_id)
    print(thread_messages.data)

    msg=thread_messages.data[0].content[0].text.value
    print(msg)
    st.session_state.messages.append({"role": "assistant", "content": msg})
    st.chat_message("assistant").write(msg)

    #response = client.chat.completions.create(model="gpt-3.5-turbo", messages=st.session_state.messages)
    #msg = response.choices[0].message.content
    #st.session_state.messages.append({"role": "assistant", "content": msg})
    #st.chat_message("assistant").write(msg)

 

[결과 화면]