Update app.py

app.py

@@ -76,14 +76,6 @@ selected_standard = st.selectbox("성취기준을 선택하세요:", selected_st
 # 선택된 성취기준을 텍스트 입력창의 기본값으로 사용
 achievement_standard = st.text_input("성취기준 입력:", value=selected_standard)
 
-# 세션 상태 초기화
-if 'selected_sentence' not in st.session_state:
-    st.session_state.selected_sentence = None
-if 'similar_sentences' not in st.session_state:
-    st.session_state.similar_sentences = []
-if 'selected_index' not in st.session_state:
-    st.session_state.selected_index = 0
-
 # # "평가 생성" 버튼 클릭 시의 동작
 # if st.button("평가 생성"):
 #     with st.spinner('답변 생성중...'):
@@ -115,14 +107,21 @@ if 'selected_index' not in st.session_state:
 #         for sentence in st.session_state.similar_sentences:
 #             st.write(sentence)
 
+# 세션 상태 초기화
+if 'selected_sentence' not in st.session_state:
+    st.session_state.selected_sentence = None
+if 'similar_sentences' not in st.session_state:
+    st.session_state.similar_sentences = []
+
 # "평가 생성" 버튼 클릭 시의 동작
 if st.button("평가 생성"):
     with st.spinner('답변 생성중...'):
         result = generate_annotated_text(achievement_standard) 
         st.session_state.generated_result = result
+        st.session_state.selected_sentence = None  # 새로운 평가 생성시 선택된 문장 초기화
     exec(result.replace('```', ''))
 
-    # annotated_text 결과에서 문장만 추출 및 셀렉트 박스 추가
+    # annotated_text 결과에서 문장만 추출
     result_lines = result.split('\n')
     sentences = []
     for line in result_lines:
@@ -133,15 +132,15 @@ if st.button("평가 생성"):
             sentences.append(sentence)
 
     if sentences:
-        st.session_state.selected_sentence = st.selectbox("문장을 선택하세요:", sentences)
+        st.session_state.selected_sentence = st.selectbox("문장을 선택하세요:", sentences, key="sentence_select")
 
 # 유사한 문장 생성 버튼 추가
 if st.button("유사한 문구 생성") and st.session_state.get('selected_sentence'):
     with st.spinner('문장 생성중...'):
-        similar_sentences = generate_similar_sentences(st.session_state.selected_sentence)
-        for generated_sentence in similar_sentences:
-            st.write(generated_sentence)
+        st.session_state.similar_sentences = generate_similar_sentences(st.session_state.selected_sentence)
 
 # 생성된 결과 및 유사한 문장들 출력
 if st.session_state.get('generated_result'):
     exec(st.session_state.generated_result.replace('```', ''))
+for sentence in st.session_state.get('similar_sentences', []):
+    st.write(sentence)