Update api.py

api.py

@@ -288,10 +288,6 @@ class SimilarityMemory:
 # 테스트
 memory = SimilarityMemory()
 
-memory.add("어제는 기분이 별로였어")
-memory.add("영화 보러 갔다가 친구랑 싸웠어")
-memory.add("카페에서 공부했는데 집중이 잘 됐어")
-
 def merge_prompt_with_memory(prompt: str, memories: list):
     context = "\n".join(f"{mem}" for mem in memories)
     return f"{context} {prompt}"
@@ -395,34 +391,56 @@ def parse_math_question(text):
     except:
         return "계산할 수 없는 수식이에요. 다시 한번 확인해 주세요!"
 
+
 def respond(input_text):
+    # 1) 사용자 입력 기억에 저장 (원하면)
+    memory.add(input_text)
+
     intent = simple_intent_classifier(input_text)
 
     if "이름" in input_text:
-        return "제 이름은 Flexi입니다."
+        response = "제 이름은 Flexi입니다."
+        memory.add(response)  # 답변도 기억에 추가 가능
+        return response
 
     if "누구" in input_text:
-        return "저는 Flexi라고 해요."
+        response = "저는 Flexi라고 해요."
+        memory.add(response)
+        return response
 
     if intent == "수학질문":
-        return parse_math_question(input_text)
+        response = parse_math_question(input_text)
+        memory.add(response)
+        return response
 
     if intent == "정보질문":
         keyword = re.sub(r"(에 대해|에 대한|에 대해서)?\s*(설명해줘|알려줘|뭐야|개념|정의|정보)?", "", input_text).strip()
         if not keyword:
-            return "어떤 주제에 대해 궁금한가요?"
+            response = "어떤 주제에 대해 궁금한가요?"
+            memory.add(response)
+            return response
         summary = summarize_from_wikipedia(keyword)
-        return f"{summary}\n다른 궁금한 점 있으신가요?"
+        response = f"{summary}\n다른 궁금한 점 있으신가요?"
+        memory.add(response)
+        return response
 
-    # ✅ 기억 기반 병합 추가
+    # 기억에서 유사 문장 꺼내서 프롬프트 만들기
     related_memories = memory.retrieve(input_text, top_k=3)
     merged_prompt = merge_prompt_with_memory(input_text, related_memories)
 
+    # 모델로 응답 생성
     response = generate_text_sample(model, merged_prompt)
+
+    # 응답 검증, 안 맞으면 재생성
     if not is_valid_response(response) or mismatch_tone(input_text, response):
         response = generate_text_sample(model, merged_prompt)
+
+    # 최종 응답도 기억에 추가
+    memory.add(response)
+
     return response
 
+
 @app.get("/generate", response_class=PlainTextResponse)
 async def generate(request: Request):
     prompt = request.query_params.get("prompt", "안녕하세요")