add debug prints for loop length investigation

app.py

@@ -203,6 +203,9 @@ def process_fn(description: str, min_length: float, max_length: float, seed: int
 
     left_hint = [hint[..., :int(0.02 * sr) * int(unit_length / 2)]]
     valid_tokens = [int(unit_length)]
+    print(f"DEBUG: unit_length={unit_length}, valid_tokens={valid_tokens}, "
+          f"left_hint_samples={left_hint[0].shape[-1]}, sr={sr}, "
+          f"loopgen.duration={loopgen.duration}, loopgen.frame_rate={loopgen.frame_rate}")
 
     setup = {
         "span_arrangement": "stride1",
@@ -227,10 +230,17 @@ def process_fn(description: str, min_length: float, max_length: float, seed: int
             left_hint=left_hint, left_hint_sr=sr, valid_tokens=valid_tokens,
         )
     result = results[0].to(device="cpu", dtype=torch.float32)
+    print(f"DEBUG: result_samples_before_doubling={result.shape[-1]}, "
+          f"result_seconds={result.shape[-1] / loopgen.sample_rate:.2f}, "
+          f"loopgen.sample_rate={loopgen.sample_rate}")
 
     max_duration = int(loopgen.duration * loopgen.frame_rate)
+    print(f"DEBUG: max_duration={max_duration}, valid_tokens[0]={valid_tokens[0]}, "
+          f"doubling={valid_tokens[0] == max_duration}")
     if valid_tokens[0] == max_duration:
         result = torch.cat([result, result], -1)
+    print(f"DEBUG: result_samples_final={result.shape[-1]}, "
+          f"result_seconds_final={result.shape[-1] / loopgen.sample_rate:.2f}")
 
     return _write_temp_wav(result, loopgen.sample_rate)