|
|
|
import socket |
|
import WarBot |
|
|
|
import warnings |
|
warnings.filterwarnings("ignore") |
|
|
|
model,tokenizer,model_punct = WarBot.initialize() |
|
|
|
HOST = '10.0.0.125' |
|
PORT = 5000 |
|
|
|
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as server_socket: |
|
server_socket.bind((HOST, PORT)) |
|
server_socket.listen() |
|
print(f'Server is listening on port {PORT}') |
|
while True: |
|
conn, addr = server_socket.accept() |
|
with conn: |
|
print(f'Connected by {addr}') |
|
data = conn.recv(1024) |
|
received_string = data.decode() |
|
print(f'Received string from client: {received_string}') |
|
|
|
response = "" |
|
while not response: |
|
response = WarBot.get_response(received_string, model, tokenizer, model_punct, temperature=0.2) |
|
response_string = response |
|
|
|
conn.sendall(response_string.encode()) |
|
conn.close() |
