Add message Protovac

This commit is contained in:
Tanner Collin 2022-10-02 19:07:30 -06:00
parent 2eb9a62c91
commit db24789f5c
2 changed files with 155 additions and 11 deletions

156
main.py
View File

@ -18,6 +18,8 @@ import pytz
import re
import os
import time
import json
import textwrap
from datetime import datetime
try:
@ -26,6 +28,12 @@ try:
except:
wa_api_key = None
try:
import secrets
character_ai_token = secrets.character_ai_token
except:
character_ai_token = None
KEY_ESCAPE = 27
KEY_ENTER = 10
KEY_SPACE = 32
@ -96,6 +104,62 @@ def fetch_protocoin():
logging.exception(e)
return 'Error'
def message_protovac(message):
try:
logging.info('Sending to Protovac: %s', message)
cookies = secrets.character_ai_cookies
headers = {
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; rv:91.0) Gecko/20100101 Firefox/91.0',
'Accept': '*/*',
'Accept-Language': 'en-US,en;q=0.5',
'Referer': 'https://beta.character.ai/chat?char=u77cm99PcKBRaczQcJEKLqD99JrT0wxK-RAOKHgFEYo',
'Authorization': 'Token ' + character_ai_token,
'Origin': 'https://beta.character.ai',
'DNT': '1',
'Connection': 'keep-alive',
'Sec-Fetch-Dest': 'empty',
'Sec-Fetch-Mode': 'cors',
'Sec-Fetch-Site': 'same-origin',
}
json_data = {
'history_external_id': 'Mfdn894zdyDsWvQ3XVZurjW_CPE5SNmEsjPTUNR_nC8',
'character_external_id': 'u77cm99PcKBRaczQcJEKLqD99JrT0wxK-RAOKHgFEYo',
'text': message,
'tgt': 'internal_id:61666:c5e98052-6704-4725-838a-7432ab01fa5b',
'ranking_method': 'random',
'faux_chat': False,
'staging': False,
'model_server_address': None,
'override_prefix': None,
'override_rank': None,
'rank_candidates': None,
'filter_candidates': None,
'enable_tti': None,
'initial_timeout': None,
'insert_beginning': None,
'translate_candidates': None,
'stream_every_n_steps': 16,
'chunks_to_pad': 1,
'is_proactive': False,
}
r = requests.post(
'https://beta.character.ai/chat/streaming/',
cookies=cookies,
headers=headers,
json=json_data,
timeout=20,
)
r.raise_for_status()
return json.loads(r.text.split('\n')[-2])['replies'][0]['text']
except BaseException as e:
logging.exception(e)
return 'Error'
if wa_api_key:
import wolframalpha
wa_client = wolframalpha.Client(wa_api_key)
@ -187,6 +251,8 @@ curses.curs_set(0)
highlight_keys = False
highlight_debounce = time.time()
sign_to_send = ''
messages = ['']*15
message_to_send = ''
think_to_send = ''
think_result = ''
stats = {}
@ -203,7 +269,7 @@ last_key = time.time()
def ratelimit_key():
global last_key
if think_to_send or sign_to_send or time.time() > last_key + 1:
if think_to_send or sign_to_send or message_to_send or time.time() > last_key + 1:
last_key = time.time()
return False
else:
@ -225,16 +291,19 @@ while True:
stdscr.addstr(7, 1, ' UNIVERSAL COMPUTER')
stdscr.addstr(8, 1, '')
menupos = 5
stdscr.addstr(9, menupos+4, '[I]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(9, menupos+8, 'Info')
stdscr.addstr(11, menupos+4, '[S]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(11, menupos+8, 'Stats')
stdscr.addstr(13, menupos+4, '[N]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(13, menupos+8, 'Sign')
stdscr.addstr(15, menupos+4, '[C]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(15, menupos+8, 'Classes')
stdscr.addstr(17, menupos+4, '[P]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(17, menupos+8, 'Protocoin')
stdscr.addstr(7, menupos+4, '[I]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(7, menupos+8, 'Info')
stdscr.addstr(9, menupos+4, '[S]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(9, menupos+8, 'Stats')
stdscr.addstr(11, menupos+4, '[N]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(11, menupos+8, 'Sign')
stdscr.addstr(13, menupos+4, '[C]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(13, menupos+8, 'Classes')
stdscr.addstr(15, menupos+4, '[P]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(15, menupos+8, 'Protocoin')
if character_ai_token:
stdscr.addstr(17, menupos+4, '[M]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(17, menupos+8, 'Message')
if wa_api_key:
stdscr.addstr(19, menupos+4, '[T]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(19, menupos+8, 'Think')
@ -417,6 +486,26 @@ while True:
stdscr.addstr(8, 4, '[E] Edit message', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(23, 1, '[B] Back', curses.A_REVERSE if highlight_keys else 0)
stdscr.clrtoeol()
stdscr.refresh()
elif current_screen == 'message':
stdscr.addstr(0, 1, 'PROTOVAC UNIVERSAL COMPUTER')
stdscr.addstr(2, 1, 'Message Protovac')
stdscr.addstr(3, 1, '===============')
stdscr.addstr(5, 1, 'Send a message to Protovac, the universal computer.')
offset = 7
for num, line in enumerate(messages[-13:]):
stdscr.addstr(num + offset, 1, line)
if message_to_send:
stdscr.addstr(21, 21, message_to_send)
stdscr.clrtoeol()
stdscr.addstr(23, 1, '[RETURN] Send [ESC] Cancel')
else:
stdscr.addstr(21, 21, '[E] Edit message', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(23, 1, '[B] Back', curses.A_REVERSE if highlight_keys else 0)
stdscr.clrtoeol()
stdscr.refresh()
elif current_screen == 'think':
@ -518,6 +607,9 @@ while True:
current_screen = 'sign'
elif button == 'c':
current_screen = 'classes'
elif button == 'm':
current_screen = 'message'
messages = ['']*15
elif button == 't' and wa_api_key:
current_screen = 'think'
elif button == 'd':
@ -627,6 +719,48 @@ while True:
sign_to_send = '_'
else:
try_highlight()
elif current_screen == 'message':
if message_to_send:
if c == curses.KEY_BACKSPACE:
message_to_send = message_to_send[:-2] + '_'
elif c == KEY_ESCAPE:
message_to_send = ''
stdscr.erase()
elif c == KEY_ENTER:
if len(message_to_send) > 1:
stdscr.addstr(22, 21, 'Sending...')
stdscr.refresh()
message_to_send = message_to_send[:-1]
lines = textwrap.wrap(
message_to_send,
width=80,
initial_indent=' '*20,
subsequent_indent=' '*20,
)
messages.append('')
messages.extend(lines)
reply = message_protovac(message_to_send)
lines = textwrap.wrap(
reply,
width=60,
)
messages.append('')
messages.extend(lines)
stdscr.erase()
message_to_send = ''
else:
if c < 127 and c > 31:
message_to_send = message_to_send[:-1] + chr(c) + '_'
elif button == 'b' or c == KEY_ESCAPE:
current_screen = 'home'
elif button == 'e':
message_to_send = '_'
else:
try_highlight()
elif current_screen == 'think':
if think_to_send:
if c == curses.KEY_BACKSPACE:

View File

@ -1 +1,11 @@
wa_api_key = ''
# Get these from your network inspector after sending a message to character.ai
character_ai_cookies = {
'_legacy_auth0.whatever.is.authenticated': 'true',
'auth0.whatever.is.authenticated': 'true',
'messages': '',
'csrftoken': '',
'sessionid': '',
}
character_ai_token = ''