Update app.py

app.py

@@ -1,10 +1,10 @@
 import os
 import google.generativeai as genai
-import time
 import gradio as gr
 
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
+# Set up the model
 generation_config = {
   "temperature": 1,
   "top_p": 0.95,
@@ -35,7 +35,7 @@ model = genai.GenerativeModel(model_name="gemini-1.5-pro",
                               generation_config=generation_config,
                               safety_settings=safety_settings)
 
-def generate_curriculum(prompt):
+def generate_curriculum(prompt, history=[]):
   prompt_parts = [
     "Concept-Based Curriculum Expert, If your answer is good, I'll give you a $10 tip.",
     "성취기준 성취 기준 및 수업 목표",
@@ -45,21 +45,14 @@ def generate_curriculum(prompt):
   ]
 
   response = model.generate_content(prompt_parts)
-  return response.text
-
-def chat_bot(message, history):
-  history.append((message, ""))
-  if message:
-    response = generate_curriculum(message)
-    for char in response:
-      history[-1] = (message, history[-1][1] + char)
-      yield history
-      time.sleep(0.05)  # Adjust the delay as needed
-    history[-1] = (message, response)
-  yield history
-
-iface = gr.ChatInterface(
-    fn=chat_bot,
+  
+  # 스트리밍 출력을 위한 처리
+  yield from response.iter_chunks()
+
+iface = gr.Interface(
+    fn=generate_curriculum,
+    inputs=gr.Textbox(lines=1, label="성취기준 입력"),
+    outputs=gr.Chatbot(label="개념 기반 교육과정"),
     title="개념 기반 교육과정 챗봇",
     description="성취기준을 입력하면 개념 기반 교육과정을 생성합니다."
 )