Update app.py

app.py

@@ -1,32 +1,32 @@
 import os
-import glob
-import asyncio
 import argparse
 import logging
 from selenium import webdriver
 from selenium.webdriver.chrome.service import Service
-from selenium.webdriver.common.by import By
 from selenium.webdriver.chrome.options import Options
 from webdriver_manager.chrome import ChromeDriverManager
 import gradio as gr
 
-# Logger setup
-logger = logging.getLogger(__name__)
+def install_chrome():
+    os.system("apt update && apt install -y curl")
+    os.system("curl -fsSL https://dl.google.com/linux/linux_signing_key.pub | gpg --dearmor -o /usr/share/keyrings/google-chrome-keyring.gpg")
+    os.system("echo 'deb [signed-by=/usr/share/keyrings/google-chrome-keyring.gpg] http://dl.google.com/linux/chrome/deb/ stable main' | tee /etc/apt/sources.list.d/google-chrome.list")
+    os.system("apt update && apt install -y google-chrome-stable")
+
+install_chrome()
 
-# Global variables for browser persistence
+logger = logging.getLogger(__name__)
 _global_browser = None
 
 def setup_browser(headless=True, window_size=(1280, 720)):
-    """Initialize a Selenium browser instance."""
     global _global_browser
-
     chrome_options = Options()
     if headless:
         chrome_options.add_argument("--headless")
     chrome_options.add_argument("--no-sandbox")
     chrome_options.add_argument("--disable-dev-shm-usage")
     chrome_options.add_argument(f"--window-size={window_size[0]},{window_size[1]}")
-    chrome_options.binary_location = "/usr/bin/google-chrome-stable"  # Ruta correcta en Hugging Face
+    chrome_options.binary_location = "/usr/bin/google-chrome-stable"
 
     service = Service(ChromeDriverManager().install())
     _global_browser = webdriver.Chrome(service=service, options=chrome_options)
@@ -34,54 +34,33 @@ def setup_browser(headless=True, window_size=(1280, 720)):
     return _global_browser
 
 def close_browser():
-    """Close the Selenium browser instance."""
     global _global_browser
     if _global_browser:
         _global_browser.quit()
         _global_browser = None
 
 async def run_agent(task_description, headless=True, window_size=(1280, 720)):
-    """Run a Selenium-based agent."""
     global _global_browser
-
     if not _global_browser:
         setup_browser(headless=headless, window_size=window_size)
-
     try:
         logger.info(f"Executing task: {task_description}")
-        _global_browser.get("https://example.com")  # Replace with your target URL
-
-        # Simulate some task
+        _global_browser.get("https://example.com")
         result = f"Task executed successfully on {task_description}"
-        return result, None  # Return result and errors
+        return result, None
     except Exception as e:
         logger.error(f"Error while running agent: {e}")
-        return None, str(e)  # Return None result and error message
+        return None, str(e)
 
 def create_ui():
-    """Create the Gradio UI."""
     with gr.Blocks() as demo:
         gr.Markdown("# Selenium-based Browser Automation")
-
-        task_input = gr.Textbox(
-            label="Task Description",
-            placeholder="Describe the task you want to execute...",
-        )
-
-        headless_mode = gr.Checkbox(
-            label="Run in Headless Mode", value=True, interactive=True
-        )
-
+        task_input = gr.Textbox(label="Task Description", placeholder="Describe the task you want to execute...")
+        headless_mode = gr.Checkbox(label="Run in Headless Mode", value=True, interactive=True)
         run_button = gr.Button("Run Task")
         output_result = gr.Textbox(label="Result")
         output_error = gr.Textbox(label="Error")
-
-        run_button.click(
-            run_agent,
-            inputs=[task_input, headless_mode],
-            outputs=[output_result, output_error],
-        )
-
+        run_button.click(run_agent, inputs=[task_input, headless_mode], outputs=[output_result, output_error])
     return demo
 
 def main():
@@ -89,7 +68,6 @@ def main():
     parser.add_argument("--ip", type=str, default="0.0.0.0", help="IP address to bind to")
     parser.add_argument("--port", type=int, default=7860, help="Port to listen on")
     args = parser.parse_args()
-
     demo = create_ui()
     demo.launch(server_name="0.0.0.0", server_port=args.port)