d0tpy commited on
Commit
1b399de
·
verified ·
1 Parent(s): 62e3f73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -9
app.py CHANGED
@@ -27,28 +27,35 @@ def greet_json():
27
  @app.post("/enhance/image/")
28
  async def enhance_image(
29
  file: UploadFile = File(...),
30
- request: EnhancementRequest = EnhancementRequest()
 
 
31
  ):
32
  try:
33
  if not file.content_type.startswith('image/'):
34
  raise HTTPException(status_code=400, detail="Invalid file type")
35
-
36
  contents = await file.read()
37
  base64_encoded_image = base64.b64encode(contents).decode('utf-8')
38
- data = _EnhanceBase(encoded_base_img=[base64_encoded_image])
39
-
 
 
 
 
 
40
  enhancer = Enhancer(request.method, request.background_enhancement, request.upscale)
41
-
42
- enhanced_img, original_resolution, enhanced_resolution = await enhancer.enhance(data)
43
-
44
  enhanced_image = Image.fromarray(enhanced_img)
45
  img_byte_arr = BytesIO()
46
  enhanced_image.save(img_byte_arr, format='PNG')
47
  img_byte_arr.seek(0)
48
- print(original_resolution, enhanced_resolution)
 
49
 
50
  return StreamingResponse(img_byte_arr, media_type="image/png")
51
-
52
  except Exception as e:
53
  raise HTTPException(status_code=500, detail=str(e))
54
 
 
27
  @app.post("/enhance/image/")
28
  async def enhance_image(
29
  file: UploadFile = File(...),
30
+ method: EnhancementMethod = EnhancementMethod.gfpgan,
31
+ background_enhancement: bool = True,
32
+ upscale: int = 2
33
  ):
34
  try:
35
  if not file.content_type.startswith('image/'):
36
  raise HTTPException(status_code=400, detail="Invalid file type")
37
+
38
  contents = await file.read()
39
  base64_encoded_image = base64.b64encode(contents).decode('utf-8')
40
+
41
+ request = EnhancementRequest(
42
+ method=method,
43
+ background_enhancement=background_enhancement,
44
+ upscale=upscale
45
+ )
46
+
47
  enhancer = Enhancer(request.method, request.background_enhancement, request.upscale)
48
+ enhanced_img, original_resolution, enhanced_resolution = await enhancer.enhance(base64_encoded_image)
49
+
 
50
  enhanced_image = Image.fromarray(enhanced_img)
51
  img_byte_arr = BytesIO()
52
  enhanced_image.save(img_byte_arr, format='PNG')
53
  img_byte_arr.seek(0)
54
+
55
+ print(f"Original resolution: {original_resolution}, Enhanced resolution: {enhanced_resolution}")
56
 
57
  return StreamingResponse(img_byte_arr, media_type="image/png")
58
+
59
  except Exception as e:
60
  raise HTTPException(status_code=500, detail=str(e))
61