Update app.py

app.py

@@ -26,31 +26,38 @@ model = genai.GenerativeModel(
 
 def generate_text(grade, num_paragraphs, sentences_per_paragraph, structure, topic):
     prompt = f"""
-        초등학교 {grade}학년 학생이 이해할 수 있는 수준의 설명문을 작성.
-        설명문의 주제는 {topic}이며, 주제를 중심으로 {structure} 구조에 따른 설명문을 작성할 것.
-        전체 문단 수는 {num_paragraphs} 문단 내외, 문단 당 문장 수는 {sentences_per_paragraph} 문장 내외.
-        설명문은 줄까만으로 작성하고 문단 구분을 꿀 할 것. 지정된 문단 수와 문단 당 문장 수를 꿀 지키기.
-        문단마다, 문단의 첫 문장 또는 마지막 문장이 문단의 중심 문장이 되도록 할 것. (중심 문장이라는 문단의 내용을 모두 포함할 수 있는 문장)
-        초등학교 {grade}학년 수준에 어렵은 단어와 한자어, 학습 도구어는 마지막에 어희 목록과 어희의 뜻을 초등학교 {grade}학년 수준에 맞게 유의언으로 또는 뜻을 풀어 설명하는 부분 따로 추가할 것.
-    """
-    
+초등학교 {grade}학년 학생이 이해할 수 있는 수준의 설명문을 작성.
+설명문의 주제는 {topic}이며, 주제를 중심으로 {structure} 구조에 따른 설명문을 작성할 것.
+전체 문단 수는 {num_paragraphs}개 문단 내외, 문단 당 문장 수는 {sentences_per_paragraph}개 문장 내외.
+설명문은 줄글로 작성하고 문단 구분을 명확히 할 것. 지정된 문단 수와 문단 당 문장 수를 준수하기.
+문단마다, 문단의 첫 문장 또는 마지막 문장이 문단의 중심 문장이 되도록 할 것. (중심 문장이란 문단의 내용을 모두 포함할 수 있는 문장)
+초등학교 {grade}학년 수준에 어려운 단어와 한자어, 학습 용어는 마지막에 어휘 목록과 어휘의 뜻을 초등학교 {grade}학년 수준에 맞게 유의어로 또는 뜻을 풀어 설명하는 부분을 따로 추가할 것.
+"""
+
     full_text = ""  # 전체 텍스트 저장 변수 초기화
-    
     try:
-        response = model.generate_content(prompt, stream=True)
-        for chunk in response:
-            full_text += chunk.text
-            # Markdown to HTML 변환
-            html_text = markdown.markdown(full_text, extensions=['tables'])
-            output_area.markdown(html_text, unsafe_allow_html=True)
-            time.sleep(0.1)  # 지연 추가 (필요에 따라 조절)
+        response = model.generate_content(prompt) #stream=True 삭제
+        full_text = response.text
 
     except Exception as e:
         st.error(f"에러 발생: {str(e)}")
         return ""
 
+    #Markdown to HTML 변환
+    html_text = markdown.markdown(full_text, extensions=['tables'])
+    
+    #출력 및 복사 기능 추가
+    output_area.markdown(html_text, unsafe_allow_html=True)
+    copy_button = st.button("출력 내용 복사")
+    if copy_button:
+        pyperclip.copy(full_text)
+        st.success("클립보드에 복사되었습니다!")
+
+
     return full_text
 
+
+
 # Streamlit 인터페이스 설정
 colored_header(
     label="📜 초등학생을 위한 읽기 자료 생성기",
@@ -62,13 +69,18 @@ add_vertical_space(1)
 
 with st.sidebar:
     st.header("옵션 설정")
-    grade = st.selectbox("학년", [f"초등학교 {i}학년" for i in range(1, 7)])
+    grade = st.selectbox("학년", 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)
+    sentences_per_paragraph = st.selectbox("문단 당 문장 수", [3, 4, 5, 6, 7, 8, 9, 10]) # 드롭다운 수정
     structure = st.selectbox("설명문 구조", ["정의와 예시", "비교와 대조", "분류", "분석", "인과", "순서"], index=0)
     topic = st.text_area("✏️ 주제 및 내용을 입력하세요 ", height=200)
 
 generate_button = st.button("읽기 자료 생성")
 
 # 출력 영역을 함수 외부에 정의
-output_area = st.empty()
+output_area = st.empty()
+
+
+if generate_button:
+    sentences_per_paragraph_range = str(sentences_per_paragraph) # 문장 수 변수 타입 변경
+    generate_text(grade, num_paragraphs, sentences_per_paragraph_range, structure, topic)