Update app.py

app.py

@@ -84,52 +84,77 @@ 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
+
+#     #디버깅
+#     st.session_state.selected_index = st.radio("문장을 선택하세요:", range(len(sentences)), format_func=lambda x: sentences[x], key="selected_index")
+    
+# # # 유사한 문장 생성 버튼 추가
+# # 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("유사한 문구 생성"):
+#     if st.session_state.get('generated_result'):
+#         # annotated_text 결과에서 다시 문장 추출
+#         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:
+#             selected_sentence = sentences[st.session_state.selected_index]
+#             with st.spinner('문장 생성중...'):
+#                 st.session_state.similar_sentences = generate_similar_sentences(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 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()  # "(" 다음부터 "," 전까지의 문자열을 추출
-            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
-
-    #디버깅
-    st.session_state.selected_index = st.radio("문장을 선택하세요:", range(len(sentences)), format_func=lambda x: sentences[x], key="selected_index")
-    
-# # 유사한 문장 생성 버튼 추가
-# 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("유사한 문구 생성"):
-    if st.session_state.get('generated_result'):
-        # annotated_text 결과에서 다시 문장 추출
-        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:
-            selected_sentence = sentences[st.session_state.selected_index]
-            with st.spinner('문장 생성중...'):
-                st.session_state.similar_sentences = generate_similar_sentences(selected_sentence)
+            sentence = line[start_idx + 2:end_idx].strip()
+            # 각 문장 옆에 버튼 추가
+            if st.button(f"유사한 문구 생성 - '{sentence[:30]}...'", key=sentence):
+                with st.spinner('문장 생성중...'):
+                    similar_sentences = generate_similar_sentences(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('```', ''))
-for sentence in st.session_state.get('similar_sentences', []):
-    st.write(sentence)
+    exec(st.session_state.generated_result.replace('```', ''))