Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| c90125944a | |||
| 4790def449 |
@ -40,8 +40,8 @@ async def gather_key():
|
||||
key = random.choice(KEYS) + random.choice(SIGN)
|
||||
while key in NONKEYS:
|
||||
key = random.choice(KEYS) + random.choice(SIGN)
|
||||
else:
|
||||
return key
|
||||
return key
|
||||
|
||||
|
||||
async def gather_theme():
|
||||
|
||||
85
script_without_mastodon.py
Normal file
85
script_without_mastodon.py
Normal file
@ -0,0 +1,85 @@
|
||||
from gpt4all import GPT4All
|
||||
import requests
|
||||
import random
|
||||
import string
|
||||
import asyncio
|
||||
|
||||
|
||||
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)
|
||||
else:
|
||||
return key
|
||||
# if key not in NONKEYS:
|
||||
# return key
|
||||
# break
|
||||
# elif key in NONKEYS:
|
||||
# pass
|
||||
|
||||
|
||||
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
|
||||
|
||||
|
||||
async def main():
|
||||
res = await asyncio.gather(gather_sig_bpm(), gather_key(), gather_theme())
|
||||
print(res)
|
||||
|
||||
|
||||
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
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
loop = asyncio.get_event_loop()
|
||||
results = loop.run_until_complete(main())
|
||||
loop.close()
|
||||
Reference in New Issue
Block a user