ChatGPT loop

loop.py

import openai
import subprocess

# Set your OpenAI API key
openai.api_key = 'your_openai_api_key'

def generate_code(prompt):
    response = openai.Completion.create(
        engine="text-davinci-003",
        prompt=prompt,
        max_tokens=150,
        n=1,
        stop=None,
        temperature=0.5
    )
    return response.choices[0].text.strip()

def run_tests(file_path):
    try:
        result = subprocess.run(['python', file_path], capture_output=True, text=True)
        return result.stdout, result.stderr
    except Exception as e:
        return '', str(e)

def correct_code(prompt):
    return generate_code(prompt)

def main():
    initial_prompt = "Write a Python function to add two numbers."

    iteration = 0
    max_iterations = 10

    while iteration < max_iterations:
        iteration += 1
        print(f"Iteration {iteration}")

        code = generate_code(initial_prompt)
        print(f"Generated Code:\n{code}")

        file_path = 'generated_code.py'
        with open(file_path, 'w') as f:
            f.write(code)

        stdout, stderr = run_tests(file_path)
        if stderr:
            print(f"Error:\n{stderr}")
            new_prompt = f"The following code produced an error:\n{code}\nError: {stderr}\nPlease correct the code."
            initial_prompt = new_prompt
        else:
            print("Code executed successfully.")
            print(f"Output:\n{stdout}")
            break

if __name__ == "__main__":
    main()