Update app.py

app.py

@@ -1,8 +1,12 @@
 import os
+import time
 import streamlit as st
 import google.generativeai as genai
+from streamlit_extras.colored_header import colored_header
+from streamlit_extras.add_vertical_space import add_vertical_space
+import markdown
 
-# Gemini API 설정
+# Google Gemini API Key 설정
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
 # 모델 설정
@@ -15,51 +19,65 @@ generation_config = {
 }
 
 model = genai.GenerativeModel(
-    model_name="gemini-1.5-flash-002",
+    model_name="gemini-1.5-pro-002",
     generation_config=generation_config,
 )
 
-
-def generate_evaluation(context):
+def generate_evaluation_questions(learning_goal):
     """
-    학습 목표를 입력받아 서술형 평가 문항과 평가 기준을 생성하는 함수
+    Generates 2-3 descriptive evaluation questions and grading criteria for the given learning goal.
     """
-
-    prompt = f"""평가 전문가, 학습 목표를 입력하면 아래와 같은 서술형 평가 문항 3개와 각각의 문항에 대한 평가 기준(상, 중, 하)을 작성해줘. 문항과 평가 기준은 초등학교 교육과정에 맞게 작성하도록 해.
-    ## 서술형 평가 예시
-    input: {context}
-    output:
-    1. 평가 문항 1
-    - 상: ...
-    - 중: ...
-    - 하: ...
-    2. 평가 문항 2
-    - 상: ...
-    - 중: ...
-    - 하: ...
-    3. 평가 문항 3
-    - 상: ...
-    - 중: ...
-    - 하: ...
+    prompt = f"""교육평가 전문가로서 다음 학습 목표를 참고하여 서술형 평가 문항을 2~3개 작성하고, 각 평가 문항에 대한 평가기준을 상, 중, 하로 제시해줘.
+    
+    ## 학습 목표
+    * {learning_goal}
+    
+    ## 서술형 평가 문항 및 평가기준
     """
 
-    response = model.generate_content([prompt])
-    return response.text
+    full_text = ""  # 초기 빈 텍스트 출력
 
+    try:
+        response = model.generate_content([prompt], stream=True)
+        for chunk in response:
+            full_text += chunk.text
+            # Convert markdown to HTML for display
+            html_text = markdown.markdown(full_text, extensions=['tables'])
+            evaluation_output_area.markdown(html_text, unsafe_allow_html=True)
+            time.sleep(0.05)
 
-# Streamlit 앱 제목
-st.title("📝 서술형 평가 문항 생성기")
-st.write("학습 목표를 입력하면 관련된 서술형 평가 문항과 평가 기준(상/중/하)을 생성해줍니다.")
+    except Exception as e:
+        st.error(f"Error: {str(e)}")
+        return ""
 
-# 입력 텍스트 박스
-context = st.text_area(
-    "학습 목표를 자유롭게 입력하세요.", height=70
+    return full_text
+
+# Streamlit Interface
+colored_header(
+    label="서술형 평가 문항 생성",
+    description="학습 목표에 맞는 서술형 평가 문항과 평가 기준을 생성하세요!",
+    color_name="blue-70",
 )
 
-# 평가 문항 생성 버튼
-if st.button("평가 문항 생성"):
-    # 평가 문항 생성 함수 호출
-    evaluation = generate_evaluation(context)
+add_vertical_space(1)
+
+with st.sidebar.expander("입력 설정", expanded=True):
+    learning_goal = st.text_input("학습 목표")
+
+generate_evaluation_button = st.button("서술형 평가 문항 생성")
+
+# 출력 영역 정의
+evaluation_output_area = st.empty()
+
+if generate_evaluation_button:
+    evaluation_result = generate_evaluation_questions(learning_goal)
+    st.session_state.generated_evaluation = evaluation_result
 
-    # 생성된 평가 문항 출력
-    st.text_area("생성된 평가 문항과 평가 기준", evaluation, height=400)
+    # 복사 버튼 추가
+    if evaluation_result:
+        if st.button("평가 문항 복사"):
+            try:
+                st.experimental_set_query_params(copied_text=evaluation_result)
+                st.success("평가 문항이 복사되었습니다!")
+            except Exception as e:
+                st.error(f"복사 중 오류가 발생했습니다: {str(e)}")