handling return type

app.py

@@ -5,6 +5,7 @@ from demucs import pretrained
 from demucs.apply import apply_model
 from pyharp import *
 from audiotools import AudioSignal
+from typing import Tuple, Dict
 
 
 # Available Demucs models
@@ -65,15 +66,41 @@ def separate_stem(audio_file_path: str, model_name: str, stem_choice: str):
     stem_signal = AudioSignal(stem.cpu().numpy().astype('float32'), sample_rate=sr)
     return stem_signal
 
-def process_fn_stem(audio_file_path: str, demucs_model: str, stem_choice: str):
+def process_fn_stem(
+    audio_file_path: str,
+    demucs_model: str,
+    stem_choice: str,
+    request: gr.Request  # Needed for OAuth
+) -> Tuple[str, Dict]:
     """
-    PyHARP process function:
-      - Separates the chosen stem using Demucs.
-      - Saves the stem as a .wav file.
+    PyHARP process function (OAuth-safe):
+    - Separates a stem from audio using Demucs.
+    - Returns a path to the output WAV + dummy metadata.
     """
-    stem_signal = separate_stem(audio_file_path, model_name=demucs_model, stem_choice=stem_choice)
-    stem_path = save_audio(stem_signal, f"{stem_choice.lower().replace(' ', '_')}.wav")
-    return stem_path, LabelList(labels=[])
+
+    # Get user's HF OAuth token if available (optional usage)
+    user_token = request.headers.get("Authorization", None)
+    if user_token:
+        user_token = user_token.replace("Bearer ", "")
+        print(f"User is authenticated with token (truncated): {user_token[:10]}...")
+
+    # Run stem separation
+    stem_signal = separate_stem(
+        audio_file_path,
+        model_name=demucs_model,
+        stem_choice=stem_choice
+    )
+
+    # Save output audio file
+    stem_filename = f"{stem_choice.lower().replace(' ', '_')}.wav"
+    stem_path = save_audio(stem_signal, stem_filename)
+
+    # Return stem path + basic metadata (dict that Gradio can serialize)
+    return stem_path, {
+        "stem": stem_choice,
+        "model": demucs_model,
+        "is_authenticated": bool(user_token)
+    }
 
 
 # Define the model card
@@ -107,4 +134,4 @@ with gr.Blocks() as demo:
     )
 
 demo.queue()
-demo.launch(show_error=True, share=True)
+demo.launch(show_error=True)