Update app.py

app.py

@@ -81,6 +81,8 @@ 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("평가 생성"):
@@ -101,10 +103,13 @@ if st.button("평가 생성"):
             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
+    # # 문장을 라디오 버튼으로 표시 및 선택된 문장 저장
+    # 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('문장 생성중...'):
@@ -112,14 +117,19 @@ if st.button("평가 생성"):
 #         for sentence in st.session_state.similar_sentences:
 #             st.write(sentence)
 
-# 유사한 문장 생성 로직 분리
-if st.session_state.get('selected_sentence'):
-    if st.button("유사한 문구 생성"):
-        with st.spinner('문장 생성중...'):
-            st.session_state.similar_sentences = generate_similar_sentences(st.session_state.selected_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.similar_sentences:
+for sentence in st.session_state.get('similar_sentences', []):
     st.write(sentence)