Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
|
|
|
|
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 |
-
|
39 |
-
|
|
|
|
|
|
|
|
|
|
|
40 |
enhancer = Enhancer(request.method, request.background_enhancement, request.upscale)
|
41 |
-
|
42 |
-
|
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 |
-
|
|
|
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 |
|