Update app.py

app.py

@@ -1,18 +1,20 @@
 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"])
 
 # 모델 설정
 generation_config = {
-    "temperature": 1.5,
-    "top_p": 0.8,
+    "temperature": 1,
+    "top_p": 0.95,
     "top_k": 40,
-    "max_output_tokens": 15000,
+    "max_output_tokens": 8192,
     "response_mime_type": "text/plain",
 }
 
@@ -49,7 +51,6 @@ SYSTEM_PROMPT = """
   - 학생 입장 유의점
   - 교사 입장 유의점
 7. 학생에게 제공할 활동 안내 자료
-
 * 활동 안내 자료는 초등학생이 이해할 수 있도록 최대한 구체적이고, 단계적, 명시적으로 안내해주세요.
 """
 
@@ -77,19 +78,34 @@ def generate_curriculum(topic):
         yield collected_text  # 실시간 출력을 위해 yield 사용
         time.sleep(0.03)  # 출력 속도 조절
 
-# Gradio 인터페이스 설정
-iface = gr.Interface(
-    fn=generate_curriculum,
-    inputs=gr.Textbox(lines=1, 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("출력 내용이 복사되었습니다!")