Update app.py

app.py

@@ -33,14 +33,16 @@ auth_button_text = '''<a href="{}"><button class="w3-button w3-light-grey w3-pad
 
 
 
-def client_log_in():
+def client_log_in(provided_server = None):
     '''
      Generates an authenticated Masterdon client for the admin user. Used for retireving the target website URL
     '''
 
-
-    global slider_choice
-    server = slider_choice
+    if provided_server == None:
+        global slider_choice
+        server = slider_choice
+    else:
+        server = provided_server
 
     if server != None:
         id = "WatchTower Ivory | Mastodon"
@@ -348,7 +350,7 @@ def update_target_website():
         value=auth_button_text.format(
             get_target_website()))
 
-    return auth_button_text.format(list_of_servers[0])
+    return auth_button_text.format(get_target_website(list_of_servers[0]))
 
 
 # The below is a JS blob used to retrieve the URL params.
@@ -375,13 +377,13 @@ def get_chatbot_text():
     return [('Welcome to Watchtower.', 'Log in via Mastodon and configure your blocking options above.')]
 
 
-def get_target_website():
+def get_target_website(provided_server = None):
     '''
     A wrapper function used for retrieving the URL a user will use to authenticate WatchTower with Mastodon.
     :return: auth url
     '''
 
-    mastodon = client_log_in()
+    mastodon = client_log_in(provided_server)
     return get_auth_url(mastodon)