Update app.py

app.py

@@ -1,78 +1,38 @@
 import os
 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
+import markdown
+import pyperclip
 
 # Google Gemini API 설정
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
-# 스타일 설정
-st.markdown(
-    """
-    <style>
-    .stApp {
-        background-color: #f5f5f5;
-    }
-    .stTextInput>div>div>textarea {
-        font-size: 16px;
-        padding: 10px;
-        border-radius: 5px;
-    }
-    .stButton>button {
-        background-color: #007bff;
-        color: white;
-        padding: 10px 20px;
-        border: none;
-        border-radius: 5px;
-        cursor: pointer;
-    }
-    .stSelectbox>div>div>select {
-        font-size: 16px;
-        padding: 5px;
-        border-radius: 5px;
-    }
-    .stNumberInput>div>div>input {
-        font-size: 16px;
-        padding: 5px;
-        border-radius: 5px;
-    }
-    .reportview-container .main .block-container {
-        padding: 2rem 3rem;
-    }
-    .stTitle {  /* 타이틀 스타일 조정 */
-        font-size: 1.5rem; /* 원하는 크기로 조정 */
-        font-weight: bold;
-    }
-    </style>
-    """,
-    unsafe_allow_html=True,
+# 모델 설정
+generation_config = {
+    "temperature": 0.7,
+    "top_p": 0.85,
+    "top_k": 40,
+    "max_output_tokens": 15000,
+    "response_mime_type": "text/plain",
+}
+
+model = genai.GenerativeModel(
+    model_name="gemini-1.5-pro-002",
+    generation_config=generation_config,
 )
 
-
 def generate_text(grade, num_paragraphs, sentences_per_paragraph, structure, topic):
-    generation_config = {
-        "temperature": 0.7,
-        "top_p": 0.85,
-        "top_k": 40,
-        "max_output_tokens": 15000,
-        "response_mime_type": "text/plain",
-    }
-
-    model = genai.GenerativeModel(
-        model_name="gemini-1.5-pro-002",
-        generation_config=generation_config,
-    )
-
     prompt = f"""
         초등학교 {grade}학년 학생이 이해할 수 있는 수준의 설명문을 작성.
         설명문의 주제는 {topic}이며, 주제를 중심으로 {structure} 구조에 따라 설명문을 작성할 것.
         전체 문단 수는 {num_paragraphs} 문단 내외, 문단 당 문장 수는 {sentences_per_paragraph} 문장 내외.
         설명문은 줄글로 작성하고 문단 구분을 꼭 할 것. 지정된 문단 수와 문단 당 문장 수를 꼭 지킬 것.
         문단마다, 문단의 첫 문장 또는 마지막 문장이 문단의 중심 문장이 되도록 할 것. (중심 문장이란 문단의 내용을 모두 포함할 수 있는 문장)
-
         초등학교 {grade}학년 수준에 어려운 단어와 한자어, 학습 도구어는 마지막에 어휘 목록과 어휘의 뜻을 초등학교 {grade}학년 수준에 맞게 유의어나 뜻을 풀어 설명하는 부분 따로 추가할 것.
-        """
+    """
 
     full_text = ""
     text_area = st.empty()
@@ -87,13 +47,17 @@ def generate_text(grade, num_paragraphs, sentences_per_paragraph, structure, top
     except Exception as e:
         st.error(f"에러 발생: {str(e)}")
 
-
 # Streamlit 인터페이스 설정
-st.title("📖 초등학생을 위한 읽기 자료 생성기 📖") # 타이틀 유지, 스타일은 CSS로 조정
-st.write("주제를 입력하면 초등학생이 이해하기 쉬운 읽기 자료를 만들어줍니다.")
+colored_header(
+    label="📖 초등학생을 위한 읽기 자료 생성기",
+    description="주제를 입력하면 초등학생이 이해하기 쉬운 읽기 자료를 만들어줍니다.",
+    color_name="purple-70",
+)
+
+add_vertical_space(1)
 
 with st.sidebar:
-    st.header("⚙️ 옵션 설정 ⚙️")
+    st.header("옵션 설정")
     grade = st.selectbox("학년", [str(i) for i in range(1, 7)], index=0)
     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)
@@ -101,5 +65,14 @@ with st.sidebar:
     topic = st.text_area("✏️ 주제 및 내용을 입력하세요 ", height=200)
 
 
-if st.button("✨ 읽기 자료 생성 ✨"):
-    generate_text(grade, num_paragraphs, sentences_per_paragraph, structure, topic)
+if st.button("읽기 자료 생성"):
+    generate_text(grade, num_paragraphs, sentences_per_paragraph, structure, topic)
+
+# 복사 버튼 추가 (이미 생성된 경우)
+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 st.button("출력 내용 복사", key="copy_button"):
+        pyperclip.copy(st.session_state.generated_curriculum)
+        st.success("복사되었습니다!")