Update app.py

app.py

@@ -8,6 +8,17 @@ import pyrubberband as rb
 import soundfile as sf
 from pedalboard import Pedalboard, Reverb
 
+# --- Create dummy files for examples. This runs on server startup. ---
+# This is the key fix: This code is now in the global scope.
+os.makedirs("generated_music", exist_ok=True)
+if not os.path.exists("example_vocal.wav"):
+    sf.write("example_vocal.wav", np.random.randn(44100 * 2), 44100)
+if not os.path.exists("example_drums.wav"):
+    sf.write("example_drums.wav", np.random.randn(44100 * 2), 44100)
+if not os.path.exists("example_synth.wav"):
+    sf.write("example_synth.wav", np.random.randn(44100 * 2), 44100)
+
+
 # --- Musical Sequence Generation ---
 
 def generate_musical_sequence(mode="Scale", scale_name="Major", root_note="C4", num_notes=50):
@@ -19,14 +30,12 @@ def generate_musical_sequence(mode="Scale", scale_name="Major", root_note="C4",
     except Exception as e:
         raise ValueError(f"Invalid root note. Use scientific notation like 'C4'. Error: {e}")
 
-    # Define scale intervals
     intervals = {
         "Chromatic": [1], "Major": [2, 2, 1, 2, 2, 2, 1], "Natural Minor": [2, 1, 2, 2, 1, 2, 2],
         "Harmonic Minor": [2, 1, 2, 2, 1, 3, 1], "Pentatonic Major": [2, 2, 3, 2, 3]
     }
     scale_intervals = intervals.get(scale_name, intervals["Chromatic"])
     
-    # Generate the base scale notes
     scale_notes = [current_midi]
     interval_index = 0
     while len(scale_notes) < num_notes:
@@ -38,11 +47,9 @@ def generate_musical_sequence(mode="Scale", scale_name="Major", root_note="C4",
         return scale_notes[:num_notes]
 
     if mode == "Arpeggio":
-        # Extract chord tones (1st, 3rd, 5th) from the generated scale
         arpeggio_tones = [scale_notes[i] for i in [0, 2, 4]]
         if not arpeggio_tones: return []
         
-        # Build the sequence by cycling through chord tones and octaves
         sequence = []
         octave_offset = 0
         tone_index = 0
@@ -89,15 +96,12 @@ def audio_orchestrator(
     if len(y) / sr < 0.1:
         raise gr.Error("Source audio is too short. Please use a file that is at least 0.1 seconds long.")
     
-    # 1. Generate the target musical sequence (scale or arpeggio)
     progress(0.1, desc=f"Generating {mode}...")
     target_midi_notes = generate_musical_sequence(mode, scale_name, root_note, num_notes)
 
-    # 2. Process each note in the sequence
     final_y = np.array([], dtype=np.float32)
     silence_samples = np.zeros(int(silence_s * sr), dtype=np.float32)
 
-    all_rms = []
     source_segments = [get_random_segment(y, sr) for _ in range(num_notes)]
     all_rms = [rms for _, rms in source_segments if rms > 0]
     max_rms = max(all_rms) if all_rms else 1.0
@@ -107,30 +111,25 @@ def audio_orchestrator(
         note_name = librosa.midi_to_note(target_note)
         progress(0.2 + (i / num_notes) * 0.7, desc=f"Processing Note {i+1}/{num_notes}: {note_name}")
 
-        # Time-stretch segment to the target note duration
         source_duration = len(segment_y) / sr
         stretch_rate = source_duration / note_duration_s if note_duration_s > 0 else 1
         stretched_y = rb.time_stretch(segment_y, sr, stretch_rate)
 
-        # Pitch-shift the stretched segment
-        source_pitch_hz = librosa.note_to_hz('C4') # Assume a neutral source pitch
+        source_pitch_hz = librosa.note_to_hz('C4')
         target_pitch_hz = librosa.midi_to_hz(target_note)
         pitch_shift_ratio = target_pitch_hz / source_pitch_hz
         shifted_y = rb.pitch_shift(stretched_y, sr, pitch_shift_ratio)
         
-        # Apply dynamic expression based on original segment's volume
         volume_factor = (rms / max_rms) * 0.9 + 0.1
         shifted_y *= volume_factor
         
         final_y = np.concatenate((final_y, shifted_y, silence_samples))
 
-    # 3. Apply final effects processing
     progress(0.95, desc="Applying final effects...")
     if reverb_mix > 0:
         board = Pedalboard([Reverb(room_size=reverb_room_size, wet_level=reverb_mix, dry_level=1.0 - reverb_mix)])
         final_y = board(final_y, sr)
 
-    # 4. Export final audio
     output_dir = "generated_music"
     os.makedirs(output_dir, exist_ok=True)
     output_filename = f"{mode.lower()}_{scale_name.lower().replace(' ', '_')}_{root_note}.wav"
@@ -158,8 +157,7 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue="teal", secondary_hue="orange"))
                 mode = gr.Radio(["Scale", "Arpeggio"], label="Generation Mode", value="Arpeggio")
                 scale_name = gr.Dropdown(["Major", "Natural Minor", "Harmonic Minor", "Pentatonic Major", "Chromatic"], label="Scale", value="Major")
                 
-                # FIX IS HERE: Converted the numpy array from librosa to a python list with .tolist()
-                root_note_choices = librosa.midi_to_note(list(range(36, 73))).tolist() # C2 to C5
+                root_note_choices = librosa.midi_to_note(list(range(36, 73))).tolist()
                 root_note = gr.Dropdown(root_note_choices, label="Root Note", value="C4")
                 
                 num_notes = gr.Slider(10, 200, value=70, step=1, label="Number of Notes")
@@ -189,6 +187,7 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue="teal", secondary_hue="orange"))
     )
     
     gr.Examples(
+        # FIX IS HERE: Using simple filenames because the files are now in the root directory.
         examples=[
             ["example_vocal.wav", "Arpeggio", "Natural Minor", "A3", 80, 0.15, 0.1, 0.4, 0.7],
             ["example_drums.wav", "Scale", "Pentatonic Major", "C3", 50, 0.25, 0.0, 0.2, 0.8],
@@ -199,17 +198,9 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue="teal", secondary_hue="orange"))
             note_duration_s, silence_s, reverb_mix, reverb_room_size
         ],
         outputs=[audio_output],
-        cache_examples=True # Recommended for HF Spaces
+        fn=audio_orchestrator,
+        cache_examples=True 
     )
 
 if __name__ == "__main__":
-    # Create dummy files for local testing of the examples
-    # Check if files exist before creating them
-    if not os.path.exists("example_vocal.wav"):
-        sf.write("example_vocal.wav", np.random.randn(44100 * 2), 44100)
-    if not os.path.exists("example_drums.wav"):
-        sf.write("example_drums.wav", np.random.randn(44100 * 2), 44100)
-    if not os.path.exists("example_synth.wav"):
-        sf.write("example_synth.wav", np.random.randn(44100 * 2), 44100)
-    
     demo.launch(debug=True)