getenv

games/game_angrybird.py

@@ -1,3 +1,7 @@
+import os
+import json
+import requests
+
 from games.angrybirds.angrybirds import angrybird
 
 
@@ -5,5 +9,15 @@ game = angrybird()
 
 
 def process(command: str):
-    res1, _ = game.query_entry_point(command)
-    return res1
+    url = f'https://{os.getenv("hfs_space")}.hf.space/games/angrybird'
+    # url = 'http://127.0.0.1:8080/nlp'
+    myobj = {
+        'command': command,
+    }
+    headers = {'Authorization': f'Bearer {os.getenv("hfs_token")}'}
+
+    x = requests.post(url, json=myobj, headers=headers)
+    print(x.text)
+
+    # res1, _ = game.query_entry_point(command)
+    return json.loads(x.text)

games/game_snake.py

@@ -1,19 +1,20 @@
+import os
 import json
 import requests
-import urllib.parse
 
 from games.snake.snake import snake
 
+
 game = snake()
 
-def process(token: str, command: str):
-    url = 'https://steveagi-pgtrain.hf.space/games/snake'
+
+def process(command: str):
+    url = f'https://{os.getenv("hfs_space")}.hf.space/games/snake'
     # url = 'http://127.0.0.1:8080/nlp'
     myobj = {
         'command': command,
     }
-    auth_token = token
-    headers = {'Authorization': f'Bearer {auth_token}'}
+    headers = {'Authorization': f'Bearer {os.getenv("hfs_token")}'}
 
     x = requests.post(url, json=myobj, headers=headers)
     print(x.text)