Refactor FluxImg2ImgReq class to make image and strength attributes required

app.py

@@ -117,15 +117,15 @@ class FluxReq(BaseModel):
 
 
 class FluxImg2ImgReq(FluxReq):
-    image: Optional[Image.Image]
-    strength: Optional[float] = 1.0
+    image: Image.Image
+    strength: float = 1.0
     
     class Config:
         arbitrary_types_allowed=True
 
 
 class FluxInpaintReq(FluxImg2ImgReq):
-    mask_image: Optional[Image.Image]
+    mask_image: Image.Image
     
     class Config:
         arbitrary_types_allowed=True
@@ -563,12 +563,13 @@ def generate_image(
             mask_image = inpaint_image['layers'][0] if image else None
             strength = float(inpaint_strength)
             
-            base_args = FluxInpaintReq(
-                **base_args.__dict__,
-                image=image,
-                mask_image=mask_image,
-                strength=strength
-            )
+            if image and mask_image:
+                base_args = FluxInpaintReq(
+                    **base_args.__dict__,
+                    image=image,
+                    mask_image=mask_image,
+                    strength=strength
+                )
         elif any([canny_image, pose_image, depth_image]):
             base_args.controlnet_config = ControlNetReq(
                 controlnets=[],