Messaging bugfixes

This commit is contained in:
Tanner Collin 2022-10-03 13:49:00 -06:00
parent 4d231bece7
commit 52e50c3ca7

11
main.py
View File

@ -106,7 +106,7 @@ def fetch_protocoin():
def message_protovac(message): def message_protovac(message):
try: try:
logging.info('Sending to Protovac: %s', message) logging.info('Message to Protovac: %s', message)
cookies = secrets.character_ai_cookies cookies = secrets.character_ai_cookies
@ -154,7 +154,11 @@ def message_protovac(message):
timeout=30, timeout=30,
) )
r.raise_for_status() r.raise_for_status()
return json.loads(r.text.split('\n')[-2])['replies'][0]['text'] reply = json.loads(r.text.split('\n')[-2])['replies'][0]['text'].replace('\n\n', ' ')
logging.info('Message reply: %s', reply)
return reply
except BaseException as e: except BaseException as e:
logging.exception(e) logging.exception(e)
return 'Error' return 'Error'
@ -312,6 +316,7 @@ while True:
if character_ai_token: if character_ai_token:
stdscr.addstr(17, menupos+4, '[M]', curses.A_REVERSE if highlight_keys else 0) stdscr.addstr(17, menupos+4, '[M]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(17, menupos+8, 'Message') stdscr.addstr(17, menupos+8, 'Message')
stdscr.addstr(17, 4, 'NEW')
if wa_api_key: if wa_api_key:
stdscr.addstr(19, menupos+4, '[T]', curses.A_REVERSE if highlight_keys else 0) stdscr.addstr(19, menupos+4, '[T]', curses.A_REVERSE if highlight_keys else 0)
stdscr.addstr(19, menupos+8, 'Think') stdscr.addstr(19, menupos+8, 'Think')
@ -761,7 +766,7 @@ while True:
messages.extend(lines) messages.extend(lines)
stdscr.erase() stdscr.erase()
message_to_send = '' message_to_send = '_'
else: else:
if c < 127 and c > 31: if c < 127 and c > 31:
message_to_send = message_to_send[:-1] + chr(c) + '_' message_to_send = message_to_send[:-1] + chr(c) + '_'