space/face-lock
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

Add buffered crop UI and API controls

Browse Source
This commit is contained in:
Luna
2026-04-11 16:33:45 +02:00
parent 660ce4e7cc
commit 3b5a9e8635
2 changed files with 28 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
app/main.py
View File

@@ -1,4 +1,4 @@
from fastapi import FastAPI, File, HTTPException, UploadFile
from fastapi import FastAPI, File, Form, HTTPException, UploadFile
from fastapi.responses import HTMLResponse, StreamingResponse
from app.config import settings
@@ -33,30 +33,45 @@ def index():
<div class="grid gap-6 md:grid-cols-2">
<section class="rounded-2xl border border-slate-800 bg-slate-900 p-4">
<input id="file" type="file" accept="image/*" class="block w-full rounded-lg border border-slate-700 bg-slate-950 p-3" />
<label class="mt-4 block text-sm text-slate-400">Buffer ratio</label>
<input id="buffer_ratio" type="number" step="0.05" min="0" max="0.5" value="0.15" class="block w-full rounded-lg border border-slate-700 bg-slate-950 p-3" />
<button id="go" class="mt-4 rounded-lg bg-cyan-500 px-4 py-2 font-semibold text-slate-950">Process</button>
<pre id="meta" class="mt-4 whitespace-pre-wrap rounded-lg bg-slate-950 p-3 text-xs text-slate-300"></pre>
</section>
<section class="rounded-2xl border border-slate-800 bg-slate-900 p-4">
<div class="mb-3 text-sm font-semibold text-slate-400">Result</div>
<img id="result" class="hidden w-full rounded-xl border border-slate-800" />
<div class="grid gap-4">
<div>
<div class="mb-2 text-xs uppercase tracking-wide text-slate-500">Crop</div>
<img id="crop" class="hidden w-full rounded-xl border border-slate-800" />
</div>
<div>
<div class="mb-2 text-xs uppercase tracking-wide text-slate-500">Annotated source</div>
<img id="annotated" class="hidden w-full rounded-xl border border-slate-800" />
</div>
</div>
</section>
</div>
</main>
<script>
const file = document.getElementById('file');
const go = document.getElementById('go');
const result = document.getElementById('result');
const crop = document.getElementById('crop');
const annotated = document.getElementById('annotated');
const meta = document.getElementById('meta');
go.onclick = async () => {
if (!file.files.length) return;
const form = new FormData();
form.append('file', file.files[0]);
form.append('buffer_ratio', document.getElementById('buffer_ratio').value);
meta.textContent = 'Working...';
const resp = await fetch('/api/focus', { method: 'POST', body: form });
const data = await resp.json();
meta.textContent = JSON.stringify(data, null, 2);
result.src = data.crop_data_url;
result.classList.remove('hidden');
crop.src = data.crop_data_url;
annotated.src = data.annotated_data_url;
crop.classList.remove('hidden');
annotated.classList.remove('hidden');
};
</script>
</body>
@@ -66,23 +81,23 @@ def index():
@app.post("/api/focus")
async def focus(file: UploadFile = File(...)):
async def focus(file: UploadFile = File(...), buffer_ratio: float = Form(0.15)):
from app.vision import process_image
try:
payload = await file.read()
return process_image(payload, file.filename or "upload")
return process_image(payload, file.filename or "upload", buffer_ratio=buffer_ratio)
except ValueError as exc:
raise HTTPException(status_code=400, detail=str(exc)) from exc
@app.post("/api/focus/image")
async def focus_image(file: UploadFile = File(...)):
async def focus_image(file: UploadFile = File(...), buffer_ratio: float = Form(0.15)):
from app.vision import process_image
try:
payload = await file.read()
result = process_image(payload, file.filename or "upload")
result = process_image(payload, file.filename or "upload", buffer_ratio=buffer_ratio)
return StreamingResponse(result["crop_bytes_io"], media_type=result["mime_type"])
except ValueError as exc:
raise HTTPException(status_code=400, detail=str(exc)) from exc