import streamlit as st
from pymongo import MongoClient
from datetime import datetime

# MongoDB 연결 설정
client = MongoClient("mongodb://localhost:27017")
db = client["product_database"]
collection = db["products"]

# 사이드바 메뉴 설정
st.sidebar.title("메뉴")
selection = st.sidebar.radio("이동할 페이지", ("상품 등록", "상품 목록"))

if selection == "상품 등록":
    # 상품 등록 페이지 레이아웃
    st.title("상품 등록 페이지")

    product_name = st.text_input("상품 이름")
    registration_date = st.date_input("등록 날짜")
    registration_datetime = datetime.combine(registration_date, datetime.now().time())  # datetime 객체로 변환
    image_upload = st.file_uploader("이미지 업로드")
    product_price = st.number_input("상품 가격")

    if st.button("상품 등록"):
        # 입력한 정보를 MongoDB에 저장
        product = {
            "상품 이름": product_name,
            "등록 날짜": registration_datetime,
            "이미지": image_upload.read(),
            "상품 가격": product_price
        }
        collection.insert_one(product)
        st.success("상품이 등록되었습니다.")

else:
    # 상품 목록 페이지 레이아웃
    st.title("상품 목록 페이지")

    # MongoDB에서 모든 상품 검색
    products = collection.find()

    for product in products:
        st.write("상품 이름:", product["상품 이름"])
        st.write("등록 날짜:", product["등록 날짜"])
        st.image(product["이미지"])
        st.write("상품 가격:", product["상품 가격"])
        st.write("---")