new UI

app.py

@@ -2,50 +2,51 @@ import requests
 import json
 from datetime import datetime
 
-import gradio
+import gradio as gr
 from gradio_client import Client
 
-function_list1 = [
-    "angry birds",
-    'snake',
-    "greet", 
-    "list AI backend functionality",
-    "show json",
-    "game"
-    # "flappy birds",
+functionality_shared = [
+    'games',
+    'greet',
+    'tokenization',
 ]
-function_list1 = [
-    "angry birds",
+
+game_list = [
+    'angry birds',
     'snake',
 ]
-function_list2 = [
-    "greet", 
+
+functionality_list = [
+    "AI",
     "list AI backend functionality",
     "show json",
-    "game"
-    # "flappy birds",
 ]
-# function_list.pop()
 
-space = 'stevez-abc123'
+space = 'stevez-bbc123'
 # space = 'stevez-test-public'
 
 # url = f'https://{space}.hf.space/run/predict'
 # url = f'https://{space}.hf.space/api/predict'
 url = f'https://{space}.hf.space'
 
-def run(hf_token, game, r, nlp_command):
+def run(hf_token, service, game, r, nlp_command):
     if hf_token is None or hf_token == '':
         return 'please specify hf token'
-    if game is None:
-        return "please specify which game"
-    if r is None:
-        return "please choose the AI functionality"
+
+    if service not in functionality_shared[1:]:
+        if game is None:
+            return "please specify which game"
+        if r is None:
+            return "please choose the AI functionality"
+        if r == 'game':
+            if nlp_command in ["", None]:
+                return "please make sure the command is not empty"
+
     try:
         print(f"{datetime.today()}\t{nlp_command}")
-        myobj = {
-            'data': [game, r, nlp_command],
-        }
+        # myobj = {
+        #     'data': [service, game, r, nlp_command],
+        # }
         # myauth = ('Bearer', hf_token)
         # x = requests.post(
         #     url, 
@@ -61,7 +62,7 @@ def run(hf_token, game, r, nlp_command):
             hf_token=hf_token,
         )
         res = client.predict(
-            r, nlp_command,
+            service, game, r, nlp_command,
             api_name='/predict',
         )
         res2 = json.loads(res[0])
@@ -72,22 +73,22 @@ def run(hf_token, game, r, nlp_command):
     except Exception as e:
         return f"{type(e)}, {str(e)}. \nyou many want to make sure your hf_token is correct"
     
-demo = gradio.Interface(
+demo = gr.Interface(
     fn=run, 
     inputs=[
         "text",
-        gradio.Radio(
-            function_list1,
-            value='angry birds',
-            label = "Which game?",
-            # info = "Greet for testing server",
-            info = "Only 'angry birds' is implemented as of now",
+        gr.Radio(
+            functionality_shared,
+            info = "Shared services",
+            ),
+        gr.Radio(
+            game_list,
+            info = "Which game you want the AI to support?",
             ),
-        gradio.Radio(
-            function_list2,
-            value='game',
-            label = "What do you want to do?",
-            info = "Greet for testing server",
+        gr.Radio(
+            functionality_list,
+            # label = "What do you want to do?",
+            info = "What functionality?",
             ),
         "text"],
     outputs="text",