파이썬 오류 좀 고쳐주세요 ㅠㅠ

Question

import time

import requests
import streamlit as st

API_BASE_URL = http://localhost:8000/qna

# Fastapi로 api 생성
def request_chat_api(user_message: str) -> str:
    url = API_BASE_URL

resp = requests.post(
        url,
        json={
            user_message: user_message,
        },
    )
    resp = resp.json()
    print(resp)
    return resp[answer]

def init_streamlit():
    st.set_page_config(page_title=Dr. KHU, page_icon=🩺)

if messages not in st.session_state:
        st.session_state.messages = [{role: assistant, content: 안녕하세요! Dr.seo입니다🩺}]
    
    # Initialize chat history
    if messages not in st.session_state:
        st.session_state.messages = []

# Display chat messages from history on app rerun
    for message in st.session_state.messages:
        with st.chat_message(message[role]):
            st.markdown(message[content])

def chat_main():
    if message := st.chat_input():
        # Add user message to chat history
        st.session_state.messages.append({role: user, content: message})
        # Display user message in chat message container
        with st.chat_message(user):
            st.markdown(message)

# Display assistant response in chat message container
        assistant_response = request_chat_api(message)

with st.chat_message(assistant):
            message_placeholder = st.empty()
            full_response = 
            for lines in assistant_response.split(
):
                for chunk in lines.split():
                    full_response += chunk +  
                    time.sleep(0.05)
                    # Add a blinking cursor to simulate typing
                    message_placeholder.markdown(full_response)
                full_response +=

message_placeholder.markdown(full_response)

# Add assistant response to chat history
        st.session_state.messages.append(
            {role: assistant, content: full_response}
        )

if __name__ == __main__:
    init_streamlit()
    chat_main()

이 코드를 실행시키면 자꾸 AttributeError: st.session_state has no attribute messages. Did you forget to initialize it? More info: https://docs.streamlit.io/library/advanced-features/session-state#initialization 라고 뜨네요..

백승윤 · Answer

와! 커리어리 QnA에서 Streamlit 질문을 만나게 되다니 이제서야 본게 괜히 죄송스럽네요. (저 Streamlit 짱 좋아합니다.)

음 제가 일단 제 로컬에서 streamlit 버전 1.35.0 로 실행해봤는데 잘 실행되는 것 같은데요? 우선 버전은 바꿔보시겠어요? 그리고 코드 안에 st.session_state에 message가 없을 때, initialize해주는 코드가 좀 중복되는 것 같은데 그 부분도 우선 수정해보시겠어요? 에러랑은 관련 없을 것 같긴합니다만..

최용빈 · Answer

제가 작성한 코드인데 여기서 보게 될 줄은 몰랐네요,,, 세션 초기화가 중복되는 거 말고는 다른 부분은 문제 없던 걸로 기억합니다! 개선하는 작업을 하는 레포가 있으니 여기 참고해주시면 될 거 같아요 :) https://github.com/whybe-choi/advanced-rag-project/blob/main/app.py

https://github.com/whybe-choi/advanced-rag-project/blob/main/app.py