import sys
from pprint import pprint

from haystack import Pipeline, component


@component
class QuestionClassifier:
    @component.output_types(topic=str)
    def run(self, question: str):
        normalized = " ".join(question.strip().split()).lower()
        if "pipeline" in normalized:
            return {"topic": "pipeline"}
        return {"topic": "general"}