import os
import socket
import ssl
import time
import openai
import re
openai.api_key = "get your own!!!!!!"
server = "irc.sageru.org"
port = 6697
channel = "#jp"
botnick = "chatgpt2"
password = ""
irc = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
irc.connect((server, port))
context = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
context.check_hostname = False
context.verify_mode = ssl.CERT_NONE
irc_ssl = context.wrap_socket(irc, server_hostname=server)
irc_ssl.send(("USER " + botnick + " " + botnick + " " + botnick + " :I am a bot!\r\n").encode())
irc_ssl.send(("NICK " + botnick + "\r\n").encode())
while True:
data = irc_ssl.recv(4096).decode('utf-8')
print(data) # print all data received from server to console
if data.find("PING") != -1:
irc_ssl.send(("PONG " + data.split()[1] + "\r\n").encode())
elif data.find("422") != -1 or data.find("376") != -1: # if no MOTD or end of MOTD
irc_ssl.send(("JOIN " + channel + "\r\n").encode())
elif data.find("PRIVMSG") != -1:
match = re.search(r'^:(.*)!(.*)@(.*) PRIVMSG (.*) :(.*)', data)
if match:
user = match.group(1)
message = match.group(5)
if message.startswith(botnick + ":"):
prompt = message.split(botnick + ":")[1]
response = openai.Completion.create(
engine="text-davinci-003",
prompt=prompt,
max_tokens=2048,
n=1,
stop=None,
temperature=0.3,
)
print(response)
if response.choices and response.choices[0].text.strip():
text = response.choices[0].text.strip()
lines = (line.strip() for line in text.split("\n") if line.strip())
for i, line in enumerate(lines):
if i >= 10:
irc_ssl.send(f"PRIVMSG {channel} :\x0312Too many lines! Sorry sosu.\x0F\r\n".encode())
break
prefix = "\x0306[ChatGPT]\x0F " if i == 0 else ""
irc_ssl.send(f"PRIVMSG {channel} :{prefix}\x0312{line}\x0F\r\n".encode())
time.sleep(1)