nodes/DemandUnderstand.py

from roles.ChiefProductOfficer import ChiefProductOfficer
from roles.ChiefCreativeOfficer import ChiefCreativeOfficer

class DemandUnderstand:
    def __init__(self):
        self.cpo = ChiefProductOfficer()
        self.cco = ChiefCreativeOfficer()

    def func_prompt_enhance(self, input_text):
        """
        Enhances the user input to understand the demand.
        """
        product_vision = self.cpo.define_product_vision(input_text)
        creative_aspects = self.cco.generate_creative_aspects(ideas="Innovative features", modality="Web")
        enhanced_demand = {
            "product_vision": product_vision,
            "creative_aspects": creative_aspects,
            "language": "Python",
            "modality": "Web",
            "task": input_text
        }
        return enhanced_demand