Merge pull request #4 from choi-ce3l/feature/streamlit

Feature/streamlit

main.py

@@ -0,0 +1,220 @@
+import streamlit as st
+
+from search_db import search_db
+from chatbot import chatbot_response
+import webbrowser
+
+with st.sidebar:
+    # Streamlit UI 구성
+    st.title("💬 맞춤형 화장품 추천 챗봇")
+    st.write(
+        "피부 타입과 고민, 원하는 기능과 제형을 선택하면 챗봇이 제품을 추천해줍니다. 추가로 궁금한 점을 물어보세요!"
+    )
+
+    # 챗봇 초기화
+    if "chat_history" not in st.session_state:
+        st.session_state.chat_history = []
+
+    # 사용자 입력
+    with st.form(key="recommendation_form"):
+        skin_type = st.selectbox(
+            "피부 타입을 선택하세요:",
+            ("건성", "지성", "복합성", "민감성", "악건성", "트러블성", "중성"),
+            index=None,
+            placeholder="피부 타입"
+        )
+        skin_tone = st.selectbox(
+            "피부 톤을 선택하세요:",
+            ("쿨톤", "웜톤", "봄웜톤", "여름쿨톤", "가을웜톤", "겨울쿨톤"),
+            index=None,
+            placeholder="피부톤"
+        )
+        skin_concern = st.multiselect(
+            "피부 고민을 선택하세요:",
+            [
+                "잡티",
+                "주름",
+                "미백",
+                "각질",
+                "트러블",
+                "블랙헤드",
+                "피지과다",
+                "민감성",
+                "모공",
+                "탄력",
+                "홍조",
+                "아토피",
+                "다크서클",
+            ],
+            placeholder="피부 고민"
+        )
+        product_function = st.selectbox(
+            "원하는 화장품 기능을 선택하세요:", ("수분", "진정", "미백", "탄력"),
+            index=None,
+            placeholder="기능"
+        )
+        product_texture = st.selectbox(
+            "원하는 제형을 선택하세요:", ("가벼운 제형", "무거운 제형"),
+            index=None,
+            placeholder="제형"
+        )
+        category = st.selectbox(
+            "원하는 카테고리를 선택하세요:", ("스킨/토너", "에센스/세럼/앰플", "크림", "로션", "미스트/오일"),
+            index=None,
+            placeholder="카테고리"
+        )
+        pricerange = st.selectbox(
+            "원하는 가격대를 선택하세요:", ("~2만원", "2만원~3만원", "3만원~5만원", "5만원~"),
+            index=None,
+            placeholder="가격대"
+        )
+
+        user_data = {
+            "skintype": skin_type,
+            "skintone": skin_tone,
+            "skinconcerns": " ".join(skin_concern),
+            "function": product_function,
+            "formulation": product_texture,
+            "category": category,
+            "pricerange": pricerange,
+        }
+
+        st.write(user_data)
+
+        # 추천 버튼
+        recommand_button = st.form_submit_button("추천받기")
+        if recommand_button:
+            # 모든 선택지가 설정되었는지 확인
+            if not all(
+                [
+                    user_data["formulation"],  # 제형
+                    user_data["function"],  # 기능
+                    user_data["skinconcerns"],  # 피부 고민
+                    user_data["skintone"],  # 피부 톤
+                    user_data["skintype"],  # 피부 타입
+                    user_data['category'],  # 카테고리
+                    user_data['pricerange'] # 가격대
+                ]
+            ):
+                st.error("모든 선택지를 설정한 후 다시 시도해주세요!")
+
+
+recommand_list = [
+    ('A000000186465', 1.0),
+    ('A000000155321', 1.0),
+    ('A000000155320', 1.0),
+    ('A000000211127', 1.0),
+    ('A000000210068', 1.0),
+    ('A000000141900', 1.0)
+]
+
+def get_product_info(recommand_list):
+    recommand_list = [i[0] for i in recommand_list]
+    product_info = search_db(recommand_list)
+    return product_info
+
+
+# 세션 상태 초기화
+if "messages" not in st.session_state:
+    st.session_state.messages = []  # 메시지 기록 초기화
+
+# 샘플 데이터
+product_info = [
+    {
+        "image_link": "https://via.placeholder.com/150",
+        "goodsName": "상품1",
+        "price": 10000,
+        "function": "보습",
+        "formulation": "크림",
+        "purchase_link": "https://example.com/product1"
+    },
+    {
+        "image_link": "https://via.placeholder.com/150",
+        "goodsName": "상품2",
+        "price": 20000,
+        "function": "미백",
+        "formulation": "로션",
+        "purchase_link": "https://example.com/product2"
+    },
+    {
+        "image_link": "https://via.placeholder.com/150",
+        "goodsName": "상품3",
+        "price": 30000,
+        "function": "진정",
+        "formulation": "세럼",
+        "purchase_link": "https://example.com/product3"
+    }
+]
+
+# 메시지 추가 함수
+def add_product_message(product_list):
+    """추천 제품 메시지를 세션 상태에 추가"""
+    st.session_state.messages.append({
+        "role": "assistant",
+        "type": "product",
+        "content": product_list
+    })
+
+# 사용자 입력에 대한 답변 생성 함수
+def generate_response(user_message):
+    """사용자 메시지에 대한 간단한 답변 생성"""
+    if "추천" in user_message:
+        return "추천받기 버튼을 눌러서 상품을 확인해 보세요!"
+    elif "안녕" in user_message:
+        return "안녕하세요! 무엇을 도와드릴까요?"
+    else:
+        return "죄송해요, 그 내용을 잘 이해하지 못했어요. 다시 말씀해 주시겠어요?"
+
+# 버튼 클릭으로 제품 추천 메시지 추가
+if recommand_button:
+    product_info = get_product_info(recommand_list)
+    add_product_message(product_info[:3])  # 첫 3개의 상품 추천
+
+# 사용자 입력 처리
+user_input = st.chat_input("Your message:")
+
+if user_input:  # 사용자가 메시지를 입력했을 경우
+    # 사용자 메시지 저장
+    st.session_state.messages.append({
+        "role": "user",
+        "type": "text",
+        "content": user_input
+    })
+    # 답변 생성 및 저장
+    assistant_response = generate_response(user_input)
+    st.session_state.messages.append({
+        "role": "assistant",
+        "type": "text",
+        "content": assistant_response
+    })
+
+# 이전 메시지 출력
+for message in st.session_state.messages:
+    if message["type"] == "product":
+        product_list = message["content"]
+        with st.chat_message("assistant"):
+            col1, col2, col3 = st.columns(3)
+            with col1:
+                st.image(product_list[0]['image_link'])
+                st.write(product_list[0]['goodsName'])
+                st.write(product_list[0]['price'], "원")
+                st.write(product_list[0]['function'])
+                st.write(product_list[0]['formulation'])
+                st.link_button("구매하기", product_list[0]['purchase_link'])
+            with col2:
+                st.image(product_list[1]['image_link'])
+                st.write(product_list[1]['goodsName'])
+                st.write(product_list[1]['price'], "원")
+                st.write(product_list[1]['function'])
+                st.write(product_list[1]['formulation'])
+                st.link_button("구매하기", product_list[1]['purchase_link'])
+            with col3:
+                st.image(product_list[2]['image_link'])
+                st.write(product_list[2]['goodsName'])
+                st.write(product_list[2]['price'], "원")
+                st.write(product_list[2]['function'])
+                st.write(product_list[2]['formulation'])
+                st.link_button("구매하기", product_list[2]['purchase_link'])
+    elif message["type"] == "text":
+        with st.chat_message(message["role"]):
+            st.write(message["content"])

