ValdeciRodrigues/Python-code-assistant

1# logic/refactor.py2from logic.generator import tokenizer, model3import torch4 5def refactor_code(code):6    try:7        prompt = f"# Refatore o seguinte código Python para melhorar clareza e desempenho:\n{code}\n# Código refatorado:\n"8        inputs = tokenizer(prompt, return_tensors="pt").to(model.device)9        outputs = model.generate(10            **inputs,11            max_new_tokens=256,12            do_sample=True,13            temperature=0.3,14            top_k=50,15            top_p=0.9516        )17        result = tokenizer.decode(outputs[0], skip_special_tokens=True)18        return result.strip()19    except Exception as e:20        return f"Erro ao tentar refatorar o código: {str(e)}"21