Update app.py

app.py

@@ -64,18 +64,19 @@ def predict(image):
                     print(f"Error decoding base64 image: {e}")
                     return json.dumps({"error": f"Error decoding base64 image: {e}"})
 
-            # Try to fetch the image from a URL
-            elif isinstance(image, str):
-                if image.startswith("http://") or image.startswith("https://"):
-                    try:
-                        response = requests.get(image)
-                        image = Image.open(BytesIO(response.content))
-                        print(f"Fetched image from URL: {image}")
-                    except Exception as e:
-                        print(f"Error fetching image from URL: {e}")
-                        return json.dumps({"error": f"Error fetching image from URL: {e}"})
-                else:
-                    return json.dumps({"error": "Invalid URL format. Please provide a valid URL starting with 'http://' or 'https://'."})
+            # Check if the input is a URL
+        if isinstance(image_input, str):
+            if image_input.startswith("http://") or image_input.startswith("https://"):
+                try:
+                    response = requests.get(image_input)
+                    response.raise_for_status()  # Check for HTTP errors
+                    image = Image.open(BytesIO(response.content))
+                    print(f"Fetched image from URL: {image}")
+                except Exception as e:
+                    print(f"Error fetching image from URL: {e}")
+                    return json.dumps({"error": f"Error fetching image from URL: {e}"})
+            else:
+                return json.dumps({"error": "Invalid URL format. Please provide a valid URL starting with 'http://' or 'https://'."})
 
             # Try to load the image from a local file path
             elif isinstance(image, str) and os.path.isfile(image):