File size: 967 Bytes
6e1eccf
ddf7ac7
 
 
17e09e4
 
 
ddf7ac7
 
17e09e4
ddf7ac7
 
 
 
 
 
 
 
 
 
 
 
 
 
6e1eccf
 
c7e44a5
ddf7ac7
 
 
6e1eccf
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
# WarServer - replace HOST with 'localhost' for local testing
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()