Update app.py

app.py

@@ -107,7 +107,47 @@ achievement_standard = st.text_input("성취기준 입력:", value=selected_stan
 #         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 결과에서 문장만 추출
+#     result_lines = result.split('\n')
+#     sentences = []
+#     for line in result_lines:
+#         start_idx = line.find('("')
+#         if start_idx != -1:
+#             end_idx = line.find('",', start_idx)
+#             sentence = line[start_idx + 2:end_idx].strip()
+#             sentences.append(sentence)
+
+#     if 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('문장 생성중...'):
+#         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)
+
 # 세션 상태 초기화
+if 'generated_result' not in st.session_state:
+    st.session_state.generated_result = None
 if 'selected_sentence' not in st.session_state:
     st.session_state.selected_sentence = None
 if 'similar_sentences' not in st.session_state:
@@ -119,28 +159,24 @@ if st.button("평가 생성"):
         result = generate_annotated_text(achievement_standard) 
         st.session_state.generated_result = result
         st.session_state.selected_sentence = None  # 새로운 평가 생성시 선택된 문장 초기화
+        st.session_state.similar_sentences = []  # 이전 유사한 문장들 초기화
     exec(result.replace('```', ''))
 
+# 셀렉트 박스 및 유사한 문장 생성 버튼 추가
+if st.session_state.generated_result:
     # annotated_text 결과에서 문장만 추출
-    result_lines = result.split('\n')
-    sentences = []
-    for line in result_lines:
-        start_idx = line.find('("')
-        if start_idx != -1:
-            end_idx = line.find('",', start_idx)
-            sentence = line[start_idx + 2:end_idx].strip()
-            sentences.append(sentence)
-
+    result_lines = st.session_state.generated_result.split('\n')
+    sentences = [line[start_idx + 2:line.find('",', start_idx)].strip() for line in result_lines if (start_idx := line.find('("')) != -1]
+    
     if 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('문장 생성중...'):
-        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)
+        selected_sentence = st.selectbox("문장을 선택하세요:", sentences, key="sentence_select")
+        st.session_state.selected_sentence = selected_sentence
+
+    # 유사한 문장 생성 버튼
+    if st.button("유사한 문구 생성") and st.session_state.selected_sentence:
+        with st.spinner('문장 생성중...'):
+            st.session_state.similar_sentences = generate_similar_sentences(st.session_state.selected_sentence)
+
+    # 유사한 문장들 출력
+    for sentence in st.session_state.similar_sentences:
+        st.write(sentence)