Update app.py

app.py

@@ -24,61 +24,61 @@ def my_custom_tool(arg1:str, arg2:str)-> str: #it's import to specify the return
         arg2: the second argument
     """
 
-# Define the agent
-agent = SmolAgent(
-    browser="chrome",  # Uses Chrome WebDriver
-    headless=False,  # Set to True to run in background
-    maximize=True,
-)
+    # Define the agent
+    agent = SmolAgent(
+        browser="chrome",  # Uses Chrome WebDriver
+        headless=False,  # Set to True to run in background
+        maximize=True,
+    )
 
-try:
-    # Step 1: Open Substack Login Page
-    agent.goto("https://tensorhealth.substack.com/")
+    try:
+        # Step 1: Open Substack Login Page
+        agent.goto("https://tensorhealth.substack.com/")
     
-    # Step 2: Click 'Sign in'
-    time.sleep(3)
-    agent.click_text("Sign in")
-
-    # Step 3: Enter Email/Username
-    time.sleep(3)
-    agent.type(USERNAME, By.NAME, "email")
-    agent.press(Keys.RETURN)
-
-    # Step 4: Enter Password
-    time.sleep(5)
-    agent.type(PASSWORD, By.NAME, "password")
-    agent.press(Keys.RETURN)
-
-    # Step 5: Open 'Dashboard' dropdown
-    time.sleep(5)
-    agent.click_text("Dashboard")
-
-    # Step 6: Select 'TensorHealth'
-    time.sleep(2)
-    agent.click_text("TensorHealth")
-
-    # Step 7: Click 'NewPost' dropdown
-    time.sleep(5)
-    agent.click_text("New Post")
-
-    # Step 8: Select 'TextPost'
-    time.sleep(2)
-    agent.click_text("Text Post")
-
-    # Step 9: Paste "Opinion AI"
-    time.sleep(5)
-    agent.type("Opinion AI", By.XPATH, "//div[@contenteditable='true']")
-
-    print("✅ Text successfully entered!")
-
-except Exception as e:
-    print(f"❌ Error occurred: {e}")
-
-finally:
-    # Keep browser open for review
-    time.sleep(10)
-    agent.quit()
-return "Text successfully entered!"  
+        # Step 2: Click 'Sign in'
+        time.sleep(3)
+        agent.click_text("Sign in")
+
+        # Step 3: Enter Email/Username
+        time.sleep(3)
+        agent.type(USERNAME, By.NAME, "email")
+        agent.press(Keys.RETURN)
+
+        # Step 4: Enter Password
+        time.sleep(5)
+        agent.type(PASSWORD, By.NAME, "password")
+        agent.press(Keys.RETURN)
+
+        # Step 5: Open 'Dashboard' dropdown
+        time.sleep(5)
+        agent.click_text("Dashboard")
+
+        # Step 6: Select 'TensorHealth'
+        time.sleep(2)
+        agent.click_text("TensorHealth")
+
+        # Step 7: Click 'NewPost' dropdown
+        time.sleep(5)
+        agent.click_text("New Post")
+
+        # Step 8: Select 'TextPost'
+        time.sleep(2)
+        agent.click_text("Text Post")
+
+        # Step 9: Paste "Opinion AI"
+        time.sleep(5)
+        agent.type("Opinion AI", By.XPATH, "//div[@contenteditable='true']")
+
+        print("✅ Text successfully entered!")
+
+    except Exception as e:
+        print(f"❌ Error occurred: {e}")
+
+    finally:
+        # Keep browser open for review
+        time.sleep(10)
+        agent.quit()
+    return "Text successfully entered!"  
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.