103 lines
2.5 KiB
Python
103 lines
2.5 KiB
Python
|
from gpt4all import GPT4All
|
||
|
|
import requests
|
||
|
|
import random
|
||
|
|
import string
|
||
|
|
import asyncio
|
||
|
|
from mastodon import Mastodon as mstd
|
||
|
|
import access_keys
|
||
|
|
|
||
|
|
MS_CONFIG = mstd(
|
||
|
|
access_token=access_keys.ACCESS_TOKEN, api_base_url="https://botsin.space"
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
MODEL = GPT4All(
|
||
|
|
model_name="gpt4all-falcon-q4_0.gguf",
|
||
|
|
# model_path=(Path.home() / ".cache" / "gpt4all"),
|
||
|
|
allow_download=False,
|
||
|
|
)
|
||
|
|
|
||
|
|
TIME_SIGNATURES = ["2/4", "3/4", "4/4", "2/2", "6/8", "9/8", "12/8"]
|
||
|
|
|
||
|
|
# Need to decide between manually entering this list or using ascii_letters, below
|
||
|
|
KEYS = ["A", "B", "C", "D", "E", "F", "G"]
|
||
|
|
SIGN = ["b", "#"]
|
||
|
|
NONKEYS = ["B#", "Cb", "E#", "Fb"]
|
||
|
|
# Option 2
|
||
|
|
MINOR = string.ascii_letters[0:7]
|
||
|
|
MAJOR = string.ascii_letters[26:33]
|
||
|
|
# and then use this:
|
||
|
|
# output = random.choice(KEYS)+random.choice(SIGN)
|
||
|
|
|
||
|
|
|
||
|
|
async def gather_sig_bpm():
|
||
|
|
time_signature = random.choice(TIME_SIGNATURES)
|
||
|
|
bpm = random.randrange(50, 201)
|
||
|
|
return time_signature, bpm
|
||
|
|
|
||
|
|
|
||
|
|
async def gather_key():
|
||
|
|
key = random.choice(KEYS) + random.choice(SIGN)
|
||
|
|
while key in NONKEYS:
|
||
|
|
key = random.choice(KEYS) + random.choice(SIGN)
|
||
|
|
return key
|
||
|
|
return key
|
||
|
|
|
||
|
|
|
||
|
|
async def gather_theme():
|
||
|
|
random_word = str(requests.get("https://random-word-api.herokuapp.com/word").text)[
|
||
|
|
2:-2
|
||
|
|
]
|
||
|
|
theme = MODEL.generate(
|
||
|
|
f"Give me a writing prompt about {random_word}.",
|
||
|
|
temp=0.7,
|
||
|
|
callback=stop_on_token_callback,
|
||
|
|
)
|
||
|
|
theme = theme.splitlines()
|
||
|
|
if len(theme) > 1:
|
||
|
|
for x, line in enumerate(theme):
|
||
|
|
if line.startswith("Sure"):
|
||
|
|
del theme[x]
|
||
|
|
elif line.endswith("."):
|
||
|
|
theme = line
|
||
|
|
return theme
|
||
|
|
else:
|
||
|
|
return theme
|
||
|
|
|
||
|
|
|
||
|
|
def stop_on_token_callback(token_id, token_string):
|
||
|
|
"""
|
||
|
|
Function to limit return length of the
|
||
|
|
gpt4all response. Period indicates a sentence.
|
||
|
|
"""
|
||
|
|
if "." in token_string:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
async def main():
|
||
|
|
res = await asyncio.gather(gather_sig_bpm(), gather_key(), gather_theme())
|
||
|
|
time_signature = res[0][0]
|
||
|
|
bpm = res[0][1]
|
||
|
|
key = res[1]
|
||
|
|
theme = res[2]
|
||
|
|
response = MS_CONFIG.status_post(
|
||
|
|
f"""
|
||
|
|
Get inspired! Here's what I've come up with:
|
||
|
|
Time Signature: {time_signature}
|
||
|
|
Key: {key}
|
||
|
|
BPM: {bpm}
|
||
|
|
Theme/Prompt: {theme}
|
||
|
|
"""
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def create_post():
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
loop = asyncio.get_event_loop()
|
||
|
|
results = loop.run_until_complete(main())
|
||
|
|
loop.close()
|