Update app.py

app.py

@@ -84,35 +84,64 @@ if 'similar_sentences' not in st.session_state:
 if 'selected_index' not in st.session_state:
     st.session_state.selected_index = 0
 
+# # "평가 생성" 버튼 클릭 시의 동작
+# if st.button("평가 생성"):
+#     with st.spinner('답변 생성중...'):
+#         result = generate_annotated_text(achievement_standard) 
+#     # 결과 출력
+#     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)
+
+
+#     # 문장을 라디오 버튼으로 표시 및 선택된 문장 저장
+#     selected_index = st.radio("문장을 선택하세요:", range(len(sentences)), format_func=lambda x: sentences[x])
+#     st.session_state.selected_sentence = sentences[selected_index] if sentences else None
+
+    
+# # 유사한 문장 생성 버튼 추가
+# if st.button("유사한 문구 생성") and st.session_state.get('selected_sentence'):
+#     with st.spinner('문장 생성중...'):
+#         similar_sentences = generate_similar_sentences(st.session_state.selected_sentence)
+#         for sentence in st.session_state.similar_sentences:
+#             st.write(sentence)
+
 # "평가 생성" 버튼 클릭 시의 동작
 if st.button("평가 생성"):
     with st.spinner('답변 생성중...'):
         result = generate_annotated_text(achievement_standard) 
-    # 결과 출력
+        st.session_state.generated_result = result
     exec(result.replace('```', ''))
 
-    # annotated_text 결과에서 문장만 추출
+    # 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()  # "(" 다음부터 "," 전까지의 문자열을 추출
+            sentence = line[start_idx + 2:end_idx].strip()
             sentences.append(sentence)
 
+    if sentences:
+        st.session_state.selected_sentence = st.selectbox("문장을 선택하세요:", sentences)
 
-    # 문장을 라디오 버튼으로 표시 및 선택된 문장 저장
-    selected_index = st.radio("문장을 선택하세요:", range(len(sentences)), format_func=lambda x: sentences[x])
-    st.session_state.selected_sentence = sentences[selected_index] if sentences else None
-
-    
 # 유사한 문장 생성 버튼 추가
 if st.button("유사한 문구 생성") and st.session_state.get('selected_sentence'):
     with st.spinner('문장 생성중...'):
         similar_sentences = generate_similar_sentences(st.session_state.selected_sentence)
-        for sentence in st.session_state.similar_sentences:
-            st.write(sentence)
-
+        for generated_sentence in similar_sentences:
+            st.write(generated_sentence)
 
+# 생성된 결과 및 유사한 문장들 출력
+if st.session_state.get('generated_result'):
+    exec(st.session_state.generated_result.replace('```', ''))