File size: 936 Bytes
132ac81 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
from os import makedirs
from os.path import join
import pandas as pd
from assistant import ask_assistant
from functions import load_prompt
def elaborate_field(row: pd.Series) -> pd.Series:
if row["column_type"] == "string":
print(
f"Elaborating field {row['column_name']} of the file {row['filename']}... ",
end="",
flush=True,
)
tag: str = ask_assistant(row["column_name"])
print("Tag: ", tag)
if tag and tag != "None":
row["tag"] = tag
return row
def main() -> None:
save_df(fields, join(OUTPUTS_FOLDER, "fields_generated.csv"))
# def main_old() -> None:
# prompt: str = load_prompt(PROMPT_FILENAME)
# ddls: list[tuple[str, str]] = extract_ddls(DDLS_FOLDER)
# for file, ddl in ddls:
# elaborate_ddl_old(file, prompt, ddl)
if __name__ == "__main__":
makedirs(OUTPUTS_FOLDER, exist_ok=True)
main()
|