Update app.py

app.py

@@ -1,10 +1,12 @@
 import os
 import time
-
+import streamlit as st
 import google.generativeai as genai
-import gradio as gr
+from streamlit_extras.colored_header import colored_header
+from streamlit_extras.add_vertical_space import add_vertical_space
+import markdown
 
-# Google Gemini API 키 설정
+# Google Gemini API Key 설정
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
 # 모델 설정
@@ -21,43 +23,39 @@ model = genai.GenerativeModel(
     generation_config=generation_config,
 )
 
-# 프롬프트 설정
-SYSTEM_PROMPT = """
-당신은 교수학습 활동 설계 전문가, 교과교육 내용 전문가, 교육공학자입니다. 
-선생님들이 활동 컨셉 및 소개를 입력하면 활동을 자세하게 설명해주세요.. 
-다음은 활동 설계에 필요한 정보입니다.
-
-**활동 구성 요소:**
-1. 활동 컨셉 및 소개
-2. 활동 목표
-3. 활동 단계별 안내
-  - 1단계: 1단계 활동 안내
-  - 2단계: 2단계 활동 안내
-  - 3단계: 3단계 활동 안내
-  - 4단계: 4단계 활동 안내...
-4. 활동 준비물 및 기구
-5. 활동을 위한 자료
-  - 활동에 필요한 자료
-  - 추가 추천 자료
-6. 활동의 유의점
-  - 학생 입장 유의점
-  - 교사 입장 유의점
-7. 학생에게 제공할 활동 안내 자료
-* 활동 안내 자료는 초등학생이 이해할 수 있도록 최대한 구체적이고, 단계적, 명시적으로 안내해주세요.
-"""
-
-def generate_curriculum(achievement_standard):
+def generate_lesson_plan(achievement_standard):
     """
     활동 컨셉을 입력받아 활동을 생성합니다.
     아웃풋을 실시간으로 출력합니다.
-
     Args:
         achievement_standard (str): 활동 컨셉
-
     Returns:
         str: 생성된 활동
     """
 
+    SYSTEM_PROMPT = """
+    당신은 교수학습 활동 설계 전문가, 교과교육 내용 전문가, 교육공학자입니다.
+    선생님들이 활동 컨셉 및 소개를 입력하면 활동을 자세하게 설명해주세요.. 
+    다음은 활동 설계에 필요한 정보입니다.
+    **활동 구성 요소:**
+    1. 활동 컨셉 및 소개
+    2. 활동 목표
+    3. 활동 단계별 안내
+      - 1단계: 1단계 활동 안내
+      - 2단계: 2단계 활동 안내
+      - 3단계: 3단계 활동 안내
+      - 4단계: 4단계 활동 안내...
+    4. 활동 준비물 및 기구
+    5. 활동을 위한 자료
+      - 활동에 필요한 자료
+      - 추가 추천 자료
+    6. 활동의 유의점
+      - 학생 입장 유의점
+      - 교사 입장 유의점
+    7. 학생에게 제공할 활동 안내 자료
+    * 활동 안내 자료는 초등학생이 이해할 수 있도록 최대한 구체적이고, 단계적, 명시적으로 안내해주세요.
+    """
+
     prompt = [
         SYSTEM_PROMPT,
         f"**활동 컨셉:** {achievement_standard}",
@@ -72,17 +70,34 @@ def generate_curriculum(achievement_standard):
         yield collected_text  # 실시간 출력을 위해 yield 사용
         time.sleep(0.03)  # 출력 속도 조절
 
-# Gradio 인터페이스 설정
-iface = gr.Interface(
-    fn=generate_curriculum,
-    inputs=gr.Textbox(lines=3, label="활동 컨셉 및 소개 입력"),
-    outputs=gr.Textbox(lines=10, label="학습 활동"),
-    title="학습 활동 설계 도우미",
+# Streamlit Interface
+colored_header(
+    label="학습 활동 설계 도우미",
     description="선생님이 활동의 컨셉을 입력하시면 활동을 구체적으로 설계해줍니다.",
-    examples=[
-        ["국경, 지형, 자원, 인구정보, 국가 간 관계가 표시된 지도와 시나리오를 주고, 나라 간 국경선을 정하는 활동을 시뮬레이션 게임 모형을 활용해 하고 싶어."],
-    ]
+    color_name="red-70",
 )
 
-# 인터페이스 실행
-iface.launch()
+add_vertical_space(1)
+
+achievement_standard = st.text_area("활동 컨셉 및 소개 입력", height=100)
+
+generate_button = st.button("학습 활동 생성")
+
+# 출력 영역 정의
+output_area = st.empty()
+
+if generate_button and achievement_standard:
+    output_text = ""
+    output_area.markdown(output_text)
+    try:
+        for partial_output in generate_lesson_plan(achievement_standard):
+            output_text = partial_output
+            output_area.markdown(output_text)
+    except Exception as e:
+        st.error(f"오류가 발생했습니다: {str(e)}")
+
+    # 복사 버튼 추가
+    if output_text:
+        if st.button("출력 내용 복사"):
+            st.write(output_text)
+            st.success("출력 내용이 복사되었습니다!")