from pathlib import Path

# FAISS vector database paths
db_paths = {
    "db1": "db/01_dialogues",
    "db2": "db/02_context_history",
    "db3": "db/03_news",
    "db4": "db/04_user_info",
    "db5_sub": "db/05_chat_history",
    "db5": "db/05_chat_history/All",
    "db6": "db/06_web_interest"
}

#countries.txt path
COUNTRIES_FILE = Path(f"{db_paths['db6']}/countries.txt")

# === DB3 paths (general news) ===
DB3_SUMMARY_FILE = Path(f"{db_paths['db3']}/news_summary.txt")
DB3_COUNTRIES_FILE = Path(f"{db_paths['db3']}/countries.txt")
DB3_COUNTRY_ORIGIN_FILE = Path(f"{db_paths['db3']}/country_origin.txt")
DB3_COUNTRY_LIVING_FILE = Path(f"{db_paths['db3']}/country_living.txt")
DB3_FAISS_DIR = Path(db_paths["db3"])   # contains index.faiss + index.pkl

# === DB6 paths (topic tracker) ===
DB6_FAISS_DIR = Path(db_paths["db6"])
DB6_RANKED_FILE = Path(f"{db_paths['db6']}/ranked_news_summary.txt")
DB6_TOPIC_TRACKER = Path(f"{db_paths['db6']}/topic_tracker.json")
DB6_TOPIC_MATCHED_TXT = Path(f"{db_paths['db6']}/topic_matched.txt")
DB6_TOPIC_MATCHED_JSON = Path(f"{db_paths['db6']}/topic_matched.json")
#DB6_DETAILED_DIR = Path(f"{db_paths['db6']}/detailed")
DB6_TOPIC_MATCHED_FULL_TXT = Path(f"{db_paths['db6']}/topic_matched_full.txt")
DB6_TOPIC_MATCHED_FULL_JSON = Path(f"{db_paths['db6']}/topic_matched_full.json")
DB6_REQUESTED_FAISS_DIR = Path(f"{db_paths['db6']}/FAISS_requested_news")

# === DB5 paths (topic tracker) ===
SUMMARY_FOLDER = Path(f"{db_paths['db5_sub']}/summaries")
SHORT_TERM_FLAT_PATH = Path(f"{db_paths['db5_sub']}/short_term_for_retrieval.json")
SHORT_TERM_STRUCTURED_PATH = Path(f"{db_paths['db5_sub']}/short_term_structured.json")
DB5_TOTAL_HISTORY_PATH = Path(f"{db_paths['db5_sub']}/total_history.json")
SESSION_META_PATH = Path(f"{db_paths['db5_sub']}/session_meta.json")

SHORT_TERM_STRUCTURED_PATH.parent.mkdir(parents=True, exist_ok=True)
SHORT_TERM_FLAT_PATH.parent.mkdir(parents=True, exist_ok=True)

# === DB4 paths (topic tracker) ===
PERSONAL_INFO_PATH = Path(f"{db_paths['db4']}/user_info.json") #maybe to remove
USER_INFO_PATH = Path(f"{db_paths['db4']}/user_info.json")