search_db.py

@@ -0,0 +1,46 @@
+import pymysql
+from dotenv import load_dotenv
+import os
+
+# product_list -> list
+def search_db(product_list):
+    load_dotenv()
+
+    # 데이터베이스 연결에 필요한 정보들 저장
+    host = os.getenv("DB_HOST")
+    port = os.getenv("DB_PORT")
+    user = os.getenv("DB_USER")
+    password = os.getenv("DB_PASSWD")
+    database = os.getenv("DB_NAME")
+
+    connection = pymysql.connect(
+        host=host,             # 예: "localhost"
+        port=int(port),        # MySQL 포트 넘버
+        user=user,             # MySQL 사용자 이름
+        password=password,     # MySQL 비밀번호
+        database=database,     # MySQL 데이터베이스 이름
+        charset="utf8mb4"      # 문자 인코딩
+    )
+
+    try:
+        with connection.cursor() as cursor:
+            result = []
+            for i in product_list:
+                query = f"select goodsName, purchase_link, image_link, price, `function`, formulation from cosmetic_data where goodsNo = '{i}';"
+                cursor.execute(query)
+                fetch = cursor.fetchone()
+                fetch = {
+                    "goodsName": fetch[0],
+                    "purchase_link": fetch[1],
+                    "image_link": fetch[2],
+                    "price": fetch[3],
+                    "function": fetch[4],
+                    "formulation": fetch[5]
+                }
+                result.append(fetch)
+
+            connection.commit()
+    finally:
+        connection.close()
+
+    return result