Upload iseamusic.py

iseamusic.py

@@ -0,0 +1,66 @@
+import gradio as gr
+import tensorflow as tf
+from magenta.models.melody_rnn import melody_rnn_model
+from magenta.models.melody_rnn import melody_rnn_sequence_generator
+from magenta.music import constants
+from magenta.music import midi_io
+from magenta.protobuf import generator_pb2
+from magenta.protobuf import music_pb2
+
+import os
+import tempfile
+
+# Cargar modelo de Melody RNN
+bundle = melody_rnn_sequence_generator.get_bundle_file('attention_rnn')
+generator_map = melody_rnn_sequence_generator.get_generator_map()
+generator = generator_map['attention_rnn'](checkpoint=None, bundle=bundle)
+
+# Establecer las configuraciones del generador
+generator.initialize()
+
+def generate_music(style, num_steps=128, primer_melody=[60]):
+    # Crear secuencia a partir de la melodía inicial
+    primer_sequence = music_pb2.NoteSequence()
+    for note in primer_melody:
+        note = primer_sequence.notes.add()
+        note.pitch = note
+        note.start_time = 0
+        note.end_time = 1
+        note.velocity = 80
+    
+    primer_sequence.total_time = 1
+
+    # Definir las configuraciones del generador
+    generator_options = generator_pb2.GeneratorOptions()
+    generator_options.generate_sections.add(
+        start_time=primer_sequence.total_time,
+        end_time=primer_sequence.total_time + num_steps / constants.DEFAULT_STEPS_PER_QUARTER)
+
+    # Generar la secuencia
+    sequence = generator.generate(primer_sequence, generator_options)
+    
+    # Guardar la secuencia en un archivo temporal MIDI
+    midi_file = os.path.join(tempfile.gettempdir(), 'output.mid')
+    midi_io.sequence_proto_to_midi_file(sequence, midi_file)
+    
+    return midi_file
+
+# Definir la interfaz de Gradio
+with gr.Blocks() as demo:
+    gr.Markdown("## Generador de Música con IA")
+    
+    with gr.Row():
+        style = gr.Dropdown(label="Estilo de Música", choices=["Classical", "Jazz", "Pop"], value="Classical")
+        num_steps = gr.Slider(label="Número de Pasos", minimum=64, maximum=512, step=64, value=128)
+        run_button = gr.Button("Generar Música")
+    
+    result = gr.Audio(label="Música Generada")
+    
+    def generate_and_play(style, num_steps):
+        midi_file = generate_music(style, num_steps)
+        return midi_file
+    
+    run_button.click(fn=generate_and_play, inputs=[style, num_steps], outputs=result)
+
+if __name__ == "__main__":
+    demo.launch()