utils.py

# utils.py

import re 
from dotenv import load_dotenv
import re 
import os 
from globalvars import tasks

def load_env_variables():
    # Load the .env file
    load_dotenv()

    # Retrieve the environment variables
    hf_token = os.getenv('HF_TOKEN')
    yi_token = os.getenv('YI_TOKEN')

    return hf_token, yi_token

def parse_and_route(example_output: str):  
    # Regex pattern to match the true task  
    pattern = r'"(\w+)":\s?true'  
  
    # Find the true task  
    match = re.search(pattern, example_output)  
  
    if match:  
        true_task = match.group(1)  
        if true_task in tasks:  
            return {true_task: tasks[true_task]}  
        else:  
            return {true_task: "Task description not found"}  
    else:  
        return "No true task found in the example output"