Update app.py

app.py

@@ -17,20 +17,26 @@ except Exception as e:
 
 def process_image(prompt, image, style, upscale_factor, inpaint):
     try:
-        # Debug the type of image received
+        # Debug the type and content of image received
+        if image is None:
+            return None, "No image received."
+
         print(f"Received image type: {type(image)}")
+        
+        if isinstance(image, np.ndarray):
+            print(f"Image shape: {image.shape}")
+            image = Image.fromarray(image)
+        elif isinstance(image, torch.Tensor):
+            print(f"Image tensor shape: {image.shape}")
+            image = transforms.ToPILImage()(image)
+        elif isinstance(image, Image.Image):
+            print("Image is already in PIL format.")
+        else:
+            return None, f"Unsupported image format: {type(image)}."
 
-        # Ensure the image is in the correct format
-        if image is not None:
-            if isinstance(image, np.ndarray):
-                image = Image.fromarray(image)
-            elif isinstance(image, torch.Tensor):
-                image = transforms.ToPILImage()(image)
-            elif isinstance(image, Image.Image):
-                # Image is already in the correct format
-                pass
-            else:
-                return None, f"Error: Unsupported image format. Received {type(image)}."
+        # Check if the image is valid
+        if not isinstance(image, Image.Image):
+            return None, "Error: Image format conversion failed."
 
         # Log the input parameters
         print(f"Prompt: {prompt}")