Switch to file upload instead of YouTube download (YouTube blocks cloud servers)

app.py

@@ -1,4 +1,4 @@
-"""SampleFlip — Describe a beat, get a beat."""
+"""SampleFlip — Upload a sample, get a beat."""
 
 import os
 import sys
@@ -27,72 +27,57 @@ sys.path.insert(0, os.path.join(SPACE_DIR, 'src'))
 
 import gradio as gr
 
-
 GENRES = [
     'trap', 'boombap', 'jazzhouse', 'progressive_house', 'rnb',
     'drill', 'melodic_trap', '2hollis', 'techno', 'breakcore',
 ]
 
 
-def generate_beat(prompt, genre_override, bpm_override):
-    """Main generation function called by Gradio."""
-    if not prompt or not prompt.strip():
-        raise gr.Error("Please describe the beat you want.")
+def generate_beat(audio_file, prompt, genre_override, bpm_override):
+    """Generate a beat from an uploaded sample."""
+    if audio_file is None:
+        raise gr.Error("Please upload an audio file (WAV or MP3).")
 
-    from sampleflip.agent import plan_beat, pick_best_result, generate_drums, pick_bass_pattern
-    from sampleflip.search import search_youtube
-    from sampleflip.download import download_youtube
+    from sampleflip.agent import plan_beat, generate_drums, pick_bass_pattern
     from sampleflip.core.render_beat import GENRE_CONFIGS
 
     logs = []
-
     def log(msg):
         logs.append(msg)
         print(msg)
 
-    # Step 1: Plan
-    log("Planning beat...")
-    genre_arg = genre_override if genre_override != "auto" else None
-    bpm_arg = bpm_override if bpm_override and bpm_override > 0 else None
-
-    plan = plan_beat(prompt, genre_override=genre_arg, bpm_override=bpm_arg)
-    g = plan['genre']
-    b = plan['bpm']
-    q = plan['search_query']
-    name = plan['name']
-    log(f"  Genre: {g} | BPM: {b} | Name: {name}")
-    log(f'  Query: "{q}"')
-
-    # Step 2: Search YouTube
-    log("Searching YouTube...")
-    results = search_youtube(q, count=5)
-    if not results:
-        raise gr.Error("No YouTube results found. Try a different description.")
-
-    # Step 3: LLM picks best result
-    log(f"  Found {len(results)} results, picking best...")
-    try:
-        best_idx = pick_best_result(results, g, prompt)
-    except Exception:
-        best_idx = 0
-    selected = results[best_idx]
-    log(f"  Selected: {selected['title']} ({selected['duration_str']})")
-
-    # Step 4: Download
-    log("Downloading...")
-    sample_path, size_mb = download_youtube(selected['url'])
-    log(f"  Done ({size_mb:.1f} MB)")
-
-    # Step 5: Bass pattern
+    # Step 1: LLM plans genre/bpm/name from prompt (or use defaults)
+    if prompt and prompt.strip():
+        log("Planning beat from description...")
+        genre_arg = genre_override if genre_override != "auto" else None
+        bpm_arg = bpm_override if bpm_override and bpm_override > 0 else None
+        try:
+            plan = plan_beat(prompt, genre_override=genre_arg, bpm_override=bpm_arg)
+            g = plan['genre']
+            b = plan['bpm']
+            name = plan['name']
+        except Exception as e:
+            log(f"  Planning failed ({e}), using defaults")
+            g = genre_override if genre_override != "auto" else "trap"
+            b = bpm_override if bpm_override and bpm_override > 0 else None
+            name = "Beat"
+    else:
+        g = genre_override if genre_override != "auto" else "trap"
+        b = bpm_override if bpm_override and bpm_override > 0 else None
+        name = "Beat"
+
+    log(f"  Genre: {g} | BPM: {b or 'auto'} | Name: {name}")
+
+    # Step 2: Bass pattern
     log("Designing bass pattern...")
     try:
-        bass_pat = pick_bass_pattern(prompt, g)
+        bass_pat = pick_bass_pattern(prompt or f"{g} beat", g)
         os.environ['SAMPLEFLIP_BASS_PATTERN'] = bass_pat
         log(f"  Bass: {bass_pat}")
     except Exception:
         pass
 
-    # Step 6: Drum pattern
+    # Step 3: Drum pattern
     cfg = GENRE_CONFIGS[g]
     nbars = cfg['bars']
     arrangement = cfg['arrangement']
