Update app.py

app.py

@@ -24,12 +24,13 @@ model = genai.GenerativeModel(
     generation_config=generation_config,
 )
 
-def generate_pairing_recommendation(input_text, input_type):
+def generate_pairing_recommendation(input_text, input_type, situation):
     """
     입력받은 안주 또는 주류에 어울리는 페어링 정보를 생성합니다.
     Args:
         input_text (str): 안주 또는 주류 정보
         input_type (str): 입력 종류 ('안주' 또는 '주류')
+        situation (str): 상황 또는 분위기 정보
     Returns:
         str: 생성된 페어링 정보
     """
@@ -38,6 +39,7 @@ def generate_pairing_recommendation(input_text, input_type):
     당신은 주류 페어링 전문가입니다.
     사용자가 안주 또는 주류를 입력하면, 해당 안주에 어울리는 주류 또는 해당 주류에 어울리는 안주를 구체적으로 설명해주세요. 또한 각각의 특징, 맛, 향, 그리고 추천 이유를 포함하여 상세하게 작성해주세요.
     한국인의 취향에 맞는 주류나 안주를 추천해주세요. 추천 점수를 10점 만점으로 표현해주세요. 주류나 안주 추천은 종류별로 다양하게 해주세요.
+    상황이나 분위기 정보를 고려하여 최적의 페어링을 추천해주세요.
     다음은 페어링 추천에 필요한 정보입니다.
     **페어링 구성 요소:**
     1. 추천 주류/안주 소개
@@ -51,6 +53,7 @@ def generate_pairing_recommendation(input_text, input_type):
         SYSTEM_PROMPT,
         f"**입력 종류:** {input_type}",
         f"**입력 내용:** {input_text}",
+        f"**상황/분위기:** {situation}",
         "**추천 페어링:**", 
     ]
 
@@ -83,15 +86,19 @@ tabs = st.tabs(["🍝 안주 입력", "🍇 주류 입력"])
 
 with tabs[0]:
     st.markdown("#### 안주를 입력해 주세요:")
-    input_text_anjoo = st.text_area("예: 치킨, 육회, 치즈 플래터 등", height=100)
+    input_text_anjoo = st.text_area("예: 치킨, 육회, 치즈 플래터 등", height=50)
 
 with tabs[1]:
     st.markdown("#### 주류를 입력해 주세요:")
-    input_text_liquor = st.text_area("예: 맥주, 와인, 소주 등", height=100)
+    input_text_liquor = st.text_area("예: 맥주, 와인, 소주 등", height=50)
 
 input_type = "안주" if input_text_anjoo else "주류"
 input_text = input_text_anjoo if input_text_anjoo else input_text_liquor
 
+# 상황 또는 분위기 입력
+st.markdown("#### 상황이나 분위기를 입력해 주세요:")
+situation = st.text_area("예: 친구들과의 모임, 로맨틱한 데이트, 가족과의 저녁 식사 등", height=100)
+
 generate_button = st.button("🍻 페어링 추천 받기")
 
 # 출력 영역 정의
@@ -103,7 +110,7 @@ if generate_button and input_text:
     output_area.markdown(output_text)
     try:
         with st.spinner("페어링 정보를 생성 중입니다... 잠시만 기다려 주세요!"):
-            for partial_output in generate_pairing_recommendation(input_text, input_type):
+            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: