tanner
/
protovac
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add message Protovac

Browse Source
master
Tanner Collin 2 years ago
parent 2eb9a62c91
commit db24789f5c
2 changed files with 155 additions and 11 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 156
      main.py
  2. 10
      secrets.py.example

156
main.py
Unescape 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:

10
secrets.py.example
Unescape 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 = ''

Write Preview
Loading…
Cancel
Save