Update fluxai.py

fluxai.py

@@ -13,7 +13,7 @@ from models import *
 class FluxAI(BaseModel):
     user_id: int
     args: str
-    auto_enhancer: bool
+    auto_enhancer: bool = False
 
 router = APIRouter()
 
@@ -65,16 +65,16 @@ async def fluxai_image(payload: FluxAI):
                 )
             
             if payload.enhancer:
-                image = Image.open(io.BytesIO(image_bytes))
-                enhancer = ImageEnhance.Sharpness(image)
-                image = enhancer.enhance(1.5)
-                enhancer = ImageEnhance.Contrast(image)
-                image = enhancer.enhance(1.2)
-                enhancer = ImageEnhance.Color(image)
-                image = enhancer.enhance(1.1)
-                enhanced_image_bytes = io.BytesIO()
-                image.save(enhanced_image_bytes, format="JPEG", quality=95)
-                enhanced_image_bytes.seek(0)
+                with Image.open(io.BytesIO(image_bytes)) as image:
+                    enhancer = ImageEnhance.Sharpness(image)
+                    image = enhancer.enhance(1.5)
+                    enhancer = ImageEnhance.Contrast(image)
+                    image = enhancer.enhance(1.2)
+                    enhancer = ImageEnhance.Color(image)
+                    image = enhancer.enhance(1.1)
+                    enhanced_image_bytes = io.BytesIO()
+                    image.save(enhanced_image_bytes, format="JPEG", quality=95)
+                    enhanced_image_bytes.seek(0)
                 return StreamingResponse(enhanced_image_bytes, media_type="image/jpeg")
             else:
                 return StreamingResponse(io.BytesIO(image_bytes), media_type="image/jpeg")