feat: create /login route to set auth token on cookie for user

app.py

@@ -293,14 +293,14 @@ def split_image(image: Image.Image) -> List[Image.Image]:
         images.append(Image.fromarray(image[:, s:e+1]))
     return [preprocess_image(image) for image in images]
 
-from fastapi import FastAPI, File, UploadFile, HTTPException, Depends, Header
+from fastapi import FastAPI, File, UploadFile, HTTPException, Depends, Header, Response
 from fastapi.responses import JSONResponse, FileResponse, StreamingResponse
 import tempfile
 import os
 
 app = FastAPI()
 
-def verify_token(authorization: str = Header(...)):
+def verify_token(authorization: str = Header(...), cookie_token: str = Header(None)):
     if not authorization.startswith("Bearer "):
         raise HTTPException(status_code=403, detail="Invalid or missing token")
     
@@ -373,9 +373,10 @@ async def remove_image_background(image: UploadFile = File(...), token: str = De
         raise HTTPException(status_code=500, detail=str(e))
 
 
-# @app.get("/")
-# async def root_():
-#     return JSONResponse(content={"message": "HI!"})
+@app.get("/login")
+async def login(response: Response, token: str = Depends(verify_token)):
+    response.set_cookie(key="auth_token", value=f"Bearer {os.getenv('AUTH_TOKEN')}", httponly=True)
+    return JSONResponse(content={"message": "HI!"})
 
 with gr.Blocks(delete_cache=(600, 600)) as demo:
     gr.Markdown("""