Spaces:
Running
Running
File size: 1,004 Bytes
dd8b826 b9c182b dd8b826 de348fb dd8b826 b9c182b dd8b826 b9c182b dd8b826 de348fb dd8b826 b9c182b dd8b826 de348fb b9c182b dd8b826 b9c182b dd8b826 de348fb dd8b826 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 | # Based on: https://github.com/allenai/s2-folks/tree/main/examples/python
#
#
import wormneuroatlas as wa
from pprint import PrettyPrinter
FORMAT_TEXT = "Text"
FORMAT_DICT = "Dict"
FORMAT_JSON = "JSON"
FORMATS = (FORMAT_TEXT, FORMAT_DICT)
DS_WORMNEUROATLAS = "WormNeuroAtlas"
pp = PrettyPrinter(depth=6)
def query_data_source(text, source, format):
if source == DS_WORMNEUROATLAS:
atlas = wa.NeuroAtlas()
ds, ss = atlas.everything_about(
text, return_values=True, return_text=True, print_text=False
)
if format == FORMAT_TEXT:
info = """Extracting info on neuron %s from %s...
""" % (
text,
source,
)
info += """```%s```""" % ss
elif format == FORMAT_DICT:
info = """```
%s```""" % pp.pformat(ds)
elif format == FORMAT_JSON:
import json
jj = json.dumps(ds)
info = """```
%s```""" % pp.pformat(jj)
return info
|