Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Proposing some changes, including OpenAI API v1.0.0 compatiblity #3

Open
wants to merge 5 commits into
base: main
Choose a base branch
from
Open

Proposing some changes, including OpenAI API v1.0.0 compatiblity #3

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
6768b18
changed for compatibility with OpenAI API v1.0.0 and supporting env v…
isakcodes May 7, 2024
ea0340e
added assert to avoid an expensive request if output dir does not exist
isakcodes May 7, 2024
72f90ef
added saving of the response as well
isakcodes May 7, 2024
c4ed95f
added index to output filenames to not overwrite using --chat
isakcodes May 7, 2024
49eae3f
fix typo fstring
isakcodes May 21, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
24 changes: 14 additions & 10 deletions g2m.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
import os, requests, argparse, re
from fractions import Fraction
from midiutil.MidiFile import MIDIFile
import openai, mido
import mido
from openai import OpenAI

#settings
openaiKey = '<YOUR API KEY>'
system = 'You are MusicGPT, a music creation and completion chat bot that. When a user gives you a prompt,' \
' you return them a song showing the notes, durations, and times that they occur. Respond with just the music.' \
'\n\nNotation looks like this:\n(Note-duration-time in beats)\nC4-1/4-0, Eb4-1/8-2.5, D4-1/4-3, F4-1/4-3 etc.'
Expand All @@ -22,8 +22,11 @@
parser.add_argument('-l', '--load', help='load a MIDI file to be appended to your prompt')
parser.add_argument('-v', '--verbose', help='display GPT-4 output', action='store_true')
parser.add_argument('-o', '--output', help='specify output directory (default: current)', default=path)
parser.add_argument('-a', '--auth', help='specify openai api key (edit this script file to set a default)', default=openaiKey)
parser.add_argument('-a', '--auth', help='specify openai api key (edit this script file to set a default)', default=os.getenv("OPENAI_API_KEY"))
args = parser.parse_args()
assert os.path.exists(args.output), "[!] The output directory does not exist. Please run $mkdir -p <output directory>"

client = OpenAI(api_key=args.auth)

#other vars n functions
notes = [['C'], ['Db', 'C#'], ['D'], ['Eb', 'D#'], ['E'], ['F'], ['Gb', 'F#'], ['G'], ['Ab', 'G#'], ['A'], ['Bb', 'A#'], ['B']]
Expand Down Expand Up @@ -70,15 +73,16 @@ def midiToStr(mPath):
history = [{'role': 'system', 'content': system}, {'role': 'user', 'content': prompt}]

# main loop
while 1:
for i in range(100):
#openai request
print('[*] Making request to OpenAI API')
openai.api_key = args.auth
r = openai.ChatCompletion.create(
model = 'gpt-4',
messages = history
r = client.chat.completions.create(
model='gpt-4',
messages=history
)
response = r['choices'][0]['message']['content']
response = r.choices[0].message.content
with open(os.path.join(args.output, f'response{i}.txt'), 'w') as f:
f.write(response)
if args.verbose:
print('\n'+response+'\n')
history.append({'role': 'assistant', 'content': response})
Expand All @@ -97,7 +101,7 @@ def midiToStr(mPath):
for i in noteInfo:
pitch, dur, time = i
melody.addNote(0, 0, pitch, time, dur, 100)
with open(os.path.join(args.output, 'output.mid'), 'wb') as f:
with open(os.path.join(args.output, f'output{i}.mid'), 'wb') as f:
melody.writeFile(f)
print('[*] Wrote the MIDI file.')

Expand Down