複数の検索結果表示に対応

app.py

@@ -2,6 +2,8 @@ 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())
@@ -9,33 +11,61 @@ 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):
-    result = "Not found"
+    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)
 
-    if fileid in fileid2json:
-        result = json.dumps(fileid2json[fileid.strip()])
-    src = "https://huggingface.co/front/assets/huggingface_logo-noborder.svg"
-    if fileid in fileid2image:
-        src = fileid2image[fileid.strip()]
-    return result, '<img src="{}" style="max-height: 240px">'.format(src)
-
-iface = gr.Interface(fn=greet, inputs="text", outputs=["text", "html"])
+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")