csv_cache.py

import pandas as pd
import os

class CSVSCache:
    def __init__(self, filename="results.csv"):
        self.filename = filename
        self.df = self._load_cache()

    def _load_cache(self):
        if os.path.exists(self.filename):
            return pd.read_csv(self.filename)
        return pd.DataFrame(columns=["task_id", "question", "answer"])

    def _save_cache(self):
        self.df.to_csv(self.filename, index=False)

    def get_all_entries(self):
        return self.df
    
    def get_answer(self, question):
        if question in self.df.index:
            answer = self.df.loc[question, 'answer']
            if pd.isna(answer) or answer == "unknown":
                return "unknown"
            return str(answer)
        else:
            return "unknown"

def main():
    csv = CSVSCache()
   
    print("done")
    q = "Where were the Vietnamese specimens described by Kuznetzov in Nedoshivina's 2010 paper eventually deposited? Just give me the city name without abbreviations."
    print(csv.get_answer(q))

if __name__ == "__main__":
    main()