anatolykopyl/politics-mark
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Added some more models
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse Source
This commit is contained in:
Anatoly Kopyl
2022-04-03 19:41:03 +03:00
parent 9ba51c0fd1
commit c0d2b1e35b
2 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
create_models.py
View File

@@ -9,7 +9,10 @@ api_id = os.getenv('API_ID')
api_hash = os.getenv('API_HASH') api_hash = os.getenv('API_HASH')
chats = [ chats = [
'tg1337const', 'tg1337const',
'cyberboec' 'cyberboec',
'topor',
'workBrothers',
'anekdot_bb'
] ]
def cleanup(msg): def cleanup(msg):
@@ -35,7 +38,7 @@ with TelegramClient('bot', api_id, api_hash) as client:
data = data + cleanup(message.text) data = data + cleanup(message.text)
text_model = markovify.Text(data) text_model = markovify.Text(data)
# text_model.compile(inplace = True) text_model.compile(inplace = True)
model_json = text_model.to_json() model_json = text_model.to_json()
f = open(f"data/{chat}.json", "w") f = open(f"data/{chat}.json", "w")
f.write(model_json) f.write(model_json)

4
main.py
View File

@@ -7,6 +7,7 @@ from argparse import ArgumentParser
parser = ArgumentParser() parser = ArgumentParser()
parser.add_argument("-d", "--dry", action='store_true', help="Run without sending a message") parser.add_argument("-d", "--dry", action='store_true', help="Run without sending a message")
parser.add_argument("-m", "--model", help="Specify model file")
args = parser.parse_args() args = parser.parse_args()
load_dotenv() load_dotenv()
@@ -15,6 +16,7 @@ api_id = os.getenv('API_ID')
api_hash = os.getenv('API_HASH') api_hash = os.getenv('API_HASH')
entity = os.getenv('ENTITY') entity = os.getenv('ENTITY')
dry_run = args.dry dry_run = args.dry
filename = args.model
def create_message(model): def create_message(model):
message = '' message = ''
@@ -30,7 +32,9 @@ def create_message(model):
avalible_files = os.listdir('data') avalible_files = os.listdir('data')
avalible_files.remove('.gitkeep') avalible_files.remove('.gitkeep')
if (not filename):
filename = random.choice(avalible_files) filename = random.choice(avalible_files)
print(f'chosen {filename}') print(f'chosen {filename}')
with open(os.path.join('data', filename), 'r') as f: with open(os.path.join('data', filename), 'r') as f:
model_json = f.read() model_json = f.read()