Update app.py

app.py

@@ -1,10 +1,11 @@
 ### IMPORTS ###
 import os
 os.environ["KERAS_BACKEND"] ="tensorflow"
+os.environ["TF_ENABLE_XLA"] = "0"
 import random
 import keras
 import gradio as gr
-import time   # <-- added
+import time 
 from src.inference import generate_chorale, draw_random_sample
 from src.dataset import NoteEncoder
 from src.metrics import Preplexity
@@ -60,7 +61,8 @@ print(f"Loaded CSS/markdown assets in {time.time()-t2:.2f}s")
 
 ### GENERATION FUNCTIONS ###
 def pick_random_seed():
-    return draw_random_sample(VAL_PATH, seed=random.randint(0, 9999))
+    path = draw_random_sample(VAL_PATH, seed=random.randint(0, 9999))
+    return path, os.path.basename(path)
 
 def generate_fn(seed_path, seed_len, gen_len, temp):
     sample_rows = slice(0, seed_len)
@@ -113,10 +115,11 @@ with gr.Blocks(css=css, title="BachNet") as demo:
             gr.Markdown("## Customize Your Chorale")
             with gr.Row():
                 sample_seed_btn = gr.Button("Pick Random Seed", variant="primary")
+                hidden_full_path = gr.State()
                 seed_path_box = gr.Textbox(label="Selected Seed Path", interactive=False)
             
             seed_len_slider = gr.Slider(40, 80, 50, step=1, label="Seed Length")
-            gen_len_slider = gr.Slider(20, 100, 30, step=1, label="Generated Length")
+            gen_len_slider = gr.Slider(20, 100, 30, step=1, label="Generated Length (Chords)")
             temp_slider = gr.Slider(0.5, 1.8, 0.9, step=0.1, label="Temperature")
             
             generate_btn = gr.Button("Generate", variant="primary")
@@ -130,11 +133,10 @@ with gr.Blocks(css=css, title="BachNet") as demo:
 
 
 ### EVENTS ###
-    demo.load(pick_random_seed, outputs=seed_path_box)
-    sample_seed_btn.click(pick_random_seed, outputs=seed_path_box)
-    generate_btn.click(generate_fn, inputs=[seed_path_box, seed_len_slider, gen_len_slider, temp_slider],
-                       outputs=audio_player)
-    
+    demo.load(pick_random_seed, outputs=[hidden_full_path, seed_path_box])
+    sample_seed_btn.click(pick_random_seed, outputs=[hidden_full_path, seed_path_box])
+    generate_btn.click(generate_fn, inputs=[hidden_full_path, seed_len_slider, gen_len_slider, temp_slider],
+    outputs=audio_player)
     english_btn.click(set_english, outputs=[title_md, summary_md, help_md])
     persian_btn.click(set_persian, outputs=[title_md, summary_md, help_md])