Update app.py

app.py

@@ -5,7 +5,6 @@ import google.generativeai as genai
 import streamlit as st
 from streamlit_extras.colored_header import colored_header
 from streamlit_extras.add_vertical_space import add_vertical_space
-import streamlit.components.v1 as components
 
 # Google Gemini API 키 설정
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
@@ -20,11 +19,11 @@ generation_config = {
 }
 
 model = genai.GenerativeModel(
-    model_name="gemini-1.5-pro-002",  
+    model_name="gemini-1.5-pro-002",
     generation_config=generation_config,
 )
 
-# 프롬프트 설정 (변경 없음)
+# 프롬프트 설정
 SYSTEM_PROMPT = """
 위 주제에 대한 개념 기반 교육과정을 다음 구성 요소에 따라 상세하게 설계해 주세요. 각 구성 요소는 서로 연결되고 유기적으로 구성되어야 합니다. 구체적인 예시를 포함하여 설명하고, 학생들의 흥미와 참여를 유도할 수 있는 창의적이고 혁신적인 활동들을 제시해 주세요. 특히, 탐구 활동을 통해 학생들이 스스로 일반화를 도출하고 핵심 개념을 깊이 있게 이해할 수 있도록 설계해야 합니다.  스트랜드(Strand)는 단원의 소주제와 같으며, 각 스트랜드는 여러 차시의 수업으로 구성됩니다.  차시별 활동 내용까지 상세히 기술해 주세요.
 
@@ -75,25 +74,20 @@ def generate_curriculum(grade, achievement_standard):
         SYSTEM_PROMPT,
         f"**대상 학년:** 초등학교 {grade}학년",
         f"**성취기준 및 주요 내용:** {achievement_standard}",
-        "**개념기반 교육과정:**", 
+        "**개념기반 교육과정:**",
     ]
 
     full_text = ""
-    output_area = st.empty()  # 출력 영역 초기화
     try:
         response = model.generate_content(prompt, stream=True)
         for chunk in response:
             full_text += chunk.text
-            # Streamlit 업데이트 (실시간 출력)
-            output_area.text_area("개념기반 교육과정", value=full_text, height=600)  # 높이 조정
-            time.sleep(0.01)  # 지연 시간 조정 (필요에 따라)
+        return full_text
     except Exception as e:
         print(f"에러 발생: {str(e)}")
-        st.error(f"에러 발생: {str(e)}") # 에러 메시지 Streamlit에 표시
+        st.error(f"에러 발생: {str(e)}")
         return ""
 
-    return full_text
-
 
 # Streamlit 인터페이스
 colored_header(
@@ -106,48 +100,22 @@ add_vertical_space(1)  # 공간 추가
 
 # 입력 위젯 (사이드바 사용, 확장 가능)
 with st.sidebar.expander("입력 설정", expanded=True):  # 확장 가능 패널
-    grade_options = ["초등학교 1학년", "초등학교 2학년", "초등학교 3학년", "초등학교 4학년", "초등학교 5학년", "초등학교 6학년"]
+    grade_options = [
+        "초등학교 1학년", "초등학교 2학년", "초등학교 3학년", "초등학교 4학년",
+        "초등학교 5학년", "초등학교 6학년"
+    ]
     grade_selected = st.selectbox("학년 선택", grade_options, index=0)
     grade = grade_options.index(grade_selected) + 1
 
     achievement_standard = st.text_area("성취기준 및 주요 내용 입력", height=200)
-    
-    generate_button = st.button("교육과정 생성") # 버튼 변수 저장
 
-# 생성 버튼 클릭시 동작
-if generate_button and achievement_standard:
-    result = generate_curriculum(grade, achievement_standard)
-    st.session_state.generated_curriculum = result
-elif generate_button and not achievement_standard:
-    st.warning("성취기준 및 주요 내용을 입력하세요.")
+generate_button = st.button("교육과정 생성")
 
-# 출력 영역 (스트리밍 출력 및 스크롤 자동 조정)
-output_area = st.empty()
+# 생성 버튼 클릭시 동작
 if generate_button and achievement_standard:
-    with output_area.container():  # 컨테이너 사용
-        # 컴포넌트 생성 및 스크롤 조정 콜백 함수 정의
-        def scroll_to_bottom():
-            js_code = """
-            <script>
-            const textArea = parent.document.querySelector('#output-frame .stTextArea textarea');
-            if (textArea) {
-                textArea.scrollTop = textArea.scrollHeight;
-            }
-            </script>
-            """
-            components.html(js_code, height=0, width=0)
-
-        try:
-            response = model.generate_content(prompt, stream=True)
-            full_text = ""
-            for chunk in response:
-                full_text += chunk.text
-                st.text_area("개념기반 교육과정", value=full_text, height=600, key=f"output_text_area_{time.time()}")  # 고유 key 값 사용
-                scroll_to_bottom()
-                time.sleep(0.01)
-            st.session_state.generated_curriculum = full_text # 세션 상태 저장
-        except Exception as e:
-            print(f"에러 발생: {str(e)}")
-            st.error(f"에러 발생: {str(e)}")
+    with st.spinner("생성 중..."): # 로딩 표시 추가
+        result = generate_curriculum(grade, achievement_standard)
+        st.session_state.generated_curriculum = result
+        st.text_area("개념기반 교육과정", value=result, height=600)
 elif generate_button and not achievement_standard:
     st.warning("성취기준 및 주요 내용을 입력하세요.")