Update app.py

app.py

@@ -89,15 +89,15 @@ if st.button("평가 생성"):
             sentence = line.split('",')[0].lstrip('("').strip()
             sentences.append(sentence)
 
-    # 문장을 라디오 버튼으로 표시
-    selected_sentence = st.radio("문장을 선택하세요:", sentences)
-
-    # 유사한 문장 생성 버튼 추가
-    if st.button("유사한 문구 생성"):
-        with st.spinner('문장 생성중...'):
-            similar_sentences = generate_similar_sentences(selected_sentence)
-            for sentence in similar_sentences:
-                st.write(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 similar_sentences:
+            st.write(sentence)