File size: 2,872 Bytes
5ae5dcd a95df60 d33b7b4 5ae5dcd a95df60 5ae5dcd cf85d3b d33b7b4 cf85d3b 5ae5dcd d33b7b4 cf85d3b a95df60 cf85d3b a95df60 d33b7b4 a95df60 d33b7b4 a95df60 cf85d3b 95e6d1e d33b7b4 a95df60 d33b7b4 cf85d3b d33b7b4 a95df60 |
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 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
import gradio as gr
import json
from pathlib import Path
import re
from bs4 import BeautifulSoup
import base64
fileid2json = json.loads(Path("fileid2json.json").read_text())
fileid2image = json.loads(Path("fileid2image.json").read_text())
autov22fileid = json.loads(Path("autov22fileid.json").read_text())
filename2fileid = json.loads(Path("filename2fileid.json").read_text())
name2fileid = json.loads(Path("name2fileid.json").read_text())
def fileids2html(fileids, query):
html = BeautifulSoup("<div></div>", "html.parser")
for fileid in sorted(list(fileids), reverse=True):
src = "https://huggingface.co./front/assets/huggingface_logo-noborder.svg"
if fileid in fileid2image:
src = fileid2image[fileid.strip()]
div = html.new_tag("div", style="height:16rem;")
h1 = html.new_tag("h1")
a = html.new_tag("a", target="_blank")
a.append(str(fileid))
h1.append(a)
textarea = html.new_tag("textarea", dir="ltr", style='color:var(--body-text-color);display:inline-block;overflow:auto;width:calc(100% - 256px);height:100%;background:var(--block-background-fill);')
html.append(h1)
html.append(div)
div.append(textarea)
div.append(html.new_tag("img", src=src, style='display:inline-block;max-width:256px;height:100%;vertical-align:initial;'))
if fileid in fileid2json:
j = fileid2json[fileid.strip()]
a["href"] = j["notes"]
textarea.append(json.dumps(j, indent=2, ensure_ascii=False))
j = base64.b64encode(json.dumps(j, indent=2, ensure_ascii=False).encode()).decode()
a2 = html.new_tag("a", download="{}.json".format(query), href="data:application/json;base64,{}".format(j), style="margin-left:3rem;")
a2.append("Save JSON")
h1.append(a2)
html.append(html.new_tag("hr", style="margin-top:1rem;margin-bottom:1rem;"))
return str(html)
def greet(query):
fileids = set()
fileid = query
hit = None
if query.upper() in autov22fileid:
fileid = str(autov22fileid[query.upper()])
fileids.add(fileid)
hit = True
if re.sub(r'\..*$', "", query) in filename2fileid:
fileid = str(filename2fileid[re.sub(r'\..*$', "", query)])
fileids.add(fileid)
hit = True
if query in name2fileid:
fileid = str(name2fileid[query])
fileids.add(fileid)
hit = True
if hit is not True:
for k, v in [(k.lower(), v) for k, v in name2fileid.items()]:
if re.search(re.compile(query), k):
fileid = str(v)
fileids.add(fileid)
return fileids2html(fileids, query)
iface = gr.Interface(fn=greet, inputs="text", outputs="html", allow_flagging='never', css='#component-4 { max-width: 16rem; }')
iface.launch(server_name="0.0.0.0")
|