Things

backend/main.py

@@ -1,11 +1,32 @@
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, File, Form, UploadFile, HTTPException, Query
+from fastapi.middleware.cors import CORSMiddleware
 import subprocess
 import threading
 import time
+import os
+import json
+import uuid
+import hashlib
+from pathlib import Path
+from pydantic import BaseModel
 import RPi.GPIO as GPIO
 
 app = FastAPI()
 
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+INCOMING_DIR = Path("/tmp/quibot-audio/incoming")
+LOCKED_DIR = Path("/tmp/quibot-audio/locked")
+PROCESSED_DIR = Path("/tmp/quibot-audio/processed")
+INCOMING_DIR.mkdir(parents=True, exist_ok=True)
+LOCKED_DIR.mkdir(parents=True, exist_ok=True)
+PROCESSED_DIR.mkdir(parents=True, exist_ok=True)
+
 # -------------------------
 # GPIO SETUP
 # -------------------------
@@ -104,6 +125,83 @@ def stop_motor(token: str):
     return {"status": "motor stopped"}
 
 
+@app.post("/audio/upload")
+async def upload_audio(file: UploadFile = File(...), format: str = "wav"):
+    raw_content = await file.read()
+    
+    checksum = hashlib.sha256(raw_content).hexdigest()[:16]
+    filename = f"{checksum[:10]}-{uuid.uuid4().hex[:8]}.wav"
+    
+    filepath = INCOMING_DIR / filename
+    filepath.write_bytes(raw_content)
+    
+    return {"status": "received", "filename": str(filepath), "lock_url": f"/audio/lock/{filepath.name}"}
+
+
+@app.get("/audio/incoming")
+def list_incoming():
+    files = []
+    for f in sorted(INCOMING_DIR.iterdir()):
+        meta = f.stat()
+        files.append({
+            "filename": f.name,
+            "size_bytes": meta.st_size,
+            "modified_iso": time.ctime(meta.st_mtime),
+        })
+    return {"count": len(files), "files": files}
+
+
+@app.post("/audio/lock/{filename}")
+def lock_audio(filename: str):
+    src = INCOMING_DIR / filename
+    dst = LOCKED_DIR / filename
+    
+    if not src.exists():
+        raise HTTPException(status_code=404, detail=f"File {filename} not found")
+    
+    if dst.exists():
+        return {"status": "already_locked", "filename": filename}
+    
+    os.rename(str(src), str(dst))
+    return {"status": "locked", "filename": filename}
+
+
+@app.post("/audio/unlock/{filename}")
+def unlock_audio(filename: str):
+    src = LOCKED_DIR / filename
+    dst = INCOMING_DIR / filename
+    
+    if not src.exists():
+        raise HTTPException(status_code=404, detail=f"File {filename} not found")
+    
+    os.rename(str(src), str(dst))
+    return {"status": "unlocked", "filename": filename}
+
+
+@app.post("/audio/cancel/{filename}")
+def cancel_audio(filename: str):
+    src = LOCKED_DIR / filename
+    dst = INCOMING_DIR / filename
+    
+    if not src.exists():
+        raise HTTPException(status_code=404, detail=f"File {filename} not found")
+    
+    os.rename(str(src), str(dst))
+    return {"status": "cancelled", "filename": filename}
+
+
+@app.post("/audio/process/{filename}")
+def process_audio(filename: str):
+    locked = LOCKED_DIR / filename
+    processed = PROCESSED_DIR / filename
+    
+    if not locked.exists():
+        raise HTTPException(status_code=404, detail=f"File {filename} not found")
+    
+    os.rename(str(locked), str(processed))
+    return {"status": "processed", "filename": filename}
+
+
 @app.on_event("shutdown")
 def shutdown():
     global motor_running