import openai 
import os

def get_api_key(local = False):
    if local:
        from dotenv import load_dotenv
        load_dotenv()
    return os.getenv('OPEN_API_KEY')

def get_user_input(prompt):
    return input(prompt)

def choose_cooperation_type():
    choice = get_user_input("Please choose the cooperation type: \n" 
                            +"1. Sequential: human provide an answer first and then AI provide the answer based on it.\n" 
                            +"2. Parallel: human and AI give answers seperately and then AI does the merge.\n")
    if choice == '1':
        return 'sequential'
    elif choice == '2':
        return 'parallel'
    else:
        print("Invalid choice. Please try again.")
        return choose_cooperation_type()

def describe_task():
    task_description = get_user_input("Please describe your task: ")
    if task_description.strip() == "":
        # print("Task description cannot be empty. Please try again.")
        # return describe_task()
        task_description = "Write a poem about the moon in 3 lines."
    return task_description

def generate_text_with_gpt(prompts, api_key = None):
    """Generate text using the GPT-3 model."""
    if api_key:
        openai.api_key = api_key

    try:
        response = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",
            messages=[
                {"role": "system", "content": "Please assist."},
                {"role": "user", "content": prompts}
            ]
        )
        return response['choices'][0]['message']['content']
    except Exception as e:
        print(f"Error occurred when generating texts: {e}")
        return ""


def generate_ai_initial_answer(task_description, api_key=None):
    prompt = f"Given the task: {task_description}, provide an answer: "
    return generate_text_with_gpt(prompt, api_key)

def merge_texts_parallel(task_description, human_text, ai_text, api_key = None):
    prompt = f"Given the task: {task_description}, there are two answers provided:\n" + \
             f"The first answer:  {human_text}\nThe second answer: {ai_text}\n" + \
             f"Merge the two answers into one in a coherent way: "
    return generate_text_with_gpt(prompt, api_key)

def merge_texts_sequential(task_description, human_text, api_key = None):
    prompt = f"Given the task:{task_description}, here is the answer provided by the human: {human_text}\n" + \
             f"Refine this response and ensure the final answer aligns with the human's intent:"
    return generate_text_with_gpt(prompt, api_key)

def modify_with_suggestion(task_description, text, suggestions, api_key = None):
    prompt = f"Given the task:{task_description}, the answer provided is: {text}\n" + \
             f"Modify the answer based on the following suggestions: {suggestions}"
    return generate_text_with_gpt(prompt, api_key)

# def get_evaluation_with_gpt(task_description, text, api_key = None):
#     prompt = f"Given the task:{task_description}, the answer provided is: {text}\n" + \
#              f"Evaluate the answer and provide scores between 0 and 10,\n" + \
#              f"where criteria for evaluation are Novelty, Feasibility, and Defensibility.\n" + \
#              f"Use a professional standard, where 5 represents an acceptable quality,\n" + \
#              f"and scores closer to 10 should reflect exceptional quality.\n" + \
#              f"Finally conclude with an overall score in the range of 0 to 10."
#     return generate_text_with_gpt(prompt, api_key)

def get_evaluation_with_gpt(task_description, text, api_key = None):
    prompt = f"Given the task: {task_description}, the provided answer is: {text}\n" + \
             f"Evaluate the answer according to the following criteria, providing both a score (0-10) and a brief comment (1 sentence) for each.\n" + \
             f"Use a professional standard, where 5 represents acceptable quality, and scores closer to 10 reflect exceptional quality.\n" + \
             f"Please format the output exactly as follows:\n" + \
             f"Novelty: [Score]\nComment: [Short comment on Novelty]\n" + \
             f"Feasibility: [Score]\nComment: [Short comment on Feasibility]\n" + \
             f"Defensibility: [Score]\nComment: [Short comment on Defensibility]\n" + \
             f"Overall Score: [Score]\nOverall Comment: [Overall feedback on the answer]\n"
    return generate_text_with_gpt(prompt, api_key)