Spaces:
Running
Running
| import os, random, re | |
| from fractions import Fraction | |
| from midiutil.MidiFile import MIDIFile | |
| import streamlit as st | |
| import mido, openai | |
| if 'path' not in st.session_state: | |
| st.session_state['path'] = os.path.realpath(os.path.dirname(__file__)) | |
| if 'sessionID' not in st.session_state: | |
| st.session_state['sessionID'] = random.randint(0,99999999) | |
| if 'history' not in st.session_state: | |
| st.session_state['history'] = [] | |
| if 'downloadable' not in st.session_state: | |
| st.session_state['downloadable'] = False | |
| notes = [['C'], ['Db', 'C#'], ['D'], ['Eb', 'D#'], ['E'], ['F'], ['Gb', 'F#'], ['G'], ['Ab', 'G#'], ['A'], ['Bb', 'A#'], ['B']] | |
| monster = r'(?<![A-Za-z\d])([A-G](?:#|b)?\d(?:-\d+(?:\/\d+)?(?:-\d+(?:\.\d+)?)?)+)(?![A-Za-z\d])' | |
| def noteToInt(n): | |
| oct = int(n[-1]) | |
| letter = n[:-1] | |
| id = 0 | |
| for ix, x in enumerate(notes): | |
| for y in x: | |
| if letter == y: | |
| id = ix | |
| return id+oct*12+12 | |
| def midiToStr(mPath): | |
| midIn = mido.MidiFile(os.path.expanduser(mPath)) | |
| ticks = midIn.ticks_per_beat | |
| midOut = [] | |
| globalT = 0 | |
| opens = {} | |
| for track in midIn.tracks: | |
| for msg in track: | |
| if msg.type == 'note_on' or msg.type == 'note_off': | |
| globalT += msg.time/ticks | |
| if msg.note in opens: | |
| noteTime = opens[msg.note] | |
| noteTime = int(noteTime) if noteTime.is_integer() else noteTime | |
| noteDur = str(Fraction((globalT-noteTime)/4)) | |
| noteDur = str(round((globalT-noteTime),3)) if len(noteDur)>=6 else noteDur | |
| midOut.append('-'.join([notes[msg.note%12][0]+str(msg.note//12-1), noteDur, str(round(noteTime,3))])) | |
| del opens[msg.note] | |
| if msg.type == 'note_on': | |
| opens[msg.note] = globalT | |
| return ', '.join(midOut) | |
| st.markdown('# GPT-4 2 Midi\n####AI Generated Polyphonic Music\n##### plus conversion tools for use with Chat-GPT\napp by [d3nt](https://github.com/d3n7/)') | |
| main, m2t, t2m = st.tabs(['GPT4-To-Midi', 'Midi-2-Text', 'Text-2-Midi']) | |
| with main: | |
| userPrompt = st.text_input('Prompt', 'Full piece of sad music with multiple parts. Plan out the structure beforehand, including chords, parts (soprano, alto, tenor, bass), meter, etc.') | |
| with st.expander('System Prompt'): | |
| sysPrompt = st.text_input('', '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.') | |
| openaikey = st.text_input('OpenAI API Key', type='password') | |
| modelV = st.selectbox('Model', ('GPT-4', 'GPT-3.5-Turbo')) | |
| col1, col2 = st.columns(2) | |
| with col1: | |
| newSession = st.checkbox('New Session', True) | |
| with col2: | |
| showOutput = st.checkbox('Show Output', True) | |
| uploadMidi = st.file_uploader('Upload a midi file (OPTIONAL)') | |
| col3, col4 = st.columns(2) | |
| with col3: | |
| if st.button('Ask GPT'): | |
| if userPrompt != '' and sysPrompt != '' and openaikey != '': | |
| if newSession: | |
| st.session_state['history'] = [{'role': 'system', 'content': sysPrompt}] | |
| prompt = userPrompt | |
| if uploadMidi: | |
| filename = ''.join(uploadMidi.name.split('.')[:-1])+str(st.session_state['sessionID'])+'.'+''.join(uploadMidi.name.split('.')[-1]) | |
| midiPath = os.path.join(st.session_state['path'], filename) | |
| with open(midiPath, 'wb') as f: | |
| f.write(uploadMidi.getbuffer()) | |
| prompt += '\n'+midiToStr(midiPath) | |
| os.remove(midiPath) | |
| st.session_state['history'].append({'role': 'user', 'content': prompt}) | |
| openai.api_key = openaikey | |
| with st.spinner('Talking to OpenAI...'): | |
| r = openai.ChatCompletion.create( | |
| model=modelV.lower(), | |
| messages=st.session_state['history'] | |
| ) | |
| response = r['choices'][0]['message']['content'] | |
| st.session_state['history'].append({'role': 'assistant', 'content': response}) | |
| noteInfo = [] | |
| for i in re.findall(monster, response): | |
| n = i.split('-') | |
| noteInfo.append([noteToInt(n[0]), float(Fraction(n[1])) * 4, float(n[2])]) #note, duration, time | |
| song = MIDIFile(1, deinterleave=False) | |
| for i in noteInfo: | |
| pitch, dur, time = i | |
| song.addNote(0, 0, pitch, time, dur, 100) | |
| with open(os.path.join(st.session_state['path'], 'output.mid'), 'wb') as f: | |
| song.writeFile(f) | |
| if not st.session_state['downloadable']: | |
| st.session_state['downloadable'] = True | |
| else: | |
| st.warning('Make sure OpenAI key, prompt, and system prompt are entered', icon='⚠️') | |
| with col4: | |
| if st.session_state['downloadable']: | |
| with open(os.path.join(st.session_state['path'], 'output.mid'), 'rb') as f: | |
| st.download_button('Download Midi', f, file_name='output.mid') | |
| if showOutput: | |
| with st.container(): | |
| for i in st.session_state['history']: | |
| st.text(i['role']+': '+i['content']+'\n') | |
| with m2t: | |
| inMidi = st.file_uploader('Input') | |
| if st.button('Convert', key='1'): | |
| if inMidi: | |
| filename = ''.join(inMidi.name.split('.')[:-1]) + str(st.session_state['sessionID']) + '.' + ''.join(inMidi.name.split('.')[-1]) | |
| midiPath = os.path.join(st.session_state['path'], filename) | |
| with open(midiPath, 'wb') as f: | |
| f.write(inMidi.getbuffer()) | |
| st.text_area('Output', midiToStr(midiPath)) | |
| os.remove(midiPath) | |
| with t2m: | |
| inText = st.text_input('Input') | |
| if st.button('Convert', key='2'): | |
| noteInfo = [] | |
| for i in re.findall(monster, inText): | |
| n = i.split('-') | |
| noteInfo.append([noteToInt(n[0]), float(Fraction(n[1])) * 4, float(n[2])]) | |
| song = MIDIFile(1, deinterleave=False) | |
| for i in noteInfo: | |
| pitch, dur, time = i | |
| song.addNote(0, 0, pitch, time, dur, 100) | |
| with open(os.path.join(st.session_state['path'], 't2m.mid'), 'wb') as f: | |
| song.writeFile(f) | |
| with open(os.path.join(st.session_state['path'], 't2m.mid'), 'rb') as f: | |
| st.download_button('Download Midi', f, file_name='output.mid') |