Update app.py

app.py

@@ -36,9 +36,11 @@ st.sidebar.markdown(ABOUT_TEXT)
 st.sidebar.markdown(CONTACT_TEXT)
 
 
-def inference(audio, num_epoch, tonic_name):
+def inference(audio, num_epoch, tonic_name, chord_duration):
     generator = Generator()
-    output_midi_data = generator(audio, num_epoch=int(num_epoch), tonic=tonic_name)[0]
+    if chord_duration == 0.0:
+        chord_duration = None
+    output_midi_data = generator(audio, num_epoch=int(num_epoch), tonic=tonic_name, chord_duration=chord_duration)[0]
     name = uuid.uuid4()
     output_midi_data.write(f'{name}.mid')
     fs = FluidSynth("font.sf2")
@@ -79,6 +81,14 @@ num_epoch = st.number_input("Number of epochs:",
                             value=1,
                             )
 
+chord_duration = st.number_input("Custom chord duration (leave zero for auto-calculation):",
+                                 min_value=0.0,
+                                 max_value=1000.0,
+                                 step=0.0001,
+                                 value=0.0,
+                                 format="%.4f"
+                                 )
+
 tonic_name = st.selectbox(
     'Select tonic:',
     ["minor", "major"]
@@ -94,7 +104,7 @@ if generate_image_button:
             f.write(uploaded_file.getvalue())
     # print(uploaded_file.getvalue())
     with st.spinner(text=f"Generating, this may take some time..."):
-        before, after = inference(input_file, num_epoch, tonic_name)
+        before, after = inference(input_file, num_epoch, tonic_name, chord_duration)
     st.markdown("Before:")
     st.audio(before)
     st.markdown("After:")