Update app.py

app.py

@@ -46,21 +46,19 @@ def generate_research_plan(question):
 
 7. 탐구를 위한 팁:"""
 
-  # 모델 응답 생성
-  response = model.generate_content([prompt])
-  response_text = response.text
-
-  # 텍스트를 청크 단위로 분할
-  chunks = response_text.split(" ")
-
-  # 빈 텍스트로 시작
-  output_text = ""
-  
-  # 청크를 하나씩 추가하며 실시간 출력
-  for chunk in chunks:
-    output_text += chunk + " "
-    yield output_text
-    time.sleep(0.05) 
+  # 초기 빈 텍스트 출력
+  full_text = ""
+  yield full_text  
+
+  try:
+    # 스트리밍 방식으로 응답 생성
+    response = model.generate_content([prompt], stream=True)
+    for chunk in response:
+      full_text += chunk.text
+      yield full_text
+      time.sleep(0.05)
+  except Exception as e:
+    yield f"에러 발생: {str(e)}"
 
 # Gradio 인터페이스 생성
 iface = gr.Interface(