Finalized

.gitignore

@@ -1,10 +1,12 @@
 # env
 .env
 *venv/
+**/__pycache__/
 
 # Binaries for installation
 *.deb
 *.whl
+.install/
 
 # Test data
 data/pgvector.sql

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()

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")

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;