bat-6 commited on
Commit
aacbdb0
·
1 Parent(s): 5ccb7fc
Data/database/sql_connector.py CHANGED
@@ -88,8 +88,8 @@ def load_preprocessed_projects():
88
  import os
89
  from pathlib import Path
90
  possible_paths = [
 
91
  Path("models/metadata.parquet"),
92
- Path(__file__).resolve().parent.parent.parent / "models" / "metadata.parquet",
93
  Path("../models/metadata.parquet"),
94
  ]
95
  for path in possible_paths:
 
88
  import os
89
  from pathlib import Path
90
  possible_paths = [
91
+ Path(__file__).resolve().parents[2] / "models" / "metadata.parquet",
92
  Path("models/metadata.parquet"),
 
93
  Path("../models/metadata.parquet"),
94
  ]
95
  for path in possible_paths:
src/similarity_model/embedding_engine.py CHANGED
@@ -31,9 +31,13 @@ TEXT_COL = "clean_text"
31
  TITLE_COL = "project_title"
32
  TECH_COL = "technologies"
33
 
34
- MODEL_DIR = Path("models")
 
 
 
 
35
  INDEX_PATH = MODEL_DIR / "faiss_index.bin"
36
- META_PATH = MODEL_DIR / "metadata.parquet"
37
 
38
  TOP_K_DEFAULT = 10
39
  MIN_SCORE_THRESHOLD = 0.35
 
31
  TITLE_COL = "project_title"
32
  TECH_COL = "technologies"
33
 
34
+ # Resolve paths relative to the project root (3 levels up from this file:
35
+ # src/similarity_model/embedding_engine.py -> src/similarity_model -> src -> project root)
36
+ _PROJECT_ROOT = Path(__file__).resolve().parents[2]
37
+
38
+ MODEL_DIR = _PROJECT_ROOT / "models"
39
  INDEX_PATH = MODEL_DIR / "faiss_index.bin"
40
+ META_PATH = MODEL_DIR / "metadata.parquet"
41
 
42
  TOP_K_DEFAULT = 10
43
  MIN_SCORE_THRESHOLD = 0.35
src/similarity_model/llm_feature_extractor.py CHANGED
@@ -135,8 +135,8 @@ def extract_features_llm(text: str):
135
  from pathlib import Path
136
 
137
  possible_paths = [
 
138
  Path("models/metadata.parquet"),
139
- Path(__file__).resolve().parent.parent.parent / "models" / "metadata.parquet",
140
  Path("../models/metadata.parquet"),
141
  ]
142
 
 
135
  from pathlib import Path
136
 
137
  possible_paths = [
138
+ Path(__file__).resolve().parents[2] / "models" / "metadata.parquet",
139
  Path("models/metadata.parquet"),
 
140
  Path("../models/metadata.parquet"),
141
  ]
142
 
src/similarity_model/semantic_search.py CHANGED
@@ -32,9 +32,13 @@ DEFAULT_MODEL = "all-MiniLM-L6-v2"
32
  TITLE_COL = "project_title"
33
  TECH_COL = "technologies"
34
 
35
- INDEX_PATH = "models/faiss_index.bin"
36
- META_PATH = "models/metadata.parquet"
37
- EMBED_PATH = "models/project_embeddings.npy"
 
 
 
 
38
 
39
  TOP_K = 20
40
  MIN_SCORE = 0.35
 
32
  TITLE_COL = "project_title"
33
  TECH_COL = "technologies"
34
 
35
+ # Resolve paths relative to the project root (3 levels up from this file:
36
+ # src/similarity_model/semantic_search.py -> src/similarity_model -> src -> project root)
37
+ _PROJECT_ROOT = Path(__file__).resolve().parents[2]
38
+
39
+ INDEX_PATH = _PROJECT_ROOT / "models" / "faiss_index.bin"
40
+ META_PATH = _PROJECT_ROOT / "models" / "metadata.parquet"
41
+ EMBED_PATH = _PROJECT_ROOT / "models" / "project_embeddings.npy"
42
 
43
  TOP_K = 20
44
  MIN_SCORE = 0.35