Update app.py

app.py

@@ -25,6 +25,7 @@ model = genai.GenerativeModel(
 )
 
 def generate_text(grade, num_paragraphs, sentences_per_paragraph, structure, topic):
+    global text_area
     prompt = f"""
         초등학교 {grade}학년 학생이 이해할 수 있는 수준의 설명문을 작성.
         설명문의 주제는 {topic}이며, 주제를 중심으로 {structure} 구조에 따른 설명문을 작성할 것.
@@ -42,6 +43,9 @@ def generate_text(grade, num_paragraphs, sentences_per_paragraph, structure, top
             text_area.text_area("읽기 자료", value=full_text, height=300)
             time.sleep(0.05)
 
+        # 생성된 텍스트를 세션 상태에 저장
+        st.session_state.generated_curriculum = full_text
+
     except Exception as e:
         st.error(f"에러 발생: {str(e)}")
 
@@ -56,7 +60,7 @@ add_vertical_space(1)
 
 with st.sidebar:
     st.header("옵션 설정")
-    grade = st.selectbox("학년", [str(i) for i in range(1, 7)], index=0)
+    grade = st.selectbox("학년", [f"초등학교 {i}학년" for i in range(1, 7)])
     num_paragraphs = st.number_input("문단 수", min_value=1, value=3)
     sentences_per_paragraph = st.selectbox("문단 당 문장 수", ["3-4", "5-6", "6-7", "7-8", "9-10"], index=0)
     structure = st.selectbox("설명문 구조", ["정의와 예시", "비교와 대조", "분류", "분석", "인과", "순서"], index=0)
@@ -68,10 +72,9 @@ if st.button("읽기 자료 생성"):
 # Streamlit UI에 대한 장소의 text_area 출력 설정
 text_area = st.empty()
 
-# 보상 버튼 추가 (이미 생성된 경우)
-if "generated_curriculum" in st.session_state and st.session_state.generated_curriculum:
-    html_text = markdown.markdown(st.session_state.generated_curriculum, extensions=['tables'])
-    st.markdown(html_text, unsafe_allow_html=True)
+# 출력 내용 복사 버튼 추가 (이미 생성된 경우)
+if "generated_text" in st.session_state and st.session_state.generated_curriculum:
+    st.text_area("생성된 읽기 자료", value=st.session_state.generated_curriculum, height=300, key="generated_text_area", disabled=True)
 
     if st.button("출력 내용 복사", key="copy_button"):
         pyperclip.copy(st.session_state.generated_curriculum)