Update app.py

app.py

@@ -1,60 +1,35 @@
-import gradio as gr
-import requests
-import os
-import time
 from gradio_client import Client
+import os
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
+BACKEND_SPACE = "CleanSong-AI/Main-tool-backend-main"
 
-# Backend endpoint
-BACKEND_SPACE_API = (
-    "https://hf.space/embed/CleanSong-AI/Main-tool-backend-main/api/predict/"
-)
-
-# ------------------------------------------------------------
-# (Optional) Support function if you actually need this later
-# ------------------------------------------------------------
-def call_space(space_repo, input_data, param_name="file_path"):
-    client = Client(space_repo, hf_token=HF_TOKEN)
-    start = time.time()
-
-    if isinstance(input_data, dict):
-        kwargs = {k: v for k, v in input_data.items()}
-    else:
-        kwargs = {param_name: input_data}
-
-    try:
-        result = client.predict(api_name="/predict", **kwargs)
-    except Exception as e:
-        raise Exception(f"Space call failed: {e}")
-
-    return result
-
-# ------------------------------------------------------------
-# Main function called by Gradio
-# ------------------------------------------------------------
 def clean_song(file_path):
-    """
-    Sends uploaded audio file to the HF backend and returns cleaned version.
-    """
     if file_path is None:
         return None
 
-    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+    client = Client(
+        BACKEND_SPACE,
+        hf_token=HF_TOKEN
+    )
 
     try:
-        with open(file_path, "rb") as f:
-            files = {"data": (os.path.basename(file_path), f, "audio/mpeg")}
-            response = requests.post(BACKEND_SPACE_API, files=files, headers=headers)
-            response.raise_for_status()
+        # This EXACTLY matches how Gradio uploads files
+        result = client.predict(
+            file_path,
+            api_name="/predict"
+        )
     except Exception as e:
-        return f"Error: {e}"
+        return f"Error calling backend: {e}"
 
-    cleaned_bytes = response.content
+    # result is usually a filepath OR bytes depending on backend
+    if isinstance(result, str) and os.path.exists(result):
+        return result
 
-    output_path = "cleaned_output.mp3"
+    # fallback: save bytes
+    output_path = "cleaned_output.wav"
     with open(output_path, "wb") as f:
-        f.write(cleaned_bytes)
+        f.write(result)
 
     return output_path