Create main.py

backend/main.py

@@ -0,0 +1,61 @@
+import os
+import asyncio
+from fastapi import FastAPI, UploadFile, File, Form, Header, BackgroundTasks
+from fastapi.responses import JSONResponse
+from pydantic import BaseModel
+from typing import Optional, List
+from queue import JobQueue
+from worker import run_job
+import db
+
+app = FastAPI()
+queue = JobQueue()
+
+@app.on_event("startup")
+async def startup():
+    await db.init_db()
+    asyncio.create_task(queue.process_jobs())
+
+class MergeRequest(BaseModel):
+    model_a_source: str      # "hf" or "civitai"
+    model_a_id: str
+    model_b_source: str
+    model_b_id: str
+    method: str = "linear"   # linear, evolutionary
+    linear_alpha: Optional[float] = 0.5
+    evo_params: Optional[dict] = None
+    dataset: Optional[str] = None  # base64 encoded zip or path in job dir
+    output_repo_name: Optional[str] = None
+    hf_token_manual: Optional[str] = None
+    civitai_key: Optional[str] = None
+
+@app.post("/api/submit-job")
+async def submit_job(
+    req: MergeRequest,
+    x_hf_user_access_token: Optional[str] = Header(None)
+):
+    token = req.hf_token_manual or x_hf_user_access_token
+    if not token:
+        return JSONResponse(status_code=400, content={"error": "HF token required"})
+    job_id = await queue.add_job(req, token)
+    return {"job_id": job_id}
+
+@app.get("/api/job-status/{job_id}")
+async def job_status(job_id: str):
+    job = await db.get_job(job_id)
+    if not job:
+        return JSONResponse(status_code=404, content={"error": "Job not found"})
+    return job
+
+@app.get("/api/queue")
+async def get_queue():
+    return await queue.get_queue_status()
+
+@app.post("/api/upload-dataset")
+async def upload_dataset(file: UploadFile = File(...)):
+    import tempfile
+    os.makedirs("/data/datasets", exist_ok=True)
+    path = f"/data/datasets/{file.filename}"
+    with open(path, "wb") as f:
+        f.write(await file.read())
+    return {"path": path}