Update app.py

app.py

@@ -61,13 +61,6 @@ def generate_pairing_recommendation(input_text, input_type, situation):
         yield collected_text
         time.sleep(0.03)
 
-# 입력 필드 초기화 함수
-def clear_input_fields():
-    """입력 필드를 초기화합니다."""
-    st.session_state.input_text_snack = ""
-    st.session_state.input_text_tea = ""
-    st.session_state.situation = ""
-
 # Streamlit Interface
 st.set_page_config(
     page_title="티 페어링 마스터",
@@ -92,11 +85,11 @@ tabs = st.sidebar.tabs(["🍭 다과 입력", "🍵 티 입력"])
 
 with tabs[0]:
     st.markdown("#### 다과를 입력해 주세요:")
-    input_text_snack = st.text_input("예: 수제 양갱, 마들렌, 과일 타르트 등", key="snack")
+    input_text_snack = st.text_input("예: 수제 양갱, 마들렌, 과일 타르트 등", key="input_text_snack")
 
 with tabs[1]:
     st.markdown("#### 티를 입력해 주세요:")
-    input_text_tea = st.text_input("예: 백차, 보이차, 루이보스 등", key="tea")
+    input_text_tea = st.text_input("예: 백차, 보이차, 루이보스 등", key="input_text_tea")
 
 # 상황 또는 분위기 입력 (사이드바)
 st.sidebar.markdown("#### 상황이나 분위기를 입력해 주세요:")
@@ -113,8 +106,8 @@ output_area = st.empty()
 
 # 추천 결과 생성 및 출력
 if generate_button:
-    input_type = "다과" if st.session_state.snack else "티"
-    input_text = st.session_state.snack if st.session_state.snack else st.session_state.tea
+    input_type = "다과" if st.session_state.input_text_snack else "티"
+    input_text = st.session_state.input_text_snack if st.session_state.input_text_snack else st.session_state.input_text_tea
     situation = st.session_state.situation
 
     if input_text:  # 입력값이 있을 때만 추천 시작
@@ -125,7 +118,6 @@ if generate_button:
                 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:
@@ -133,7 +125,9 @@ if generate_button:
 
 # 입력 초기화 기능
 if clear_button:
-    clear_input_fields()
+    st.session_state.input_text_snack = ""
+    st.session_state.input_text_tea = ""
+    st.session_state.situation = ""
 
 # 추가 정보 및 FAQ
 add_vertical_space(2)