afdx2 commited on
Commit
3868453
·
verified ·
1 Parent(s): b57c777

Update server1.py

Browse files
Files changed (1) hide show
  1. server1.py +21 -3
server1.py CHANGED
@@ -8,6 +8,8 @@ from fastapi import FastAPI, File, UploadFile
8
  from fastapi.responses import JSONResponse
9
  from PIL import Image, UnidentifiedImageError
10
  import open_clip
 
 
11
  import os
12
  os.environ.setdefault("HF_HOME", "/app/cache")
13
  os.environ.setdefault("XDG_CACHE_HOME", "/app/cache")
@@ -135,13 +137,29 @@ def process_image_bytes(image_bytes: bytes):
135
  def root():
136
  return {"status": "ok", "device": DEVICE}
137
 
 
138
  @app.post("/predict")
139
- async def predict(front: UploadFile = File(...), back: Optional[UploadFile] = File(None)):
140
  try:
 
 
 
 
 
 
 
 
141
  front_bytes = await front.read()
 
 
142
  if back is not None:
 
143
  _ = await back.read()
 
144
  result = process_image_bytes(front_bytes)
145
  return JSONResponse(content={"code": 200, "data": result})
146
- except Exception:
147
- return JSONResponse(content={"code": 404, "data": {}}, status_code=200)
 
 
 
 
8
  from fastapi.responses import JSONResponse
9
  from PIL import Image, UnidentifiedImageError
10
  import open_clip
11
+ from fastapi import FastAPI, File, UploadFile, Request
12
+ import traceback
13
  import os
14
  os.environ.setdefault("HF_HOME", "/app/cache")
15
  os.environ.setdefault("XDG_CACHE_HOME", "/app/cache")
 
137
  def root():
138
  return {"status": "ok", "device": DEVICE}
139
 
140
+
141
  @app.post("/predict")
142
+ async def predict(front: UploadFile = File(None), back: Optional[UploadFile] = File(None), request: Request = None):
143
  try:
144
+ # log de cabeceras y tipos
145
+ if request:
146
+ print("headers:", dict(request.headers))
147
+ if front is None:
148
+ print("no llego 'front'")
149
+ return JSONResponse(content={"code": 400, "error": "faltan archivos: 'front' es obligatorio"}, status_code=200)
150
+
151
+ print("front filename:", front.filename, "content_type:", front.content_type)
152
  front_bytes = await front.read()
153
+ print("front size:", len(front_bytes))
154
+
155
  if back is not None:
156
+ print("back filename:", back.filename, "content_type:", back.content_type)
157
  _ = await back.read()
158
+
159
  result = process_image_bytes(front_bytes)
160
  return JSONResponse(content={"code": 200, "data": result})
161
+
162
+ except Exception as e:
163
+ print("EXCEPTION:", repr(e))
164
+ traceback.print_exc()
165
+ return JSONResponse(content={"code": 404, "data": {}, "error": str(e)}, status_code=200)