from shortGPT.gpt import gpt_utils
import json

def generate_title_description_dict(content):
    out = {"title": "", "description":""}
    chat, system = gpt_utils.load_local_yaml_prompt('prompt_templates/yt_title_description.yaml')
    chat = chat.replace("<<CONTENT>>", f"{content}")
    
    while out["title"] == "" or out["description"] == "":
        result = gpt_utils.gpt3Turbo_completion(chat_prompt=chat, system=system, temp=1)
        try:
            response = json.loads(result)
            if "title" in response:
                out["title"] = response["title"]
            if "description" in response:
                out["description"] = response["description"]
        except Exception as e:
            pass
        
    return out['title'], out['description']