Update app.py

app.py

@@ -64,6 +64,13 @@ def generate_pairing_recommendation(input_text, input_type, situation):
         yield collected_text  # 실시간 출력을 위해 yield 사용
         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="티 페어링 도우미",
@@ -88,39 +95,52 @@ tabs = st.sidebar.tabs(["🍭 다과 입력", "🍵 티 입력"])
 
 with tabs[0]:
     st.markdown("#### 다과를 입력해 주세요:")
-    input_text_snack = st.text_input("예: 마카롱, 스콘, 다크 초콜릿 등")
+    input_text_snack = st.text_input("예: 마카롱, 스콘, 다크 초콜릿 등", key="input_text_snack")
 
 with tabs[1]:
     st.markdown("#### 티를 입력해 주세요:")
-    input_text_tea = st.text_input("예: 녹차, 홍차, 캐모마일 등")
-
-input_type = "다과" if input_text_snack else "티"
-input_text = input_text_snack if input_text_snack else input_text_tea
+    input_text_tea = st.text_input("예: 녹차, 홍차, 캐모마일 등", key="input_text_tea")
 
 # 상황 또는 분위기 입력 (사이드바로 이동)
 st.sidebar.markdown("#### 상황이나 분위기를 입력해 주세요:")
-situation = st.sidebar.text_area("예: 오후의 여유로운 시간, 친구들과의 티타임, 집중이 필요한 공부 시간 등", height=100)
+situation = st.sidebar.text_area("예: 오후의 여유로운 시간, 친구들과의 티타임, 집중이 필요한 공부 시간 등", height=100, key="situation")
 
+# 페어링 추천 받기 버튼
 generate_button = st.sidebar.button("🍻 페어링 추천 받기")
 
+# 입력 초기화 버튼
+clear_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_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: # 입력값이 있을 때만 추천 시작
+        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("티 또는 다과를 입력해주세요.")
+
+# 입력 초기화 기능
+if clear_button:
+    clear_input_fields()
 
 # 추가 정보 및 FAQ
 add_vertical_space(2)
 stoggle(
     "📜 티 페어링 FAQ",
     "이 앱은 티와 다과에 대한 전문적인 페어링 정보를 제공합니다. 한국인의 입맛에 맞춘 추천을 통해 최고의 티와 다과 조합을 즐기세요!\n\n\n**Q: 모든 티와 다과를 추천받을 수 있나요?**\n\nA: 예, 다양한 티와 다과에 대해 추천받을 수 있으며, 지속적으로 업데이트됩니다.",
-)
+)