Update app.py

app.py

@@ -57,11 +57,9 @@ async def capture_screenshot(image_type: str):
         image_url
         if image_type: 
             image_url = get_image_url(image_type)
-        elif:
-            image_url = url
         else:
-            Return: f"Unable to fetch image"
-
+            image_url = url
+        
         print(f"Opening image from p5 sketch: {image_url}")
         await page.goto(image_url, timeout=120000)  # Wait for the image page to load