from indexer import Indexer if __name__ == '__main__': query = input("Enter the Genre: ") query = query.split(' ') model = Indexer(filename = "movies.dat") model.fit() if len(query) == 1: genre = query[0] names = model.get(genre) for name in names: print(name) else: genre1 = query[0] genre2 = query[2] if query[1].lower() == 'or': names = model.getOr(genre1, genre2) for name in names: print(name) elif query[1].lower() == 'and': names = model.getAnd(genre1, genre2) for name in names: print(name)