add redirect to different proxy

app.py

@@ -33,9 +33,11 @@ url = f'https://{space}.hf.space'
 
 def run(hf_token, service, game, functionality, nlp_command):
     # reuse hf_token field as json string
-    token, user_name, _ = extract(hf_token)
+    token, user_name, redirect, _ = extract(hf_token)
     if user_name is None:
         user_name = '__fake__'
+    if redirect is not None:
+        space = redirect
 
     if token is None or token == '':
         return 'please specify hf token'

extract.py

@@ -1,11 +1,15 @@
 import json
 
 def extract(hf_token):
-    info = {}
+    """
+    Extract token, user, redirect, and info from input hf_token.
+    If hf_token is simple, it is the token itself.
+    """
+    info = {} # a copy of hf_token in json format
     try:
         info = json.loads(hf_token)
     except json.decoder.JSONDecodeError:
-        return hf_token, None, None
+        return hf_token, None, None, None
 
     try:
         token = info['token']
@@ -17,4 +21,9 @@ def extract(hf_token):
     except KeyError:
         user = None
 
-    return token, user, info
+    try:
+        redirect = info['redirect']
+    except KeyError:
+        redirect = None
+
+    return token, user, redirect, info

test_extract.py

@@ -5,40 +5,53 @@ from extract import extract
 
 def test_pure_string():
     inp = 'abc'
-    token, user, info = extract(inp)
+    token, user, redirect, info = extract(inp)
     assert token == inp
     assert user is None
+    assert redirect is None
     assert info is None
 
 def test_json_format_01():
     inp_s = '{"a":"b"}'
-    token, user, info = extract(inp_s)
+    token, user, redirect, info = extract(inp_s)
     inp = json.loads(inp_s)
     assert token == None
     assert user == None
+    assert redirect is None
     assert info == inp
 
 def test_json_format_02():
     inp_s = '{"token":"b"}'
-    token, user, info = extract(inp_s)
+    token, user, redirect, info = extract(inp_s)
     inp = json.loads(inp_s)
     assert token == inp['token']
     assert user == None
+    assert redirect is None
     assert info == inp
 
 def test_json_format_03():
     inp_s = '{"user":"xb"}'
-    token, user, info = extract(inp_s)
+    token, user, redirect, info = extract(inp_s)
     inp = json.loads(inp_s)
     assert token == None
     assert user == inp['user']
+    assert redirect is None
     assert info == inp
 
 def test_json_format_04():
     inp_s = '{"user":"t1", "token":"t2"}'
-    token, user, info = extract(inp_s)
+    token, user, redirect, info = extract(inp_s)
     inp = json.loads(inp_s)
     assert token == inp['token']
     assert user == inp['user']
+    assert redirect is None
     assert info == inp
 
+def test_json_format_05():
+    inp_s = '{"user":"t1", "token":"t2", "redirect":"xx"}'
+    token, user, redirect, info = extract(inp_s)
+    inp = json.loads(inp_s)
+    assert token == inp['token']
+    assert user == inp['user']
+    assert redirect == inp['redirect']
+    assert info == inp