from pr.config import MAX_AUTONOMOUS_ITERATIONS
from pr.ui import Colors


def is_task_complete(response, iteration):
    if "error" in response:
        return True

    if "choices" not in response or not response["choices"]:
        return True

    message = response["choices"][0]["message"]
    content = message.get("content", "").lower()

    completion_keywords = [
        "task complete",
        "task is complete",
        "finished",
        "done",
        "successfully completed",
        "task accomplished",
        "all done",
        "implementation complete",
        "setup complete",
        "installation complete",
    ]

    error_keywords = [
        "cannot proceed",
        "unable to continue",
        "fatal error",
        "cannot complete",
        "impossible to",
    ]

    has_tool_calls = "tool_calls" in message and message["tool_calls"]
    mentions_completion = any(keyword in content for keyword in completion_keywords)
    mentions_error = any(keyword in content for keyword in error_keywords)

    if mentions_error:
        return True

    if mentions_completion and not has_tool_calls:
        return True

    if iteration > 5 and not has_tool_calls:
        return True

    if iteration >= MAX_AUTONOMOUS_ITERATIONS:
        print(f"{Colors.YELLOW}⚠ Maximum iterations reached{Colors.RESET}")
        return True

    return False