diff --git a/pyproject.toml b/pyproject.toml index 7103ee0..5f353fb 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -22,7 +22,6 @@ dependencies = [ "torch>=2.0.0", "tqdm>=4.65.0", "transformers>=4.27.4", - "setuptools", ] description = "A semantic search CLI tool" name = "semantra" diff --git a/src/semantra/semantra.py b/src/semantra/semantra.py index 9636117..c16987e 100644 --- a/src/semantra/semantra.py +++ b/src/semantra/semantra.py @@ -3,10 +3,10 @@ import json import math import os +from importlib.metadata import version import click import numpy as np -import pkg_resources from dotenv import load_dotenv from flask import Flask, jsonify, make_response, request, send_file, send_from_directory from tqdm import tqdm @@ -31,11 +31,12 @@ write_embedding, ) -VERSION = pkg_resources.require("semantra")[0].version +VERSION = version("semantra") DEFAULT_ENCODING = "utf-8" DEFAULT_PORT = 8080 package_directory = os.path.dirname(os.path.abspath(__file__)) +client_public_directory = os.path.join(package_directory, "client_public") class Content: @@ -655,7 +656,7 @@ def get_content(filename): @app.route("/") def base(): return send_from_directory( - pkg_resources.resource_filename("semantra.semantra", "client_public"), + client_public_directory, "index.html", ) @@ -663,7 +664,7 @@ def base(): @app.route("/") def home(path): return send_from_directory( - pkg_resources.resource_filename("semantra.semantra", "client_public"), + client_public_directory, path, )