Update app.py

start adding more options

app.py

@@ -8,6 +8,7 @@ import nest_asyncio
 from playwright.async_api import async_playwright
 from PIL import Image
 import subprocess
+import json
 import gradio as gr
 
 from tools.final_answer import FinalAnswerTool
@@ -35,18 +36,17 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     return "What magic will you build ?"
 
 
-def chooseImage(type: str)-> str:
-    """Returns a url based on a person's reuqested type of image.
-    Args:
-        type: the type of image the person would like to see
-    Returns:
-        str: a url
-    """
-    
-    
+def load_image_sources():
+    with open("image_sources.json", "r") as file:
+        return json.load(file)
+
+image_sources = load_image_sources()
 
-async def capture_screenshot():
-    """Launches Playwright and captures a screenshot of an image from p5.js."""
+def get_image_url(image_type: str):
+    return image_sources.get(image_type, None)  
+
+async def capture_screenshot(image_type: str):
+    """Launches Playwright and uses user input,  if any, to captures a screenshot of an image from p5.js."""
     print("Launching Playwright...")
     async with async_playwright() as p:
         browser = await p.chromium.launch(headless=True)
@@ -54,7 +54,13 @@ async def capture_screenshot():
 
         #url = "https://openprocessing.org/sketch/2539973"
         url = "https://editor.p5js.org/kfahn/full/2XD5Y8MiV"
-        image_url = url
+        image_url
+        if (image_type): 
+            image_url = get_image_url()
+        elif:
+            image_url = url
+        else:
+            Return: f"Unable to fetch image"
 
         print(f"Opening image from p5 sketch: {image_url}")
         await page.goto(image_url, timeout=120000)  # Wait for the image page to load