Update app.py

add

@tool
decorator which got accidentally deleted

app.py

@@ -71,11 +71,13 @@ async def capture_screenshot(image_type: str):
         await browser.close()
         print("Screenshot saved!")
      
-
-def grab_image() -> Image:
+@tool
+def grab_image(image_type: str) -> Image:
     """
     Fetches an op art image from a p5.js sketch.
     This function sends uses Playwright to launch a headless server and grab a screenshot of a p5.js sketch.
+    Args:
+        image_type: A geneerate art type
     Returns:
         image: The screen shot of the p5.js sketch as an image.
     """
@@ -83,7 +85,7 @@ def grab_image() -> Image:
     
     loop = asyncio.new_event_loop()  # Create a new event loop (Fixes issues on HF Spaces)
     asyncio.set_event_loop(loop)
-    loop.run_until_complete(capture_screenshot())
+    loop.run_until_complete(capture_screenshot(image_type))
 
     print("Loading image for Gradio...")
     img = Image.open("img.png")