from selenium import webdriverdef create_driver(browser_name: str = "chrome"):
b = browser_name.lower().strip()
if b == "firefox":
return webdriver.Firefox()
elif b == "edge":
return webdriver.Edge()
else:
return webdriver.Chrome()