Update app.py

app.py

@@ -4,14 +4,17 @@ import requests
 import pytz
 import yaml
 import asyncio
-#import nest_asyncio
+import nest_asyncio
 from playwright.async_api import async_playwright
 from PIL import Image
+import gradio as gr
 
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
+nest_asyncio.apply()  # Ensure async works in a Jupyter/Colab/HF Spaces environment
+
 subprocess.run(["playwright", "install"])
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
@@ -53,22 +56,25 @@ async def capture_screenshot():
         print("Screenshot saved!")
         
 @tool
-def grab_image() -> img:
-     """
-    Fetches an op art image from a p5 sketch.
-    This function sends uses Playwright to launch a headles server and grab a screen shot of a p5.sketch.
-   
+def grab_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.
+
     Returns:
-        image: img
+        PIL.Image: Captured image
     """
-    loop = asyncio.get_event_loop()
-    future = asyncio.run_coroutine_threadsafe(capture_screenshot(), loop)
-    future.result()  # Wait for coroutine to finish
+    print("Running async Playwright screenshot...")
     
+    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())
+
     print("Loading image for Gradio...")
     img = Image.open("img.png")
     return img
 
+
 #https://github.com/huggingface/smolagents/blob/main/examples/multiple_tools.py
 @tool
 def get_joke() -> str: