|
|
import io |
|
|
from fastapi import FastAPI, UploadFile, File, Form |
|
|
from fastapi.middleware.cors import CORSMiddleware |
|
|
from fastapi.responses import JSONResponse |
|
|
import gradio as gr |
|
|
|
|
|
app = FastAPI() |
|
|
app.add_middleware( |
|
|
CORSMiddleware, |
|
|
allow_origins=["*"], |
|
|
allow_credentials=True, |
|
|
allow_methods=["*"], |
|
|
allow_headers=["*"], |
|
|
) |
|
|
|
|
|
THRESH = 0.65 |
|
|
|
|
|
@app.post("/verify") |
|
|
async def verify(name: str = Form(...), image: UploadFile = File(...)): |
|
|
try: |
|
|
data = await image.read() |
|
|
score = min(0.99, 0.6 + (len(data) % 400) / 1000) |
|
|
decision = "Access Granted" if score >= THRESH else "Access Denied" |
|
|
return {"decision": decision, "score": round(score, 3), "token": ""} |
|
|
except Exception as e: |
|
|
return JSONResponse( |
|
|
{"decision": "Access Denied", "score": 0.0, "error": str(e)}, |
|
|
status_code=500, |
|
|
) |
|
|
|
|
|
def demo_verify(name, img): |
|
|
if not name or img is None: |
|
|
return "Missing input", 0.0 |
|
|
buf = io.BytesIO() |
|
|
img.save(buf, format="PNG") |
|
|
data = buf.getvalue() |
|
|
score = min(0.99, 0.6 + (len(data) % 400) / 1000) |
|
|
decision = "Access Granted" if score >= THRESH else "Access Denied" |
|
|
return decision, round(score, 3) |
|
|
|
|
|
demo = gr.Interface( |
|
|
fn=demo_verify, |
|
|
inputs=["text", "image"], |
|
|
outputs=["text", "number"], |
|
|
title="EMMYS Verify API" |
|
|
) |
|
|
|
|
|
app = gr.mount_gradio_app(app, demo, path="/") |
|
|
|
