대학교 AI 질의응답 챗봇 만들기 - 9. 챗봇 엔진 서버 테스트

0. 포스팅 목록 & GitHub

(깃허브 스타 버튼 눌러주시면 글 작성에 큰 도움이 됩니다!)

https://github.com/Seokii/Chatbot4Univ

GitHub - Seokii/Chatbot4Univ: 대학생을 위한 AI 질의응답 챗봇 만들기

https://seokii.tistory.com/146

[Project] 대학생을 위한 AI 질의응답 챗봇 만들기

 

이전 글에서 챗봇 엔진 서버 코드에 대한 설명을 했습니다.

이번에는 완성된 챗봇 엔진 서버 코드를 테스트해보도록 하겠습니다.

 

1. 챗봇 엔진 실행

아래 사진은 파이참에서 챗봇 엔진 프로그램인 chatbot.py를 실행시킨 모습입니다.

단어 사전 및 텍스트 전처리기 불러오기,

의도 파악 모델 불러오기,

엑셀 파일 불러오기,

임베딩 pt 파일 갱신 및 불러오기

의 과정이 모두 진행되며 챗봇 엔진이 정상 작동하는 것을 확인할 수 있습니다.

 

 

2. 챗봇 엔진 테스트

# /test/chatbot_client_test.py

import socket
import json

# 챗봇 엔진 서버 접속 정보
host = "127.0.0.1"
port = 5050

# 클라이언트 프로그램 시작
while True:
    print("질문 : ")
    query = input()
    if(query == "exit"):
        exit(0)
    print("-" * 40)

    # 챗봇 엔진 서버 연결
    mySocket = socket.socket()
    mySocket.connect((host, port))

    # 챗봇 엔진 질의 요청
    json_data = {
        'Query' : query,
        'BotType' : "TEST"
    }
    message = json.dumps(json_data)
    mySocket.send(message.encode())

    # 챗봇 엔진 답변 출력
    data = mySocket.recv(2048).decode()
    ret_data = json.loads(data)
    print("답변 : ")
    print(ret_data['Answer'])
    print("\n")

# 챗봇 엔진 서버 연결 소켓 닫기
mySocket.close()

챗봇 엔진 프로그램을 종료하지 않고, 위의 코드를 작성한 후 실행하도록 합니다.

질문을 입력하면 답변을 출력 받을 수 있습니다.

아래의 그림은 질문을 입력한 순간 챗봇 엔진에 입력받은 json데이터의 출력입니다.