Update api.py

api.py

@@ -202,6 +202,89 @@ def generate_text_typical(model, prompt, max_len=100, max_gen=98,
             yield decoded
             break
 
+def is_valid_response(response):
+    if len(response.strip()) < 2:
+        return False
+    if re.search(r'[ㄱ-ㅎㅏ-ㅣ]{3,}', response):
+        return False
+    if len(response.split()) < 2:
+        return False
+    if response.count(' ') < 2:
+        return False
+    if any(tok in response.lower() for tok in ['hello', 'this', 'ㅋㅋ']):
+        return False
+    return True
+
+def extract_main_query(text):
+    sentences = re.split(r'[.?!]\s*', text)
+    sentences = [s.strip() for s in sentences if s.strip()]
+    if not sentences:
+        return text
+    last = sentences[-1]
+    last = re.sub(r'[^가-힣a-zA-Z0-9 ]', '', last)
+    particles = ['이', '가', '은', '는', '을', '를', '의', '에서', '에게', '한테', '보다']
+    for p in particles:
+        last = re.sub(rf'\b(\w+){p}\b', r'\1', last)
+    return last.strip()
+
+def get_wikipedia_summary(query):
+    cleaned_query = extract_main_query(query)
+    url = f"https://ko.wikipedia.org/api/rest_v1/page/summary/{cleaned_query}"
+    res = requests.get(url)
+    if res.status_code == 200:
+        return res.json().get("extract", "요약 정보를 찾을 수 없습니다.")
+    else:
+        return "위키백과에서 정보를 가져올 수 없습니다."
+
+def simple_intent_classifier(text):
+    text = text.lower()
+    greet_keywords = ["안녕", "반가워", "이름", "누구", "소개", "어디서 왔", "정체", "몇 살", "너 뭐야"]
+    info_keywords = ["설명", "정보", "무엇", "뭐야", "어디", "누구", "왜", "어떻게", "종류", "개념"]
+    math_keywords = ["더하기", "빼기", "곱하기", "나누기", "루트", "제곱", "+", "-", "*", "/", "=", "^", "√", "계산", "몇이야", "얼마야"]
+
+    if any(kw in text for kw in greet_keywords):
+        return "인사"
+    elif any(kw in text for kw in info_keywords):
+        return "정보질문"
+    elif any(kw in text for kw in math_keywords):
+        return "수학질문"
+    else:
+        return "일상대화"
+
+def parse_math_question(text):
+    text = text.replace("곱하기", "*").replace("더하기", "+").replace("빼기", "-").replace("나누기", "/").replace("제곱", "*2")
+    text = re.sub(r'루트\s(\d+)', r'math.sqrt(\1)', text)
+    try:
+        result = eval(text)
+        return f"정답은 {result}입니다."
+    except:
+        return "계산할 수 없는 수식이에요. 다시 한번 확인해 주세요!"
+
+# 전체 응답 함수
+def respond(input_text):
+    intent = simple_intent_classifier(input_text)
+
+    if "이름" in input_text:
+        return "제 이름은 InteractGPT입니다."
+
+    if "누구" in input_text:
+        return "저는 InteractGPT이라고 해요."
+
+    if intent == "수학질문":
+        return parse_math_question(input_text)
+
+    if intent == "인사":
+        return "반가워요! 무엇을 도와드릴까요?"
+        
+    if intent == "정보질문":
+        keyword = re.sub(r"(에 대해|에 대한|에 대해서)?\s*(설명해줘|알려줘|뭐야|개념|정의|정보)?", "", input_text).strip()
+        if not keyword:
+            return "어떤 주제에 대해 궁금한가요?"
+        summary = get_wikipedia_summary(keyword)
+        return f"{summary}\n다른 궁금한 점 있으신가요?"
+
+    return generate_text_typical(input_text)
+
 async def async_generator_wrapper(prompt: str):
     gen = generate_text_typical(model, prompt)
     for text_piece in gen: