Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
-from playwright.sync_api import sync_playwright
+from playwright.async_api import async_playwright
 import json
 
 # Define model and inference parameters
@@ -22,41 +22,41 @@ def generate_actions(input_text):
     actions = response.split("\n")
     return actions
 
-# Function to initialize browser and page
-def initialize_browser():
-    with sync_playwright() as p:
-        browser = p.chromium.launch()
-        page = browser.new_page()
+# Function to initialize browser and page (now asynchronous)
+async def initialize_browser():
+    async with async_playwright() as p:
+        browser = await p.chromium.launch()
+        page = await browser.new_page()
         return browser, page
 
-# Gradio interface
-def run_agent(input_text):
-    with sync_playwright() as p:
-        browser, page = initialize_browser()
+# Gradio interface (now using the asynchronous function)
+async def run_agent(input_text):
+    async with async_playwright() as p:
+        browser, page = await initialize_browser()
         actions = generate_actions(input_text)
 
         for action in actions:
             if "open website" in action:
                 website = action.split(" ")[-1]
-                page.goto(website)
+                await page.goto(website)
             elif "click" in action:
                 selector = action.split(" ")[-1]
-                page.click(selector)
+                await page.click(selector)
             elif "type" in action:
                 text = action.split(" ")[-1]
-                page.type(text)
+                await page.type(text)
             elif "submit" in action:
-                page.press("Enter")
+                await page.press("Enter")
             else:
                 print(f"Action not recognized: {action}")
 
         return f"Successfully executed actions based on: {input_text}"
 
 iface = gr.Interface(
-    fn=run_agent,
+    fn=run_agent,  # Pass the async function directly
     inputs=gr.Textbox(label="Enter your request"),
     outputs=gr.Textbox(label="Response"),
     title="Automated Agent",
     description="Enter a task or instruction for the agent to perform."
 )
-iface.launch()
+iface.launch(share=True)  # Enable sharing