Implement the prompt handling for different datasets

[chakravarthik27]

Implement functionality to handle prompts specific to different datasets, enhancing versatility and adaptability of the system for prompt-based tasks across diverse datasets.