doctorvivin's picture
Update app.py
8d3bd7f
from matplotlib.pyplot import title
from yaml import Mark
from music import music_gen #Funtion to generate music based on ABC files
import gradio as gr
import os
from MC.markov_chain import main_markov #Function to generate music based on midi files
keysignature = ["C","G","D","A","No selection"]
difficulty = ["beginner","intermediate","expert"]
timesignature = ['3/4','4/4','2/2','2/4']
# output = gr.Gallery() if GlobalUIGallery else "image"
# interface = gr.Interface(fn = music_gen,
# inputs=[gr.Radio(difficulty,label="Difficulty"),
# gr.Radio(timesignature,label="Time Signature"),
# gr.Dropdown(keysignature,label="Key Signature")],
# outputs = [gr.Gallery(label="Sheet Music"),gr.Audio(label="Audio")],
# title="Sheet Music Generation for Sight-Reading",
# description="TO be added")
# interface.launch(inline=False)
with gr.Blocks() as demo:
with gr.Tabs():
with gr.TabItem("MusicGen™ Advanced (v2)"):
gr.Markdown("ABC Model")
with gr.Row():
with gr.Column():
user_name_input = gr.Textbox(placeholder="Only Melody supported for now!!!", label="Describe the music you want to create...")
difficulty_input_abc = gr.Radio(difficulty,label="Music Complexity") #input
time_sig_input_abc = gr.Radio(timesignature,label="Time Signature") #input
key_sig_input_abc = gr.Dropdown(keysignature,label="Key Signature") #input
with gr.Row():
abc_button = gr.Button("Create AI Music with SpriFi MusicGen™!!") #Submit
with gr.Column():
output_audio_abc = gr.Audio(label="Generated music")
output_gallery_abc = gr.Gallery(label="Music Note Sheet")
with gr.TabItem("MusicGen™ (v1)"):
gr.Markdown("MIDI Model")
with gr.Row():
with gr.Column():
# difficulty_input_midi = gr.Radio(difficulty,label="Difficulty") #input
time_sig_input_midi = gr.Radio(timesignature,label="Time Signature") #input
# key_sig_input_midi = gr.Dropdown(keysignature,label="Key Signature") #input
with gr.Row():
midi_button = gr.Button("Create Music!!") #Submit
with gr.Column():
output_audio_midi = gr.Audio(label="Generated music")
output_gallery_midi = gr.Gallery(label="Music Note Sheet")
abc_button.click(music_gen, inputs=[difficulty_input_abc,time_sig_input_abc,key_sig_input_abc], outputs=[output_gallery_abc,output_audio_abc])
midi_button.click(main_markov, inputs= time_sig_input_midi, outputs=[output_gallery_midi,output_audio_midi])
if __name__ == "__main__":
demo.launch()