expose more params

app.py

@@ -3,6 +3,21 @@ import numpy as np
 from fastapi import FastAPI, UploadFile, File, Form
 import tempfile, io, base64, math, threading
 from fastapi.middleware.cors import CORSMiddleware
+from contextlib import contextmanager
+
+@contextmanager
+def mrt_overrides(mrt, **kwargs):
+    """Temporarily set attributes on MRT if they exist; restore after."""
+    old = {}
+    try:
+        for k, v in kwargs.items():
+            if hasattr(mrt, k):
+                old[k] = getattr(mrt, k)
+                setattr(mrt, k, v)
+        yield
+    finally:
+        for k, v in old.items():
+            setattr(mrt, k, v)
 
 # loudness utils
 try:
@@ -239,9 +254,13 @@ def generate(
     beats_per_bar: int = Form(4),
     styles: str = Form("acid house"),
     style_weights: str = Form(""),
-    loop_weight: float = Form(1.0),               # NEW
-    loudness_mode: str = Form("auto"),            # NEW
-    loudness_headroom_db: float = Form(1.0),      # NEW
+    loop_weight: float = Form(1.0),
+    loudness_mode: str = Form("auto"),
+    loudness_headroom_db: float = Form(1.0),
+    # NEW per-request knobs
+    guidance_weight: float = Form(5.0),
+    temperature: float = Form(1.1),
+    topk: int = Form(40),
 ):
     # Read file
     data = loop_audio.file.read()
@@ -256,19 +275,23 @@ def generate(
     weights = [float(x) for x in style_weights.split(",")] if style_weights else None
 
     mrt = get_mrt()  # warm once, in this worker thread
-    mrt = get_mrt()
-    wav, loud_stats = generate_loop_continuation_with_mrt(
-        mrt,
-        input_wav_path=tmp_path,
-        bpm=bpm,
-        extra_styles=extra_styles,
-        style_weights=weights,
-        bars=bars,
-        beats_per_bar=beats_per_bar,
-        loop_weight=loop_weight,
-        loudness_mode=loudness_mode,
-        loudness_headroom_db=loudness_headroom_db,
-    )
+    # Temporarily override MRT inference knobs for this request
+    with mrt_overrides(mrt,
+                       guidance_weight=guidance_weight,
+                       temperature=temperature,
+                       topk=topk):
+        wav, loud_stats = generate_loop_continuation_with_mrt(
+            mrt,
+            input_wav_path=tmp_path,
+            bpm=bpm,
+            extra_styles=extra_styles,
+            style_weights=weights,
+            bars=bars,
+            beats_per_bar=beats_per_bar,
+            loop_weight=loop_weight,
+            loudness_mode=loudness_mode,
+            loudness_headroom_db=loudness_headroom_db,
+        )
 
     # Return base64 WAV + minimal metadata
     buf = io.BytesIO()
@@ -286,10 +309,14 @@ def generate(
             "styles": extra_styles,
             "style_weights": weights,
             "loop_weight": loop_weight,
-            "loudness": loud_stats,                       # NEW
+            "loudness": loud_stats,
             "sample_rate": mrt.sample_rate,
             "channels": mrt.num_channels,
             "crossfade_seconds": mrt.config.crossfade_length,
+            # Echo the actual knobs used
+            "guidance_weight": guidance_weight,
+            "temperature": temperature,
+            "topk": topk,
         },
     }