Update app.py

app.py

@@ -1,6 +1,6 @@
-#===================================================================
+#==========================================================
 # https://huggingface.co/spaces/asigalov61/Orpheus-Morpheus
-#===================================================================
+#==========================================================
 
 # -----------------------------
 # CONFIGURATION & GLOBALS
@@ -522,6 +522,16 @@ def generate_music_and_state(input_midi,
         
     return [final_composition] + outputs_flat
 
+# -----------------------------
+# MISC FUNCTIONS
+# -----------------------------
+def reset(final_composition=[]):
+    """Reset composition state."""
+    print_sep()
+    print('Reset composition...')
+    print_sep()
+    return []
+
 # -----------------------------
 # GRADIO INTERFACE SETUP
 # -----------------------------
@@ -571,24 +581,16 @@ with gr.Blocks() as orpheus_mosrpheus_app:
     gr.Markdown("## Upload seed MIDI or click 'Generate' button for random output")
     
     input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"])
-    input_midi.upload(reset, [final_composition, generated_batches, block_lines],
-                      [final_composition, generated_batches, block_lines])
+    input_midi.upload(reset, [final_composition],
+                      [final_composition])
 
     gr.Markdown("## Generation options")
-    prime_instruments = gr.Dropdown(label="Prime instruments (select up to 5)", choices=list(Patch2number.keys()),
-                                    multiselect=True, max_choices=5, type="value",
-                                    info="NOTE: Custom MIDI overrides prime instruments"
-                                   )
-
-    prime_instruments.input(reset, [final_composition, generated_batches, block_lines],
-                          [final_composition, generated_batches, block_lines])
     
-    num_prime_tokens = gr.Slider(16, 6656, value=6656, step=1, label="Number of prime tokens")
-    num_gen_tokens = gr.Slider(16, 1024, value=512, step=1, label="Number of tokens to generate")
+    num_prime_tokens = gr.Slider(0, 32, value=0, step=1, label="Number of prime tokens")
+    num_gen_tokens = gr.Slider(256, 1280, value=1280, step=1, label="Number of tokens to generate")
     requested_model = gr.Dropdown(label="Model to use",
                                   choices=list(models_dict.keys()),
                                   value=list(models_dict.keys())[0],
-                                  info="Use medium model when speed is important, use large models when quality is important"
                                  )
     model_temperature = gr.Slider(0.1, 1, value=0.9, step=0.01, label="Model temperature",
                                   info="Increase for more creative output, decrease for more repetitive output"
@@ -596,13 +598,11 @@ with gr.Blocks() as orpheus_mosrpheus_app:
     model_top_p = gr.Slider(0.1, 1.0, value=0.96, step=0.01, label="Model sampling top p value",
                             info="1 == Disabled"
                            )
-    add_drums = gr.Checkbox(value=False, label="Add drums")
-    add_outro = gr.Checkbox(value=False, label="Add an outro")
     
     generate_btn = gr.Button("Generate", variant="primary")
 
     gr.Markdown("## Batch Previews")
-    outputs = [final_composition, generated_batches, block_lines]
+    outputs = [final_composition]
     # Two outputs (audio and plot) for each batch
     for i in range(NUM_OUT_BATCHES):
         with gr.Tab(f"Batch # {i}"):
@@ -610,59 +610,25 @@ with gr.Blocks() as orpheus_mosrpheus_app:
             plot_output = gr.Plot(label=f"Batch # {i} MIDI Plot")
             midi_file = gr.File(label=f"Batch # {i} MIDI File")
             outputs.extend([audio_output, plot_output, midi_file])
-
-    requested_model.change(
-        fn=update_state_from_dropdown,
-        inputs=[requested_model, model_selector],
-        outputs=model_selector
-    )
             
     generate_btn.click(
         generate_music_and_state,
         [input_midi,
-         prime_instruments, 
          num_prime_tokens, 
          num_gen_tokens, 
          model_temperature, 
          model_top_p, 
-         add_drums, 
-         add_outro,
          final_composition, 
-         generated_batches, 
-         block_lines,
          model_selector
         ],
         outputs
     )
 
-    gr.Markdown("## Add/Remove Batch")
-    batch_number = gr.Slider(0, NUM_OUT_BATCHES - 1, value=0, step=1, label="Batch number to add/remove")
-    add_btn = gr.Button("Add batch", variant="primary")
-    remove_btn = gr.Button("Remove batch", variant="stop")
-    clear_btn = gr.ClearButton()
-
-    final_audio_output = gr.Audio(label="Final MIDI audio", format=AUDIO_FORMAT)
-    final_plot_output = gr.Plot(label="Final MIDI plot")
-    final_file_output = gr.File(label="Final MIDI file")
-
-    add_btn.click(
-        add_batch,
-        [batch_number, final_composition, generated_batches, block_lines],
-        [final_audio_output, final_plot_output, final_file_output, final_composition, generated_batches, block_lines]
-    )
-    remove_btn.click(
-        remove_batch,
-        [batch_number, num_gen_tokens, final_composition, generated_batches, block_lines],
-        [final_audio_output, final_plot_output, final_file_output, final_composition, generated_batches, block_lines]
-    )
-    clear_btn.click(clear, inputs=None,
-                    outputs=[final_audio_output, final_plot_output, final_file_output, final_composition, block_lines])
-
 # -----------------------------
 # APP LAUNCHER
 # -----------------------------
 if __name__ == "__main__":
-    orpheus_app.launch(
+    orpheus_morpheus_app.launch(
         mcp_server=RUNNING_IN_SPACE,   # MCP only on HF
         share=not RUNNING_IN_SPACE,    # Share only locally
         server_name="0.0.0.0",