Update app.py

app.py

@@ -3,6 +3,8 @@ import time
 
 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
 
 # Google Gemini API 키 설정
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
@@ -93,28 +95,46 @@ def generate_curriculum(grade, achievement_standard):
 
 
 # Streamlit 인터페이스
-st.title("개념기반 교육과정 설계 도우미")
-st.write("초등학교 교육과정에 맞는 개념기반 교육과정을 생성합니다.")
+colored_header(
+    label="개념기반 교육과정 설계 도우미",
+    description="초등학교 교육과정에 맞는 개념기반 교육과정을 생성합니다.",
+    color_name="blue-70",  # 색상 변경 가능
+)
+
+add_vertical_space(1)  # 공간 추가
 
-# 입력 위젯 (사이드바 사용)
-with st.sidebar:
+# 입력 위젯 (사이드바 사용, 확장 가능)
+with st.sidebar.expander("입력 설정", expanded=True):  # 확장 가능 패널
     grade_options = ["초등학교 1학년", "초등학교 2학년", "초등학교 3학년", "초등학교 4학년", "초등학교 5학년", "초등학교 6학년"]
     grade_selected = st.selectbox("학년 선택", grade_options, index=0)
-    grade = grade_options.index(grade_selected) + 1 # 선택된 학년 문자열을 숫자로 변환
+    grade = grade_options.index(grade_selected) + 1
 
     achievement_standard = st.text_area("성취기준 및 주요 내용 입력", height=200)
+    
+    generate_button = st.button("교육과정 생성") # 버튼 변수 저장
 
-    if st.button("교육과정 생성"):
-        if achievement_standard:  # 입력 확인
-            result = generate_curriculum(grade, achievement_standard)
-            st.session_state.generated_curriculum = result # 결과를 세션 상태에 저장
-        else:
-            st.warning("성취기준 및 주요 내용을 입력하세요.")
+# 생성 버튼 클릭시 동작 (if문 밖으로 이동)
+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("성취기준 및 주요 내용을 입력하세요.")
 
 
-# 출력 영역 (복사 버튼 포함)
+# 출력 영역 (탭 추가, 다운로드 버튼 추가)
 if "generated_curriculum" in st.session_state:
-    st.text_area("개념기반 교육과정", value=st.session_state.generated_curriculum, height=600)
-    st.button("결과 복사", on_click=lambda: st.experimental_set_query_params(copied=st.session_state.generated_curriculum))
-    if st.experimental_get_query_params().get("copied"):
-        st.success("결과가 복사되었습니다!")
+    tab1, tab2 = st.tabs(["결과 보기", "결과 다운로드"])
+
+    with tab1:
+        st.text_area("개념기반 교육과정", value=st.session_state.generated_curriculum, height=600)
+        st.button("결과 복사", on_click=lambda: st.experimental_set_query_params(copied=st.session_state.generated_curriculum))
+        if st.experimental_get_query_params().get("copied"):
+            st.success("결과가 복사되었습니다!")
+
+    with tab2:
+        st.download_button(
+            label="텍스트 파일 다운로드",
+            data=st.session_state.generated_curriculum,
+            file_name="generated_curriculum.txt",
+            mime="text/plain",
+        )