memory_stream.py

#!/usr/bin/python
# -*- coding: utf-8 -*-

import pandas as pd
import json

class MemoryStream:
    def __init__(self):
        self.columns = ["ID", "TargetDocID", "Question", "Reply", "Point"]
        self.data = pd.DataFrame(columns=self.columns)
        self.current_id = 1

    def add_data(self, target_doc_id, question, reply, point):
        ret_id = self.current_id
        data = [[self.current_id, target_doc_id, question, reply, point]]
        new_data = pd.DataFrame(data, columns=self.columns)
        self.data = pd.concat([self.data, new_data], ignore_index=True)
        self.current_id += 1
        return ret_id

    def get_data(self):
        return self.data

    def get_reply(self, index = None):
        if (index == None):
            index = len(self.data) - 1
        if index >= 0 and index < len(self.data):
            return self.data.loc[index, "Reply"]
        else:
            return None

    def get_data(self, id: int):
        filtered_data = self.data.loc[self.data["ID"] == id]
        if filtered_data.empty:
            return None
        else:
            return filtered_data.to_dict(orient="records")[0]

    def get_point(self, index = None):
        if (index == None):
            index = len(self.data) - 1
        if index >= 0 and index < len(self.data):
            return self.data.loc[index, "Point"]
        else:
            return None

    def save_to_json(self, file_path):
        json_data = self.data.to_dict(orient="records")

        with open(file_path, "w", encoding="utf-8") as f:
            json.dump(json_data, f, indent=4, ensure_ascii=False)


    def load_from_json(self, file_path):
        with open(file_path, 'r') as file:
            json_data = json.load(file)
            self.data = pd.DataFrame(json_data)

if __name__ == "__main__":
    memory_stream = MemoryStream()
    while True:
        target_doc_id = input("TargetDocID> ")
        question = input("question> ")
        reply = input("reply> ")
        point = input("point> ")
        memory_stream.add_data(target_doc_id, question, reply, int(point))
        print(memory_stream.get_data())
else:
    pass