Function chat_page

app.py:271–381

()

                    st.markdown('<hr style="margin: 2rem 0;">', unsafe_allow_html=True)
                    
def chat_page():
    # Initialize session state
    if "messages" not in st.session_state:
        st.session_state.messages = []
    if "query" not in st.session_state:
        st.session_state.query = ""

    st.markdown("""
        <div style="text-align: center; padding: 2rem 0;">
            <h1 style="color: #81E831; font-size: 3em; font-weight: 800;">🤵‍♂️ Fashion AI Assistant</h1>
            <p style="color: #666; font-size: 1.2em;">Your personal style advisor powered by AI</p>
        </div>
    """, unsafe_allow_html=True)

    # Navigation buttons
    st.markdown("""
        <div class="nav-container">
            <a href="add_product_page" class="nav-button">Add Product Data</a>
            <a href="visual_search" class="nav-button">Visual Search</a>
        </div>
    """, unsafe_allow_html=True)

    # Show suggestions if no messages yet
    if not st.session_state.messages:
        render_suggestions()

    # Chat messages display
    for message in st.session_state.messages:
        with st.chat_message(message["role"], avatar="👤" if message["role"] == "user" else "👗"):
            if message["role"] == "assistant":
                st.markdown(message["content"]["response"])
                if message["content"].get("metadata") and message["content"]["metadata"].get("sources"):
                    render_results_section(message["content"])
            else:
                st.markdown(message["content"])

    # Handle query from suggestion buttons
    if st.session_state.query:
        query = st.session_state.query
        st.session_state.query = ""  # Clear the query
        
        # Add user message
        st.session_state.messages.append({
            "role": "user",
            "content": query
        })
        
        with st.chat_message("assistant", avatar="👗"):
            with st.spinner("Finding perfect matches..."):
                try:
                    response_data = get_rag_response(query)
                    st.markdown(response_data["response"])
                    if response_data.get("metadata") and response_data["metadata"].get("sources"):
                        render_results_section(response_data)
                except Exception as e:
                    st.error(f"An error occurred: {str(e)}")
                    response_data = {
                        "response": "I encountered an error while processing your request. Please try again.",