@@ -100,7 +85,7 @@ def generate_beat(prompt, genre_override, bpm_override):
     log("Generating drum pattern...")
     drums_json = None
     try:
-        drum_data = generate_drums(prompt, g, b, nbars, arrangement)
+        drum_data = generate_drums(prompt or f"{g} beat", g, b or cfg['bpm_default'], nbars, arrangement)
         n_pats = len([k for k in drum_data['patterns'] if k != 'silent'])
         log(f"  Created {n_pats} patterns for {nbars} bars")
         drums_json = os.path.join(tempfile.gettempdir(), f'sf_drums_{name}.json')
@@ -109,18 +94,17 @@ def generate_beat(prompt, genre_override, bpm_override):
     except Exception as e:
         log(f"  Drum generation failed ({e}), using defaults")
 
-    # Step 7: Render
-    log(f"Generating {g} beat: \"{name}\"...")
+    # Step 4: Render
+    log(f"Generating {g} beat...")
     output_dir = os.environ['SAMPLEFLIP_OUTPUT_DIR']
 
-    # Add core to path
     core_dir = os.path.join(SPACE_DIR, 'src', 'sampleflip', 'core')
     if core_dir not in sys.path:
         sys.path.insert(0, core_dir)
 
     from sampleflip.core.render_beat import render
     render(
-        sample_path, name, genre=g,
+        audio_file, name, genre=g,
         bpm_hint=b, nbars=None,
         loop_start=None, loop_end=None,
         lofi=None, no_bass=False,
@@ -141,20 +125,24 @@ def generate_beat(prompt, genre_override, bpm_override):
 # Gradio UI
 with gr.Blocks(title="SampleFlip", theme=gr.themes.Base()) as demo:
     gr.Markdown("# SampleFlip")
-    gr.Markdown("Describe a beat, get a beat. AI searches YouTube for the right sample and generates a full produced beat.")
+    gr.Markdown("Upload a sample, get a full produced beat with drums, bass, arrangement, and mastering.")
 
     with gr.Row():
-        with gr.Column(scale=3):
+        with gr.Column(scale=2):
+            audio_input = gr.Audio(
+                label="Upload your sample (WAV or MP3)",
+                type="filepath",
+            )
             prompt = gr.Textbox(
-                label="Describe your beat",
-                placeholder='e.g. "dark trap beat with piano" or "jazzy house with sax"',
-                lines=2,
+                label="Describe the beat (optional)",
+                placeholder='e.g. "dark trap beat" or "jazzy house groove"',
+                lines=1,
             )
         with gr.Column(scale=1):
             genre = gr.Dropdown(
                 choices=["auto"] + GENRES,
                 value="auto",
-                label="Genre (optional)",
+                label="Genre",
             )
             bpm = gr.Slider(
                 minimum=0, maximum=200, step=1, value=0,
@@ -166,17 +154,23 @@ with gr.Blocks(title="SampleFlip", theme=gr.themes.Base()) as demo:
     with gr.Row():
         audio_out = gr.Audio(label="Your Beat", type="filepath")
 
-    logs_out = gr.Textbox(label="Generation Log", lines=12, interactive=False)
+    logs_out = gr.Textbox(label="Generation Log", lines=10, interactive=False)
 
     generate_btn.click(
         fn=generate_beat,
-        inputs=[prompt, genre, bpm],
+        inputs=[audio_input, prompt, genre, bpm],
         outputs=[audio_out, logs_out],
         api_name="generate_beat",
     )
 
     gr.Markdown("""
-**Try these:**  dark trap beat with sad piano melody  |  jazzy house with saxophone  |  aggressive UK drill with strings  |  chill lo-fi boom bap
+### How to use
+1. **Upload** a melody loop, chord progression, or any audio sample (WAV/MP3)
+2. **Describe** what kind of beat you want (optional — helps the AI pick drums and bass)
+3. **Select** a genre and BPM (or leave on auto)
+4. **Click Generate** — AI adds drums, bass, arrangement, FX, and masters to -14 LUFS
+
+**Find samples:** Search YouTube for "free melody loop" or "sample pack preview", download with [yt-dlp](https://github.com/yt-dlp/yt-dlp), and upload here.
 """)
 
 demo.launch(server_name="0.0.0.0", server_port=7860)