singleprocess.py

import os
import sys
import random
import string
import subprocess
from openai import OpenAI


APIKey = "API-KEY"
APIUrl = "https://api.deepseek.com"
AIModel = "deepseek-chat"

def DeepseekAPI(input, instructions, api_key = APIKey, base_url = APIUrl, model = AIModel):
    client = OpenAI(api_key = api_key, base_url = base_url)
    response = client.chat.completions.create(
        model = model,
        messages=[
            {"role": "system", "content": instructions},
            {"role": "user", "content": input},
        ],
        stream=False
    )
    APIResponse = response.choices[0].message.content
    return APIResponse

def PromptRefining(input):
    promptinstructions = "Rewrite the given input as the perfect ai-ready prompt to obtain the maximum of prompt-engeneering, you have to write no code, but in the prompt ask for full code in one single python code, just the roadmap for another ai, do not add your extenal text, just a ready copy and paste full detailed prompt."
    PromptResponse = DeepseekAPI(input, promptinstructions)
    return PromptResponse

def CodeLibSeparate(text):
    lines = text.split('\n')
    state = None
    current_block = []
    code = None
    libs = None
    for line in lines:
        stripped = line.strip()
        if stripped.startswith('```'):
            marker = stripped[3:].strip()
            if marker.startswith('python') and not code:
                state = 'python'
                current_block = []
            elif marker.startswith('bash') and not libs:
                state = 'bash'
                current_block = []
            elif state:
                if state == 'python' and not code:
                    code = '\n'.join(current_block)
                elif state == 'bash' and not libs:
                    if any(l.strip().startswith('pip install') for l in current_block):
                        libs = '\n'.join(current_block)
                state = None
            continue
        if state:
            current_block.append(line)
    return code, libs

def CodeDevelop(input):
    codeinput = PromptRefining(input)
    codeinstructions = "Write the code, first give the code in the form of main.py and then code, then give the pip install requirements in the form of pip install *** in a separated codeblock, avoid introductions or everything else than code and libs"
    codeandlibs = DeepseekAPI(input, codeinstructions)
    code, libs = CodeLibSeparate(codeandlibs)
    return code, libs

def rand(lunghezza=10):
    caratteri = string.ascii_letters + string.digits
    return ''.join(random.choice(caratteri) for _ in range(lunghezza))

def Runner(code, libs):
    folder = rand()
    path = os.path.join(os.getcwd(), folder)
    os.makedirs(path, exist_ok=True)
    os.chdir(folder)
    filename = rand()+".py"
    with open(filename, 'w', encoding='utf-8') as f:
        f.write(code)
    subprocess.run([sys.executable, "-m", "venv", "venv"])
    log_file = "log.txt"
    command = f"""
    source venv/bin/activate && 
    {libs} && 
    python {filename}
    """
    with open(log_file, "w") as log:
        result = subprocess.run(
            command,
            shell=True,
            executable="/bin/zsh",
            check=False,
            text=True,
            stdout=log,
            stderr=subprocess.STDOUT
        )
    with open(log_file, "a") as log:
        log.write(f"\n\nProcesso terminato con codice: {result.returncode}")
    filewrite = open(log_file, 'r', encoding='utf-8')
    logresult = filewrite.read()
    filewrite.close()
    status = 0
    if result.returncode == 0:
        status = 200
    elif result.returncode == 1:
        status = 400
    else:
        status = 5000
    return status, logresult, filename, code, libs

def FixRunner(fixedcode, fixedlibs, count):
    mid = rand()
    countn = str(count)
    filename = countn + "__" + "temp" + mid + ".py"
    with open(filename, 'w', encoding='utf-8') as f:
        f.write(fixedcode)
    subprocess.run([sys.executable, "-m", "venv", "venv"])
    log_file = "temp" + mid + "_log.txt"
    command = f"""
    {fixedlibs} && 
    python {filename}
    """
    with open(log_file, "w") as log:
        result = subprocess.run(
            command,
            shell=True,
            executable="/bin/zsh",
            check=False,
            text=True,
            stdout=log,
            stderr=subprocess.STDOUT
        )
    with open(log_file, "a") as log:
        log.write(f"\n\nProcesso terminato con codice: {result.returncode}")
    status = 0
    if result.returncode == 0:
        status = 200
    elif result.returncode == 1:
        status = 400
    else:
        status = 5000
    return status

def CodeFixer(logresult, filename, code, libs):
    status = 400
    count = 0
    while status == 400:
        prompt = code + "\n" + libs + "\n" + logresult + "\n GIVE BACK FULL CODE UPDATED"
        instructions = "Fix it and give back full code updated. Give the code in the form of main.py and then code, then give the pip install requirements in the form of pip install *** in a separated codeblock, avoid introductions or everything else than code and libs"
        fixedcode, fixedlibs = CodeLibSeparate(DeepseekAPI(prompt, instructions))
        status = FixRunner(fixedcode, fixedlibs, count)
        count = count + 1
        if count > 3:
            instructions2 = "REWRITE COMPLETELY THE CODE CHANGING APROACH. Give the code in the form of main.py and then code, then give the pip install requirements in the form of pip install *** in a separated codeblock, avoid introductions or everything else than code and libs"
            fixedcode, fixedlibs = CodeLibSeparate(DeepseekAPI(prompt, instructions2))
            status = FixRunner(fixedcode, fixedlibs, count)
        if status == 200:
            with open(filename, 'w', encoding='utf-8') as f:
                f.write(fixedcode)
    return fixedcode

def WriteMD(code):
    input = code
    instructions = "write a GitHub ready MD file of the code. Give back only the readme file. DO NOT INCLUDE CODE FROM THE PYTHON CODE PROVIDED. Exclude licence and user references like github-usernames and github-profiles-links etc."
    TextMD = DeepseekAPI(input, instructions)
    filename = "README.md"
    with open(filename, 'w', encoding='utf-8') as f:
        f.write(TextMD)

def Run():
    userinput = input("Prompt: ")
    code, libs = CodeDevelop(userinput)
    status, logresult, filename, code, libs = Runner(code, libs)
    if status == 400:
        fixedcode = CodeFixer(logresult, filename, code, libs)
        WriteMD(fixedcode)
    if status == 200:
        WriteMD(code)
    return 

if __name__ == "__main__":
    Run()