from pocketflow import Flow

from nodes import Decide, Search, Wikipedia, Answer

def createFlow(logger):
    decide = Decide()
    decide.setLogger(logger)
    search = Search()
    search.setLogger(logger)
    wikipedia = Wikipedia()
    wikipedia.setLogger(logger)
    answer = Answer()
    answer.setLogger(logger)

    decide - "search" >> search
    decide - "wikipedia" >> wikipedia
    decide - "answer" >> answer
    search - "decide" >> decide
    wikipedia - "decide" >> decide

    return Flow(start=decide)