Update app.py

app.py

@@ -27,28 +27,35 @@ def greet_json():
 @app.post("/enhance/image/")
 async def enhance_image(
     file: UploadFile = File(...),
-    request: EnhancementRequest = EnhancementRequest()
+    method: EnhancementMethod = EnhancementMethod.gfpgan,
+    background_enhancement: bool = True,
+    upscale: int = 2
 ):
     try:
         if not file.content_type.startswith('image/'):
             raise HTTPException(status_code=400, detail="Invalid file type")
-            
+        
         contents = await file.read()
         base64_encoded_image = base64.b64encode(contents).decode('utf-8')
-        data = _EnhanceBase(encoded_base_img=[base64_encoded_image])
-
+        
+        request = EnhancementRequest(
+            method=method,
+            background_enhancement=background_enhancement,
+            upscale=upscale
+        )
+        
         enhancer = Enhancer(request.method, request.background_enhancement, request.upscale)
-
-        enhanced_img, original_resolution, enhanced_resolution = await enhancer.enhance(data)
-
+        enhanced_img, original_resolution, enhanced_resolution = await enhancer.enhance(base64_encoded_image)
+        
         enhanced_image = Image.fromarray(enhanced_img)
         img_byte_arr = BytesIO()
         enhanced_image.save(img_byte_arr, format='PNG')
         img_byte_arr.seek(0)
-        print(original_resolution, enhanced_resolution)
+        
+        print(f"Original resolution: {original_resolution}, Enhanced resolution: {enhanced_resolution}")
         
         return StreamingResponse(img_byte_arr, media_type="image/png")
-
+    
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))