server.py

import os
if not os.path.isfile("config.json"):
    print("config.json not found, please create one")
    exit()

import json
with open("config.json", "r") as f:
    IMAGE_ROOT_PATH = json.load(f)["image-path"]

with open("searchsite.html", "r") as f:
    website = "".join(f.readlines())

from search import create_image_string, count

# simple http server
from http.server import HTTPServer, BaseHTTPRequestHandler
from urllib.parse import urlparse, parse_qs
import re

image_regex = re.compile('[/\\\\]\d+[/\\\\]')

class SimpleHTTPRequestHandler(BaseHTTPRequestHandler):
    def do_GET(self):
        try:
            if self.path == "/":
                self.send_response(200)
                self.headers["Cache-Control"] = "max-age=31536000"
                self.end_headers()
                self.wfile.write(bytes(website, "utf-8"))
            elif self.path.startswith("/search"):
                self.send_response(200)
                self.end_headers()
                args = parse_qs(urlparse(self.path).query)
                query = args["query"][0]
                page = int(args["page"][0]) if "page" in args else 0
                page_size = int(args["page_size"][0]) if "page_size" in args else 30
                self.wfile.write(bytes(f'<div id="imagecontainer" onkeydown="inputEventHandler()">{create_image_string(query, True, page, page_size)}</div>', "utf-8"))
            elif self.path.startswith("/count"):
                self.send_response(200)
                self.end_headers()
                args = parse_qs(urlparse(self.path).query)
                query = args["query"][0]
                self.wfile.write(bytes(str(count(query)), "utf-8"))
            elif self.path.startswith("/refresh"):
                self.send_response(200)
                self.end_headers()
                from ingest import main
                from asyncio import run
                run(main())
            elif image_regex.match(self.path):
                self.send_response(200)
                self.headers["Cache-Control"] = "max-age=31536000"
                self.end_headers()
                self.wfile.write(open(IMAGE_ROOT_PATH + self.path, "rb").read())
            else:
                self.send_response(404)
                self.end_headers()
                self.wfile.write(bytes("404 Not Found", "utf-8"))
        except KeyError:
            pass

httpd = HTTPServer(("0.0.0.0", 8000), SimpleHTTPRequestHandler)
print("Serving on http://localhost:8000")
httpd.serve_forever()