Update app.py

app.py

@@ -883,5 +883,58 @@ gr.Interface(
     allow_flagging="never"
 ).launch()
 
+# === Hugging Face API Integration ===
+def hf_api_process(audio_data_url, effects_json, isolate, preset, export_format):
+    import base64, tempfile, json, os
+    from pydub import AudioSegment
+
+    try:
+        # Step 1: Decode base64 audio string
+        header, base64_data = audio_data_url.split(",", 1)
+        audio_bytes = base64.b64decode(base64_data)
+        suffix = ".mp3" if "mpeg" in header else ".wav"
+        with tempfile.NamedTemporaryFile(delete=False, suffix=suffix) as f:
+            f.write(audio_bytes)
+            input_path = f.name
+
+        # Step 2: Parse effects
+        if isinstance(effects_json, str):
+            effects = json.loads(effects_json)
+        else:
+            effects = effects_json
+
+        # Step 3: Call the processing function
+        output_path, _, _, _, _, _ = process_audio(
+            input_path, effects, isolate, preset, export_format
+        )
+
+        # Step 4: Convert result to base64
+        with open(output_path, "rb") as f:
+            out_b64 = base64.b64encode(f.read()).decode()
+        mime = "audio/wav" if export_format.lower() == "wav" else "audio/mpeg"
+        return f"data:{mime};base64,{out_b64}"
+
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Define an API interface for Hugging Face Spaces
+api_interface = gr.Interface(
+    fn=hf_api_process,
+    inputs=[
+        gr.Text(label="Audio Base64 Data URL"),
+        gr.Textbox(label="Effects (JSON)"),
+        gr.Checkbox(label="Isolate Vocals"),
+        gr.Textbox(label="Preset"),
+        gr.Textbox(label="Export Format")
+    ],
+    outputs=gr.Text(label="Processed Audio as Base64 URL"),
+    allow_flagging="never"
+)
+
+# Include this in the Blocks launch to enable API access
+demo.launch(show_api=True)
+api_interface.launch(inline=False, share=False)
+
+
 # Launch Gradio App
 demo.launch()