mathtext / app.py
hobs
rename vars and fix docs
20c813e
raw
history blame
3.4 kB
import gradio as gr
import inspect
from gradio import routes
from typing import List, Type
"""
>>> !curl -X POST https://Hobson-gradio-rest-api.hf.space/api/predict/ -H 'Content-Type: application/json' -d '{"data": ["one hundred twenty-three"], "fn_index": 0}'
{"data":["123"],"duration":0.00019359588623046875,"average_duration":0.00019359588623046875}
"""
def normalize_text(text):
return text.replace('-', ' ')
def text2int(text, 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:
units = [
"zero", "one", "two", "three", "four", "five", "six", "seven", "eight",
"nine", "ten", "eleven", "twelve", "thirteen", "fourteen", "fifteen",
"sixteen", "seventeen", "eighteen", "nineteen",
]
tens = ["", "", "twenty", "thirty", "forty", "fifty", "sixty", "seventy", "eighty", "ninety"]
scales = ["hundred", "thousand", "million", "billion", "trillion"]
numwords["and"] = (1, 0)
for idx, word in enumerate(units):
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
text = normalize_text(text)
for word in text.split():
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 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 = gr.Textbox()
button = gr.Button("text2int")
output = gr.Textbox()
button.click(text2int, inputs=[textbox], outputs=[output])
gr.Markdown(r"""
## API
You can select which function to run using the `fn_index` argument:
```python
import requests
requests.post(
url="https://Hobson-gradio-rest-api.hf.space/api/predict/", json={"data": ["one hundred forty-two"], "fn_index": 0}
).json()
```
Or using `curl`:
```bash
curl -X POST https://Hobson-gradio-rest-api.hf.space/api/predict/ -H 'Content-Type: application/json' -d '{"data": ["one hundred forty-two"], "fn_index": 0}'
```
""")
interface = gr.Interface(lambda: None, inputs=[textbox], outputs=[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()