search

Semantic image search using FastEmbed and sqlite-vec

  1. 1
  2. 2
  3. 3
  4. 4
  5. 5
  6. 6
  7. 7
  8. 8
  9. 9
  10. 10
  11. 11
  12. 12
  13. 13
  14. 14
  15. 15
  16. 16
  17. 17
  18. 18
  19. 19
  20. 20
  21. 21
  22. 22
  23. 23
  24. 24
  25. 25
  26. 26
  27. 27
  28. 28
  29. 29
  30. 30
  31. 31
  32. 32
  33. 33
  34. 34
from http.client import HTTPConnection
import os
import socket
import sys
import xmlrpc.client


class UnixStreamHTTPConnection(HTTPConnection):
    def connect(self):
        self.sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
        self.sock.connect(self.host)


class UnixStreamTransport(xmlrpc.client.Transport, object):
    def __init__(self, socket_path):
        self.socket_path = socket_path
        super().__init__()

    def make_connection(self, host):
        return UnixStreamHTTPConnection(self.socket_path)


proxy = xmlrpc.client.ServerProxy(
    "http://localhost", transport=UnixStreamTransport("search.sock")
)
res = proxy.search(sys.argv[1], sys.argv[2])
print("\n".join(res))

if len(sys.argv) > 3:
    for path in os.listdir():
        if path.startswith("res"):
            os.unlink(path)
    for i, path in enumerate(res):
        os.symlink(path, f"res{i}")