import streamlit as st
from templates.user_info import show_user_info_form
from templates.recommendations import show_recommendations_tab
from templates.recommendations import show_virtual_tryon_tab

# 页面设置
st.set_page_config(page_title="智能穿搭助手", page_icon="👗", layout="wide")

# 标题和介绍
st.title("👗 智能穿搭助手")

# 初始化session state
if 'user_info' not in st.session_state:
    st.session_state.user_info = {}
if 'recommendations' not in st.session_state:
    st.session_state.recommendations = None

# 侧边栏 - 用户信息收集
show_user_info_form()

# 主界面
tab1, tab2 = st.tabs(["穿搭推荐", "虚拟试衣"])

with tab1:
    show_recommendations_tab()

with tab2:
    show_virtual_tryon_tab()

if __name__ == "__main__":
    st.write("")