Copied original content

app.py

@@ -50,7 +50,6 @@ def replace_chars(text, char_mapping=CHAR_MAPPING):
 
 def tokens2int(tokens, numwords={}):
     """ Convert an English str containing number words into an int
-
     >>> text2int("nine")
     9
     >>> text2int("forty two")
@@ -151,7 +150,26 @@ with gr.Blocks() as html_block:
     )
     button_text2int.click(try_text2int, inputs=[textbox_input], outputs=[textbox_output])
     button_text2int_preprocessed.click(try_text2int_preprocessed, inputs=[textbox_input], outputs=[textbox_output])
-
-
-interface = gr.Interface(fn=try_text2int_preprocessed, inputs=[textbox_input], outputs=[textbox_output])
-interface.launch()
+    gr.Markdown(r"""
+## API
+You can select which function to run using the `fn_index` argument:
+```python
+import requests
+requests.post(
+    url="https://cetinca-mathtext-nlu.hf.space/run/predict/run/predict", json={"data": ["one hundred forty-two"], "fn_index": 0}
+).json()
+```
+Or using `curl`:
+```bash
+curl -X POST https://cetinca-mathtext-nlu.hf.space/run/predict/ -H 'Content-Type: application/json' -d '{"data": ["one hundred forty-two"], "fn_index": 0}'
+```
+""" + f"{json.loads(BQ_JSON)['type']}")
+
+interface = gr.Interface(lambda: None, inputs=[textbox_input], outputs=[textbox_output])
+
+html_block.input_components = interface.input_components
+html_block.output_components = interface.output_components
+html_block.examples = None
+html_block.predict_durations = []
+
+bapp = html_block.launch()

app2.py

@@ -0,0 +1,157 @@
+import inspect
+import json
+import logging
+import os
+import gradio as gr
+from gradio import routes
+import spacy  # noqa
+from typing import List, Type
+from dotenv import load_dotenv
+
+load_dotenv()
+
+TOKENS2INT_ERROR_INT = 32202
+
+log = logging.getLogger()
+
+ONES = [
+    "zero", "one", "two", "three", "four", "five", "six", "seven", "eight",
+    "nine", "ten", "eleven", "twelve", "thirteen", "fourteen", "fifteen",
+    "sixteen", "seventeen", "eighteen", "nineteen",
+]
+
+# token_mapping = json.load(open('str_mapping.json'))
+CHAR_MAPPING = {
+    "-": " ",
+    "_": " ",
+}
+CHAR_MAPPING.update((str(i), word) for i, word in enumerate([" " + s + " " for s in ONES]))
+
+TOKEN_MAPPING = dict(enumerate([" " + s + " " for s in ONES]))
+
+BQ_JSON = os.environ['BQ_JSON']
+
+
+def tokenize(text):
+    return text.split()
+
+
+def detokenize(tokens):
+    return ' '.join(tokens)
+
+
+def replace_tokens(tokens, token_mapping=TOKEN_MAPPING):
+    return [token_mapping.get(tok, tok) for tok in tokens]
+
+
+def replace_chars(text, char_mapping=CHAR_MAPPING):
+    return ''.join((char_mapping.get(c, c) for c in text))
+
+
+def tokens2int(tokens, numwords={}):
+    """ Convert an English str containing number words into an int
+
+    >>> text2int("nine")
+    9
+    >>> text2int("forty two")
+    42
+    >>> text2int("1 2 three")
+    123
+    """
+    if not numwords:
+
+        tens = ["", "", "twenty", "thirty", "forty", "fifty", "sixty", "seventy", "eighty", "ninety"]
+
+        scales = ["hundred", "thousand", "million", "billion", "trillion"]
+
+        numwords["and"] = (1, 0)
+        for idx, word in enumerate(ONES):
+            numwords[word] = (1, idx)
+        for idx, word in enumerate(tens):
+            numwords[word] = (1, idx * 10)
+        for idx, word in enumerate(scales):
+            numwords[word] = (10 ** (idx * 3 or 2), 0)
+
+    current = result = 0
+
+    for word in tokens:
+        if word not in numwords:
+            raise Exception("Illegal word: " + word)
+
+        scale, increment = numwords[word]
+        current = current * scale + increment
+        if scale > 100:
+            result += current
+            current = 0
+
+    return str(result + current)
+
+
+def text2int(text):
+    return tokens2int(tokenize(replace_chars(text)))
+
+
+def try_text2int(text):
+    text = str(text)
+    try:
+        intstr = tokens2int(tokens2int(tokenize(replace_chars(text))))
+    except Exception as e:
+        log.error(str(e))
+        log.error(f'User input: {text}')
+        intstr = TOKENS2INT_ERROR_INT
+    return str(intstr)
+
+
+def try_text2int_preprocessed(text):
+    text = str(text)
+    try:
+        tokens = replace_tokens(tokenize(replace_chars(str(text))))
+    except Exception as e:
+        log.error(str(e))
+        tokens = text.split()
+    try:
+        intstr = tokens2int(tokens)
+    except Exception as e:
+        log.error(str(e))
+        intstr = str(TOKENS2INT_ERROR_INT)
+    return intstr
+
+
+def get_types(cls_set: List[Type], component: str):
+    docset = []
+    types = []
+    if component == "input":
+        for cls in cls_set:
+            doc = inspect.getdoc(cls)
+            doc_lines = doc.split("\n")
+            docset.append(doc_lines[1].split(":")[-1])
+            types.append(doc_lines[1].split(")")[0].split("(")[-1])
+    else:
+        for cls in cls_set:
+            doc = inspect.getdoc(cls)
+            doc_lines = doc.split("\n")
+            docset.append(doc_lines[-1].split(":")[-1])
+            types.append(doc_lines[-1].split(")")[0].split("(")[-1])
+    return docset, types
+
+
+routes.get_types = get_types
+
+with gr.Blocks() as html_block:
+    gr.Markdown("# Gradio Blocks (3.0) with REST API")
+    textbox_input = gr.Textbox(
+        value="forty-two",
+        label="Input number words:",
+    )
+    button_text2int = gr.Button("text2int")
+    button_text2int_preprocessed = gr.Button("text2int with preprocessing")
+    textbox_output = gr.Textbox(
+        value="42",
+        label="Output integer:"
+    )
+    button_text2int.click(try_text2int, inputs=[textbox_input], outputs=[textbox_output])
+    button_text2int_preprocessed.click(try_text2int_preprocessed, inputs=[textbox_input], outputs=[textbox_output])
+
+
+interface = gr.Interface(fn=try_text2int_preprocessed, inputs=[textbox_input], outputs=[textbox_output])
+interface.launch()

mathtext-nlu-api.py

@@ -1,18 +1,18 @@
 import os
-from dotenv import load_dotenv
+
 import requests
+from dotenv import load_dotenv
 
 load_dotenv()
 
-# data = input("Enter a number as text:\n").split()
-data = "four two".split()
+data = [input("Enter a number as text:\n")]
 
 HUG_ACC_TOKEN = os.environ.get("HUG_ACC_TOKEN")
 
 response = requests.post(
     url="https://cetinca-mathtext-nlu.hf.space/run/predict",
-    json={"data": data, "fn_index": 0},
+    json={"data": data, "fn_index": 1},
     # headers={'Authorization': HUG_ACC_TOKEN},
-)
+).json()
 
-print(response.status_code)
+print(response.get("data", "No response!"))