commit changes

app.py

@@ -1,7 +1,5 @@
 import gradio as gr
+from pyserini_wikipedia_kilt_doc import search
 
-def greet(name):
-    return "Hello " + name + "!!"
-
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+iface = gr.Interface(fn=search, inputs="text", outputs="text")
 iface.launch()

pyserini_wikipedia_kilt_doc.py

@@ -0,0 +1,32 @@
+from pyserini.search.lucene import LuceneSearcher
+from transformers import Tool
+import json
+
+
+# searcher = LuceneSearcher.from_prebuilt_index('wikipedia-kilt-doc')
+searcher = LuceneSearcher('index-wikipedia-kilt-doc-20210421-f29307.b8ec8feb654f7aaa86f9901dc6c804a8')
+
+
+
+def search(query):
+    hits = searcher.search(query, k=1)
+
+    # for i in range(0, 5):
+    #     print(f'{i+1:2} {hits[i].docid:15} {hits[i].score:.5f}')
+
+    hit = hits[0]
+    contents = json.loads(hit.raw)['contents']
+    return contents
+
+class PyseriniWikipediaKiltDoc(Tool):
+    name = "pyserini-wikipedia-kilt-doc"
+    description = (
+        "This is a tool that returns the top 5 results from the Wikipedia KILT index."
+    )
+
+    inputs = ["text"]
+    outputs = ["text"]
+
+    def __call__(self, query: str):
+        return search(query)
+    

requirements.txt

@@ -0,0 +1,95 @@
+appdirs==1.4.4
+autokey==0.95.10
+bcc==0.18.0
+bcrypt==3.1.7
+blinker==1.4
+Brlapi==0.8.2
+Brotli==1.0.9
+certifi==2020.6.20
+chardet==4.0.0
+chrome-gnome-shell==0.0.0
+click==7.1.2
+colorama==0.4.4
+command-not-found==0.3
+cryptography==3.3.2
+cupshelpers==1.0
+dbus-python==1.2.16
+decorator==4.4.2
+defer==1.0.6
+distlib==0.3.1
+distro==1.5.0
+distro-info==1.0
+faiss-cpu==1.7.4
+filelock==3.0.12
+gssapi==1.6.12
+gyp==0.1
+hidpidaemon==18.4.6
+httplib2==0.18.1
+idna==2.10
+ifaddr==0.1.7
+importlib-metadata==4.0.1
+jeepney==0.7.1
+keyring==23.0.1
+language-selector==0.1
+launchpadlib==1.10.13
+lazr.restfulclient==0.14.2
+lazr.uri==1.0.5
+louis==3.18.0
+lz4==3.1.3+dfsg
+macaroonbakery==1.3.1
+more-itertools==4.2.0
+netaddr==0.8.0
+netifaces==0.10.9
+numpy==1.19.5
+oauthlib==3.1.0
+olefile==0.46
+paramiko==2.7.2
+Pillow==8.1.2
+pop-transition==1.1.2
+protobuf==3.12.4
+py-cpuinfo==5.0.0
+py3dns==3.2.1
+pycairo==1.16.2
+pycups==2.0.1
+pydbus==0.6.0
+Pygments==2.7.1
+PyGObject==3.40.1
+pyinotify==0.9.6
+PyJWT==1.7.1
+pykerberos==1.1.14
+pymacaroons==0.13.0
+PyNaCl==1.4.0
+PyOpenGL==3.1.5
+pyRFC3339==1.1
+python-apt===2.2.1pop0-1628521831-21.10-5dbe6cf
+python-dateutil==2.8.1
+python-debian===0.1.39ubuntu1
+python-lzo==1.12
+python-xlib==0.29
+pytz==2021.1
+pyxdg==0.27
+PyYAML==5.3.1
+rencode==1.0.6
+repolib==1.5.2
+repoman==1.4.0
+requests==2.25.1
+screen-resolution-extra==0.0.0
+SecretStorage==3.3.1
+sessioninstaller==0.0.0
+setproctitle==1.2.1
+simplejson==3.17.2
+six==1.16.0
+ssh-import-id==5.11
+systemd-python==234
+ubuntu-advantage-tools==27.8
+ubuntu-drivers-common==0.0.0
+ufw==0.36.1
+uritools==3.0.0
+urllib3==1.26.5
+virtualenv==20.4.6+ds
+wadllib==1.3.5
+xdg==5
+xkit==0.0.0
+xpra==3.0.13
+zeroconf==0.26.1
+zipp==1.0.0

tool_config.json

@@ -0,0 +1,5 @@
+{
+  "description": "This is a tool that returns the top 5 results from the Wikipedia KILT index.",
+  "name": "pyserini_wikipedia_kilt_doc",
+  "tool_class": "pyserini_wikipedia_kilt_doc.PyseriniWikipediaKiltDoc"
+}