GENERATE_PYTHON_CODE_PROMPT = """ Generate a functional and correct Python code for the given problem. Problem: """