| # example.py | |
| from generation_fast import CodeGenerator | |
| generator = CodeGenerator() | |
| nl_inputs = [ | |
| "Write a Python function to calculate the sum of two numbers.", | |
| "Generate Python code to check if a number is prime.", | |
| "Create a Python class for a simple calculator.", | |
| ] | |
| for nl_input in nl_inputs: | |
| generated_code = generator.generate_code(nl_input) | |
| print(f"Input: {nl_input}\nGenerated Code:\n{generated_code}\n{'-'*40}") |