サムネイルの表示に対応

app.py

@@ -1,8 +1,10 @@
 import gradio as gr
 import json
 from pathlib import Path
+import re
 
 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())
@@ -12,17 +14,28 @@ name2fileid = json.loads(Path("name2fileid.json").read_text())
 def greet(query):
     result = "Not found"
     fileid = query
+    hit = None
 
-    if query in autov22fileid:
-        fileid = str(autov22fileid[query])
-    if query in filename2fileid:
-        fileid = str(filename2fileid[query])
+    if query.upper() in autov22fileid:
+        fileid = str(autov22fileid[query.upper()])
+        hit = True
+    if re.sub(r'\..*$', "", query) in filename2fileid:
+        fileid = str(filename2fileid[re.sub(r'\..*$', "", query)])
+        hit = True
     if query in name2fileid:
         fileid = str(name2fileid[query])
+        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)
 
     if fileid in fileid2json:
         result = json.dumps(fileid2json[fileid.strip()])
-    return result
+    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")
-iface.launch()
+iface = gr.Interface(fn=greet, inputs="text", outputs=["text", "html"])
+iface.launch(server_name="0.0.0.0")

autov22fileid.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:83e8c2345358226015f46167b7db2b8679fe52a196bbdda5ffa732046d4eae1d
-size 1501256
+oid sha256:7fb788ad61bb22f9c096a8175ecb85d31a66a706a2f3b2dd23bb63796b63b3c1
+size 1501802

fileid2json.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1721d3cfc780d37437e5cdd46c47fc0614686a790bed4bd45e0e873e384361a0
-size 82846964
+oid sha256:84cd1f5579aa81770e4b0942b825c8fe793b06fd68f2d279b770143dcaeb7ca6
+size 83064973

filename2fileid.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:871cc43e3a31af8f08a8946c70878c1e5694af0af8e7a5c1026c87920919ad4e
-size 2044320
+oid sha256:ba9093d95e56df976930868251890024795330c3472190522784d947af865fc3
+size 2045049

name2fileid.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d7fe02437df0faa80dac8e1b0ecfaa327e2c0b087878d78b73b18b8680aea386
-size 2297509
+oid sha256:407b0c945e24d7dd3b0775d2494be9e96d0686d264a2114022977d55e6919c0b
+size 2298256