Reflect changes to gradio_app.py

api_server.py

@@ -83,23 +83,110 @@ sys.stdout = _stdout_proxy
 
 class PersonaRequest(BaseModel):
     info: str
+    stream: bool = True
+
+
+PERSONA_STATUS_MESSAGES = [
+    "인물 정보 수집 중...",
+    "웹 검색을 통해 배경 조사 중...",
+    "금융 사고 방식 분석 중...",
+    "데이터 분석 접근법 평가 중...",
+    "답변 스타일 특성 파악 중...",
+    "핵심 투자 원칙 추출 중...",
+    "대표 어록 정리 중...",
+    "페르소나 프로필 구성 중...",
+    "최종 검증 및 저장 준비 중...",
+]
+
+
+def _build_persona_payload(persona) -> dict:
+    return {
+        "type": "result",
+        "name": persona.name,
+        "full_name": persona.full_name,
+        "summary": persona.summary,
+        "financial_mindset": persona.financial_mindset,
+        "data_analysis_approach": persona.data_analysis_approach,
+        "response_style": persona.response_style,
+        "key_principles": persona.key_principles,
+        "famous_quotes": getattr(persona, "famous_quotes", None),
+    }
 
 
 @app.post("/persona/")
 async def create_persona(request: PersonaRequest):
     info = (request.info or "").strip()
+    stream = request.stream
+
     if not info:
         return JSONResponse(status_code=400, content={"error": "info 필드가 비어 있습니다."})
 
-    try:
-        persona = make_persona(info)
-    except Exception as exc:
-        return JSONResponse(status_code=500, content={"error": str(exc)})
+    if not stream:
+        try:
+            persona = make_persona(info)
+        except Exception as exc:
+            return JSONResponse(status_code=500, content={"error": str(exc)})
+
+        if persona is None:
+            return JSONResponse(status_code=500, content={"error": "페르소나 생성에 실패했습니다."})
+
+        return JSONResponse(content=persona.model_dump())
+
+    def event_stream():
+        event_queue: Queue = Queue()
+
+        def status_sender():
+            import asyncio
+
+            async def send_status():
+                for i, message in enumerate(PERSONA_STATUS_MESSAGES[:-1]):  # 마지막 메시지는 완료 시점에 사용
+                    event_queue.put({"type": "status", "message": message})
+                    await asyncio.sleep(7)
+
+            # 비동기 이벤트 루프에서 실행
+            loop = asyncio.new_event_loop()
+            asyncio.set_event_loop(loop)
+            loop.run_until_complete(send_status())
+
+        def worker():
+            thread_id = threading.get_ident()
+            _stdout_proxy.register(thread_id, _QueueingStdoutTee(_stdout_proxy._target, event_queue))
+            try:
+                # status 메시지 전송 스레드 시작
+                status_thread = Thread(target=status_sender, daemon=True)
+                status_thread.start()
+
+                persona = make_persona(info)
+
+                if persona is None:
+                    event_queue.put({"type": "error", "message": "페르소나 생성에 실패했습니다."})
+                else:
+                    event_queue.put(_build_persona_payload(persona))
+            except Exception as exc:
+                event_queue.put({"type": "error", "message": str(exc)})
+            finally:
+                _stdout_proxy.unregister(thread_id)
+                event_queue.put({"type": "done"})
+
+        yield _sse({"type": "status", "message": "페르소나 생성 준비 중..."})
+        Thread(target=worker, daemon=True).start()
 
-    if persona is None:
-        return JSONResponse(status_code=500, content={"error": "페르소나 생성에 실패했습니다."})
+        done = False
+        while not done:
+            try:
+                event = event_queue.get(timeout=0.2)
+            except Empty:
+                continue
+            yield _sse(jsonable_encoder(event))
+            if event.get("type") == "done":
+                done = True
 
-    return JSONResponse(content=persona.model_dump())
+    headers = {
+        "Cache-Control": "no-cache",
+        "Connection": "keep-alive",
+        "X-Accel-Buffering": "no",
+    }
+    return StreamingResponse(event_stream(), media_type="text/event-stream", headers=headers)
 
 class QueryRequest(BaseModel):
     query: str

llm/generator.py

@@ -62,7 +62,7 @@ def generate_news_info(client, user_query, intent):
 class Persona(BaseModel):
     name: str                        # 인물 이름
     full_name: str                   # 인물 이름
-    background: str                  # 인물 배경 (경력, 주요 업적 등)
+    summary: str                     # 인물 요약 (간단한 소개)
     financial_mindset: str           # 금융 사고 방식
     data_analysis_approach: str      # 데이터 분석 방식
     response_style: str              # 질문에 대한 답변 스타일
@@ -76,6 +76,7 @@ def generate_persona(client, user_query):
         "- full_name: 인물의 원문/정식 전체 이름.\n"
         "- name: 사용자가 이해하기 쉬운 표시 이름(한국어 통용명 우선, 없으면 간결한 영어).\n"
         "- 괄호 별칭/원문 병기는 full_name에만 포함하고 name에는 넣지 말 것.\n"
+        "- summary: 인물의 간단한 소개 요약 (2-3문장 정도로 간결하게).\n"
         "나머지 필드는 사실 기반으로 충실히 작성하시오."
     )
 
@@ -123,7 +124,7 @@ def build_full_prompt(user_query, context, intent, persona=None):
 
 [선택된 페르소나]
 이름: {persona.name}
-배경: {persona.background}
+요약: {persona.summary}
 금융 사고방식: {persona.financial_mindset}
 데이터 분석 방식: {persona.data_analysis_approach}
 답변 스타일: {persona.response_style}

persona/make_persona.py

@@ -68,13 +68,13 @@ def save_persona_jsonl(persona, query, file_name=None):
 def print_persona(persona):
     print("\n[Persona 생성 결과]")
     print(f"- 이름: {persona.name}")
-    print(f"- 배경: {persona.background}")
+    print(f"- 요약: {persona.summary}")
     print(f"- 금융 사고 방식: {persona.financial_mindset}")
     print(f"- 데이터 분석 방식: {persona.data_analysis_approach}")
     print(f"- 답변 스타일: {persona.response_style}")
-    print(f"- 핵심 원칙: {', '.join(persona.key_principles)}")
+    print(f"- 핵심 원칙: {', '.join(persona.key_principles)}", flush=True)
     if getattr(persona, "famous_quotes", None):
-        print(f"- 어록: {', '.join(persona.famous_quotes)}")
+        print(f"- 어록: {', '.join(persona.famous_quotes)}", flush=True)
 
 
 def make_persona(info):

persona/persona_loader.py

@@ -13,6 +13,10 @@ def load_personas():
                 line = line.strip()
                 if line:
                     data = json.loads(line)
+                    # 호환성: background 필드가 있으면 summary로 매핑
+                    if isinstance(data, dict) and "background" in data and "summary" not in data:
+                        data["summary"] = data["background"]
+                        # background 필드는 제거하지 않고 유지 (호환성)
                     # full_name 없으면 name 사용하여 채움
                     if isinstance(data, dict) and not data.get("full_name"):
                         data["full_name"] = data.get("name", "")