Update app.py

app.py

@@ -50,7 +50,40 @@ st.markdown(
 
 
 def generate_text(grade, num_paragraphs, sentences_per_paragraph, structure, topic):
-    # ... (생략) ...
+    generation_config = {
+        "temperature": 0.3,
+        "top_p": 0.95,
+        "top_k": 64,
+        "max_output_tokens": 15000,  # 맥스 토큰 설정 추가
+        "response_mime_type": "text/plain",
+    }
+
+    model = genai.GenerativeModel(
+        model_name="gemini-1.5-pro-exp-0827",
+        generation_config=generation_config,
+    )
+
+    # 프롬프트 생성
+    prompt = f"""
+        초등학교 {grade}학년 학생이 이해할 수 있는 수준의 설명문을 작성.
+        설명문의 주제는 {topic}이며, 주제를 중심으로 {structure} 구조에 따라 설명문을 작성할 것,
+        전체 문단 수는 {num_paragraphs} 문단 내외, 문단 당 문장 수는 {sentences_per_paragraph} 문장 내외,
+        설명문은 줄글로 작성하고 문단 구분을 꼭 할 것, 지정된 문단 수와 문단 당 문장 수를 꼭 지킬것. 
+        """
+
+    full_text = ""
+    text_area = st.empty()  # 텍스트 출력을 위한 빈 텍스트 에어리어 생성
+
+    try:
+        # 스트리밍 방식으로 응답 생성
+        response = model.generate_content(prompt, stream=True)
+        for chunk in response:
+            full_text += chunk.text
+            text_area.text_area("읽기 자료", value=full_text, height=300)  # 텍스트 에어리어 높이 조정
+            time.sleep(0.05)
+
+    except Exception as e:
+        st.error(f"에러 발생: {str(e)}")
 
 
 # Streamlit 인터페이스 설정