s444354/AMUseBot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

path to file with characters moved to .env

Browse Source
This commit is contained in:
Adrian Charkiewicz 2023-06-17 12:42:32 +02:00
parent 90218b8a58
commit 88011b0e42
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ai_talks/AMUseBotBackend/src/DP/dp.py
View File

@ -7,6 +7,9 @@ import AMUseBotBackend.consts as c
import json import json
import streamlit as st import streamlit as st
from dotenv import load_dotenv
load_dotenv()
CHARACTERS = os.getenv('CHARACTERS')
class DP: class DP:
@ -14,12 +17,12 @@ class DP:
def __init__(self, dst: DST, llm_rephrasing=False, character='default'): #TODO: a way to set llm_rephrasing status and a character def __init__(self, dst: DST, llm_rephrasing=False, character='default'): #TODO: a way to set llm_rephrasing status and a character
self.dst_module = dst self.dst_module = dst
self.llm_rephrasing = llm_rephrasing self.llm_rephrasing = llm_rephrasing
with open('ai_talks/AMUseBotBackend/utils/characters_dict.json') as f: with open(CHARACTERS) as f:
characters_dict = json.load(f) characters_dict = json.load(f)
self.character = characters_dict[character] self.character = characters_dict[character]
def change_character(self, character): def change_character(self, character):
with open('ai_talks/AMUseBotBackend/utils/characters_dict.json') as f: with open(CHARACTERS) as f:
characters_dict = json.load(f) characters_dict = json.load(f)
self.character = characters_dict[character] self.character = characters_dict[character]
if self.character == 'default': if self.character == 'default':