Update app.py
Browse files
app.py
CHANGED
|
@@ -30,7 +30,6 @@ def setup_browser(headless=True, window_size=(1280, 720)):
|
|
| 30 |
|
| 31 |
return _global_browser
|
| 32 |
|
| 33 |
-
|
| 34 |
def close_browser():
|
| 35 |
"""Close the Selenium browser instance."""
|
| 36 |
global _global_browser
|
|
@@ -38,7 +37,6 @@ def close_browser():
|
|
| 38 |
_global_browser.quit()
|
| 39 |
_global_browser = None
|
| 40 |
|
| 41 |
-
|
| 42 |
async def run_agent(task_description, headless=True, window_size=(1280, 720)):
|
| 43 |
"""Run a Selenium-based agent."""
|
| 44 |
global _global_browser
|
|
@@ -57,7 +55,6 @@ async def run_agent(task_description, headless=True, window_size=(1280, 720)):
|
|
| 57 |
logger.error(f"Error while running agent: {e}")
|
| 58 |
return None, str(e) # Return None result and error message
|
| 59 |
|
| 60 |
-
|
| 61 |
def create_ui():
|
| 62 |
"""Create the Gradio UI."""
|
| 63 |
with gr.Blocks() as demo:
|
|
@@ -84,17 +81,14 @@ def create_ui():
|
|
| 84 |
|
| 85 |
return demo
|
| 86 |
|
| 87 |
-
|
| 88 |
def main():
|
| 89 |
parser = argparse.ArgumentParser(description="Gradio UI for Selenium Automation")
|
| 90 |
-
parser.add_argument("--ip", type=str, default="
|
| 91 |
parser.add_argument("--port", type=int, default=7860, help="Port to listen on")
|
| 92 |
args = parser.parse_args()
|
| 93 |
|
| 94 |
demo = create_ui()
|
| 95 |
-
demo.launch(server_name=
|
| 96 |
-
|
| 97 |
|
| 98 |
if __name__ == "__main__":
|
| 99 |
main()
|
| 100 |
-
|
|
|
|
| 30 |
|
| 31 |
return _global_browser
|
| 32 |
|
|
|
|
| 33 |
def close_browser():
|
| 34 |
"""Close the Selenium browser instance."""
|
| 35 |
global _global_browser
|
|
|
|
| 37 |
_global_browser.quit()
|
| 38 |
_global_browser = None
|
| 39 |
|
|
|
|
| 40 |
async def run_agent(task_description, headless=True, window_size=(1280, 720)):
|
| 41 |
"""Run a Selenium-based agent."""
|
| 42 |
global _global_browser
|
|
|
|
| 55 |
logger.error(f"Error while running agent: {e}")
|
| 56 |
return None, str(e) # Return None result and error message
|
| 57 |
|
|
|
|
| 58 |
def create_ui():
|
| 59 |
"""Create the Gradio UI."""
|
| 60 |
with gr.Blocks() as demo:
|
|
|
|
| 81 |
|
| 82 |
return demo
|
| 83 |
|
|
|
|
| 84 |
def main():
|
| 85 |
parser = argparse.ArgumentParser(description="Gradio UI for Selenium Automation")
|
| 86 |
+
parser.add_argument("--ip", type=str, default="0.0.0.0", help="IP address to bind to")
|
| 87 |
parser.add_argument("--port", type=int, default=7860, help="Port to listen on")
|
| 88 |
args = parser.parse_args()
|
| 89 |
|
| 90 |
demo = create_ui()
|
| 91 |
+
demo.launch(server_name="0.0.0.0", server_port=args.port, share=True)
|
|
|
|
| 92 |
|
| 93 |
if __name__ == "__main__":
|
| 94 |
main()
|
|
|