Update app.py

app.py

@@ -4,8 +4,7 @@ import streamlit as st
 import google.generativeai as genai
 from streamlit_extras.colored_header import colored_header
 from streamlit_extras.add_vertical_space import add_vertical_space
-import markdown
-from streamlit_extras.stoggle import stoggle  # 추가된 라이브러리
+from streamlit_extras.stoggle import stoggle
 
 # Google Gemini API Key 설정
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
@@ -27,10 +26,13 @@ model = genai.GenerativeModel(
 def generate_pairing_recommendation(input_text, input_type, situation):
     """
     입력받은 안주 또는 주류에 어울리는 페어링 정보를 생성합니다.
+    상황/분위기를 고려하여 한국인의 취향에 맞는 주류나 안주를 추천합니다.
+
     Args:
         input_text (str): 안주 또는 주류 정보
         input_type (str): 입력 종류 ('안주' 또는 '주류')
         situation (str): 상황 또는 분위기 정보
+
     Returns:
         str: 생성된 페어링 정보
     """
@@ -38,13 +40,14 @@ def generate_pairing_recommendation(input_text, input_type, situation):
     SYSTEM_PROMPT = f"""
     당신은 주류 페어링 전문가입니다.
     사용자가 안주 또는 주류를 입력하면, 상황/분위기: {situation}를 고려하여 해당 안주에 어울리는 주류 또는 해당 주류에 어울리는 안주를 구체적으로 설명해주세요. 또한 각각의 특징, 맛, 향, 그리고 추천 이유를 포함하여 상세하게 작성해주세요.
-    한국인의 취향에 맞는 주류나 안주를 추천해주세요. 추천 점수를 10점 만점으로 표현해주세요. 주류나 안주 추천은 종류별로 다양하게 해주세요.(4가지 이상)
+    한국인의 취향에 맞는 주류나 안주를 추천해주세요. 추천 점수는 10점 만점으로 표현해주세요. 주류나 안주 추천은 종류별로 다양하게 해주세요.(4가지 이상)
     상황이나 분위기를 반드시 고려하여 최적의 페어링을 추천해주세요. 상황이나 분위기에 맞춘 특별한 추천 이유를 추가하고, 해당 상황에 적합한 맛과 향의 요소를 강조해주세요.
-    다음은 페어링 추천에 필요한 정보입니다.
+    예를 들어, '{situation}' 상황이라면, 더욱 즐거운 시간을 보낼 수 있도록 특별히 어울리는 페어링을 추천해주세요.
+
     **페어링 구성 요소:**
     1. 추천 주류/안주 소개
     2. 맛과 향 설명
-    3. 추천 이유
+    3. 추천 이유 (상황/분위기 고려)
     4. 추가 추천 페어링
     5. 추천 점수 (10점 만점)
     """
@@ -53,7 +56,7 @@ def generate_pairing_recommendation(input_text, input_type, situation):
         SYSTEM_PROMPT,
         f"**입력 종류:** {input_type}",
         f"**입력 내용:** {input_text}",
-        "**추천 페어링:**", 
+        "**추천 페어링:**",
     ]
 
     response = model.generate_content(prompt, stream=True)
@@ -61,8 +64,15 @@ def generate_pairing_recommendation(input_text, input_type, situation):
     for token in response:
         chunk = token.text
         collected_text += chunk
-        yield collected_text  # 실시간 출력을 위해 yield 사용
-        time.sleep(0.03)  # 출력 속도 조절
+        yield collected_text
+        time.sleep(0.03)
+
+# 입력 필드 초기화 함수
+def clear_input_fields():
+    """입력 필드를 초기화합니다."""
+    st.session_state.input_text_anjoo = ""
+    st.session_state.input_text_liquor = ""
+    st.session_state.situation = ""
 
 # Streamlit Interface
 st.set_page_config(
@@ -83,40 +93,46 @@ add_vertical_space(1)
 # 사이드바 입력 영역
 st.sidebar.markdown("## 🛌️ 입력 정보")
 
-# 사용자 입력 영역 (사이드바로 이동)
+# 사용자 입력 영역 (사이드바 탭으로 분리)
 tabs = st.sidebar.tabs(["🍝 안주 입력", "🍇 주류 입력"])
 
 with tabs[0]:
     st.markdown("#### 안주를 입력해 주세요:")
-    input_text_anjoo = st.text_input("예: 치킨, 육회, 치즈 플래터 등")
+    input_text_anjoo = st.text_input("예: 치킨, 육회, 치즈 플래터 등", key="input_text_anjoo")  # key 추가
 
 with tabs[1]:
     st.markdown("#### 주류를 입력해 주세요:")
-    input_text_liquor = st.text_input("예: 맥주, 와인, 소주 등")
-
-input_type = "안주" if input_text_anjoo else "주류"
-input_text = input_text_anjoo if input_text_anjoo else input_text_liquor
+    input_text_liquor = st.text_input("예: 맥주, 와인, 소주 등", key="input_text_liquor")  # key 추가
 
-# 상황 또는 분위기 입력 (사이드바로 이동)
+# 상황 또는 분위기 입력 (사이드바)
 st.sidebar.markdown("#### 상황이나 분위기를 입력해 주세요:")
-situation = st.sidebar.text_area("예: 친구들과의 모임, 로맨틱한 데이트, 가족과의 저녁 식사 등", height=100)
+situation = st.sidebar.text_area("예: 친구들과의 모임, 로맨틱한 데이트, 가족과의 저녁 식사 등", height=100, key="situation")  # key 추가
 
+# 페어링 추천 받��� 버튼
 generate_button = st.sidebar.button("🍻 페어링 추천 받기")
 
 # 출력 영역 정의
 output_area = st.empty()
 
 # 추천 결과 생성 및 출력
-if generate_button and input_text:
-    output_text = ""
-    output_area.markdown(output_text)
-    try:
-        with st.spinner("페어링 정보를 생성 중입니다... 잠시만 기다려 주세요!"):
-            for partial_output in generate_pairing_recommendation(input_text, input_type, situation):
-                output_text = partial_output
-                output_area.markdown(output_text)
-    except Exception as e:
-        st.error(f"오류가 발생했습니다: {str(e)}")
+if generate_button:
+    input_type = "안주" if st.session_state.input_text_anjoo else "주류"
+    input_text = st.session_state.input_text_anjoo if st.session_state.input_text_anjoo else st.session_state.input_text_liquor
+    situation = st.session_state.situation
+
+    if input_text:  # 입력값이 있을 때만 추천 시작
+        output_text = ""
+        output_area.markdown(output_text)
+        try:
+            with st.spinner("페어링 정보를 생성 중입니다... 잠시만 기다려 주세요!"):
+                for partial_output in generate_pairing_recommendation(input_text, input_type, situation):
+                    output_text = partial_output
+                    output_area.markdown(output_text)
+            clear_input_fields()  # 입력 필드 초기화
+        except Exception as e:
+            st.error(f"오류가 발생했습니다: {str(e)}")
+    else:
+        st.warning("안주 또는 주류를 입력해주세요.")
 
 # 추가 정보 및 FAQ
 add_vertical_space(2)