Update src/core.py
Browse files- src/core.py +5 -4
src/core.py
CHANGED
|
@@ -1,16 +1,17 @@
|
|
| 1 |
from src.services.utils import *
|
| 2 |
from src.services.processor import *
|
| 3 |
|
| 4 |
-
|
|
|
|
|
|
|
|
|
|
| 5 |
prompt = set_prompt(data.problem)
|
| 6 |
|
| 7 |
constraints = retrieve_constraints(prompt)
|
| 8 |
constraints_stemmed = stem(constraints, "constraints")
|
| 9 |
|
| 10 |
save_dataframe(constraints_stemmed, "constraints_stemmed.xlsx")
|
| 11 |
-
|
| 12 |
-
global_tech, global_tech_embeddings = load_technologies()
|
| 13 |
-
|
| 14 |
#global_tech, keys, original_tech = preprocess_tech_data(df)
|
| 15 |
|
| 16 |
save_dataframe(global_tech, "global_tech.xlsx")
|
|
|
|
| 1 |
from src.services.utils import *
|
| 2 |
from src.services.processor import *
|
| 3 |
|
| 4 |
+
global_tech, global_tech_embeddings = load_technologies()
|
| 5 |
+
|
| 6 |
+
|
| 7 |
+
def process_input(data, global_tech, global_tech_embeddings):
|
| 8 |
prompt = set_prompt(data.problem)
|
| 9 |
|
| 10 |
constraints = retrieve_constraints(prompt)
|
| 11 |
constraints_stemmed = stem(constraints, "constraints")
|
| 12 |
|
| 13 |
save_dataframe(constraints_stemmed, "constraints_stemmed.xlsx")
|
| 14 |
+
|
|
|
|
|
|
|
| 15 |
#global_tech, keys, original_tech = preprocess_tech_data(df)
|
| 16 |
|
| 17 |
save_dataframe(global_tech, "global_tech.xlsx")
|