Switch to Gradio

Dockerfile

@@ -17,4 +17,4 @@ COPY api.py ./
 
 EXPOSE 7860
 
-CMD ["uvicorn", "api:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["python", "api.py"]

api.py

@@ -1,11 +1,11 @@
 """
-Snare Scout API - Runs on Hugging Face Spaces
+Snare Scout API - Gradio version for HuggingFace
 """
 
 import os
 import sys
-import base64
-import traceback
+import json
+import gradio as gr
 
 sys.path.insert(0, os.path.join(os.path.dirname(os.path.abspath(__file__)), "src"))
 
@@ -24,21 +24,6 @@ if not os.path.exists(DB_PATH):
 import scout
 scout.DEFAULT_DB_PATH = DB_PATH
 
-from fastapi import FastAPI
-from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import JSONResponse
-from pydantic import BaseModel
-
-app = FastAPI(title="Snare Scout API")
-
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
 print("🔄 Loading AI models...")
 embedder = scout.get_embedder()
 print("🔄 Loading library...")
@@ -46,37 +31,27 @@ lib = scout.load_library_matrices(DB_PATH, False)
 print(f"✅ Ready! {len(lib['ids'])} clips loaded")
 
 
-class SearchRequest(BaseModel):
-    audio_base64: str
-    top_k: int = 15
-
-
-@app.get("/")
-def root():
-    return {"status": "ok", "clips": len(lib['ids'])}
-
-
-@app.post("/search")
-async def search(req: SearchRequest):
-    print(f"[API] Received search request, top_k={req.top_k}")
+def search(audio_path, top_k=15):
+    print(f"[API] Search request: {audio_path}, top_k={top_k}")
+    
+    if audio_path is None:
+        return json.dumps({"results": []})
     
     try:
-        audio_bytes = base64.b64decode(req.audio_base64)
-        print(f"[API] Audio size: {len(audio_bytes)} bytes")
+        with open(audio_path, 'rb') as f:
+            audio_bytes = f.read()
         
-        if len(audio_bytes) < 100:
-            return JSONResponse(content={"results": []})
+        print(f"[API] Audio size: {len(audio_bytes)} bytes")
         
-        print("[API] Starting search...")
         results = scout.search_library(
             embedder, 
             audio_bytes, 
             lib, 
-            top_k=req.top_k, 
+            top_k=int(top_k), 
             debug=False, 
             db_path=DB_PATH
         )
-        print(f"[API] Search complete, {len(results)} results")
+        print(f"[API] Found {len(results)} results")
         
         output = {
             "results": [
@@ -91,9 +66,22 @@ async def search(req: SearchRequest):
             ]
         }
         
-        return JSONResponse(content=output)
+        return json.dumps(output)
         
     except Exception as e:
         print(f"[API] ERROR: {e}")
-        print(traceback.format_exc())
-        return JSONResponse(content={"error": str(e), "results": []})
+        return json.dumps({"error": str(e), "results": []})
+
+
+demo = gr.Interface(
+    fn=search,
+    inputs=[
+        gr.Audio(type="filepath", label="Upload audio"),
+        gr.Number(value=15, label="Results")
+    ],
+    outputs=gr.Textbox(label="Results JSON"),
+    title="Snare Scout API",
+    api_name="search"
+)
+
+demo.launch(server_name="0.0.0.0", server_port=7860)

requirements.txt

@@ -1,6 +1,4 @@
-fastapi
-uvicorn[standard]
-python-multipart
+gradio
 numpy>=1.24.0,<2.0.0
 scipy>=1.12.0
 soundfile>=0.12.0