add search

client/src/lib/Api.ts

@@ -40,6 +40,18 @@ export async function similarSearch(imagePath: string, limit: number, excludeLab
     return response;
 }
 
+export async function TextSearch(queryStr: string, limit: number, excludeLabeled: boolean): Promise<Hits> {
+    const response = await fetchJSON<Hits>("/text_search", {
+        "q": queryStr,
+        "exclude_labeled": excludeLabeled,
+        limit: limit.toString(),
+    }).catch((e) => {
+        console.error(e);
+        throw new Error("Failed to retrieve text search results.", { cause: e })
+    });
+    return response;
+}
+
 export async function random(limit: number): Promise<Hits> {
     const response = await fetchJSON<Hits>("/random", {
         limit: limit.toString(),

mmdx/search.py

@@ -138,9 +138,12 @@ def search_by_image_path(
         )
         return df_hits
 
-    def search_by_text(
+    def search_by_seller(
         self, query_string: str, limit: int, exclude_labeled: bool
     ) -> pd.DataFrame:
+        original_path = os.environ.get("CSV_PATH")
+        original_df = pd.read_csv(original_path)
+
         query_str_embedding = self.model.embed_text(query=query_string)
         df_hits = self.__vector_embedding_search(
             query_str_embedding, limit, exclude_labeled

server.py

@@ -79,6 +79,14 @@ def image_search():
     return {"total": len(hits.index), "hits": hits.to_dict("records")}
 
 
+@app.route("/api/v1/text_search")
+def text_search():
+    query: str = request.args.get("q")
+    exclude_labeled: bool = request.args.get("exclude_labeled", "false") == "true"
+    limit: int = request.args.get("limit", 12, type=int)
+    hits = db.search_by_seller(query_string=query, limit=limit, exclude_labeled=exclude_labeled)
+    return {"total": len(hits.index), "hits": hits.to_dict("records")}
+
 @app.route("/api/v1/labeled")
 def labeled_search():
     limit: int = request.args.get("limit", 12, type=int)