diff --git a/.gitignore b/.gitignore index 1ca6115..a7c73d9 100644 --- a/.gitignore +++ b/.gitignore @@ -1,10 +1,12 @@ # env .env *venv/ +**/__pycache__/ # Binaries for installation *.deb *.whl +.install/ # Test data data/pgvector.sql diff --git a/pipeline.indexing.png b/pipeline.indexing.png new file mode 100644 index 0000000..a4900ea Binary files /dev/null and b/pipeline.indexing.png differ diff --git a/hybrid-retrieval.png b/pipeline.retrieval.png similarity index 100% rename from hybrid-retrieval.png rename to pipeline.retrieval.png diff --git a/server-implementation/__pycache__/backend.cpython-310.pyc b/server-implementation/__pycache__/backend.cpython-310.pyc index aaba666..782d9ec 100644 Binary files a/server-implementation/__pycache__/backend.cpython-310.pyc and b/server-implementation/__pycache__/backend.cpython-310.pyc differ diff --git a/server-implementation/app.py b/server-implementation/app.py index df75429..14123c0 100644 --- a/server-implementation/app.py +++ b/server-implementation/app.py @@ -1,7 +1,9 @@ import pathlib + from gevent.pywsgi import WSGIServer -from flask import Flask, request, render_template from tabulate import tabulate + +from flask import Flask, request, render_template from wtforms import Form, StringField, SubmitField from wtforms.validators import DataRequired @@ -47,5 +49,6 @@ def query_backend(): return backend.format_for_api(result) -http_server = WSGIServer(("0.0.0.0", 8000), app) -http_server.serve_forever() +if __name__ == "__main__": + http_server = WSGIServer(("0.0.0.0", 8000), app) + http_server.serve_forever() diff --git a/server-implementation/backend.py b/server-implementation/backend.py index 8032108..94473e6 100644 --- a/server-implementation/backend.py +++ b/server-implementation/backend.py @@ -27,7 +27,7 @@ class AIBackend: document_store: PgvectorDocumentStore documents: list[Document] = [] - def __init__(self, load_dataset = False): + def __init__(self, load_dataset=False): get_torch_info() try: self.gpu = ComponentDevice.from_str("cuda:0") diff --git a/server-implementation/templates/results.html b/server-implementation/templates/results.html index 1a09d22..09af5eb 100644 --- a/server-implementation/templates/results.html +++ b/server-implementation/templates/results.html @@ -73,7 +73,7 @@ flex-direction: column; position: relative; width: 220px; - height: 250px; + height: 275px; border: 2px solid #b86cff; margin: 1rem; justify-items: flex-start;