Initial commit with the basic bones of the app and webpage up. Need to fill in the functions and routing next.
This commit is contained in:
67
application/routes.py
Normal file
67
application/routes.py
Normal file
@ -0,0 +1,67 @@
|
||||
"""
|
||||
Flask app that uses gpt4all to generate random song writing prompt.
|
||||
"""
|
||||
import os
|
||||
import string
|
||||
import random
|
||||
from flask import (
|
||||
render_template,
|
||||
session,
|
||||
)
|
||||
from gpt4all import GPT4All
|
||||
from datetime import datetime, timezone, timedelta
|
||||
from pathlib import Path
|
||||
from application import app
|
||||
|
||||
app.config.update(SECRET_KEY=os.urandom(24))
|
||||
app.permanent_session_lifetime = timedelta(minutes=30)
|
||||
|
||||
MODEL = GPT4All(
|
||||
model_name="orca-mini-3b-gguf2-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", "#"]
|
||||
# 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)
|
||||
|
||||
|
||||
@app.route("/")
|
||||
def main_prompt():
|
||||
"""
|
||||
Main function that loads the prompt
|
||||
"""
|
||||
return render_template("index.html", title="Home")
|
||||
|
||||
@app.route("/")
|
||||
def prompt_instrument():
|
||||
pass
|
||||
|
||||
@app.route("/")
|
||||
def prompt_key():
|
||||
pass
|
||||
|
||||
@app.route("/")
|
||||
def prompt_timesig():
|
||||
pass
|
||||
|
||||
@app.route("/")
|
||||
def prompt_all():
|
||||
pass
|
||||
|
||||
@app.route("/")
|
||||
def prompt_influence():
|
||||
pass
|
||||
response = MODEL.generate("The writing prompt is about the weather:", temp=0)
|
||||
session["list_response"] = response.splitlines()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
app.run(debug=True)
|
||||
Reference in New Issue
Block a user