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