ValdeciRodrigues/Python-code-assistant

1# app.py2import gradio as gr3from logic.generator import generate_code, tokenizer, model4from logic.fixer import fix_code5from logic.refactor import refactor_code6from logic.file_reader import read_uploaded_file7import torch8 9def process_file_command(arquivos, prompt):10    if not arquivos:11        return "Nenhum arquivo foi enviado."12 13    conteudo_total = ""14    for file in arquivos:15        conteudo = read_uploaded_file(file)16        conteudo_total += f"\n# ===== {file.name} =====\n{conteudo}\n"17 18    comando = f"""19# Abaixo estão os códigos enviados:20{conteudo_total}21 22# Instrução: {prompt}23# Código resultante:24"""25 26    inputs = tokenizer(comando, return_tensors="pt", truncation=True, max_length=512).to(model.device)27    outputs = model.generate(28        **inputs,29        max_new_tokens=256,30        temperature=0.3,31        top_k=50,32        top_p=0.95,33        eos_token_id=tokenizer.eos_token_id34    )35    return tokenizer.decode(outputs[0], skip_special_tokens=True).strip()36 37with gr.Blocks(title="Python Code Assistant", css="#file-output textarea { overflow-y: scroll !important; }") as demo:38    gr.Markdown("# 🧠 Python Code Assistant\nUma IA especializada em programação Python — gere, corrija, refatore e interprete arquivos.")39 40    with gr.Tab("📌 Gerar Código"):41        with gr.Row():42            prompt = gr.Textbox(label="Descreva o que o código deve fazer", placeholder="Ex: Ler um CSV e plotar um gráfico", lines=3)43        output_gen = gr.Code(label="Código gerado")44        btn_gen = gr.Button("Gerar")45        btn_gen.click(fn=generate_code, inputs=prompt, outputs=output_gen)46 47    with gr.Tab("🛠️ Corrigir Código"):48        buggy_code = gr.Code(label="Cole o código com erro")49        output_fix = gr.Code(label="Código corrigido")50        btn_fix = gr.Button("Corrigir")51        btn_fix.click(fn=fix_code, inputs=buggy_code, outputs=output_fix)52 53    with gr.Tab("🧹 Refatorar Código"):54        original_code = gr.Code(label="Cole o código para refatorar")55        output_refac = gr.Code(label="Código refatorado")56        btn_refac = gr.Button("Refatorar")57        btn_refac.click(fn=refactor_code, inputs=original_code, outputs=output_refac)58 59    with gr.Tab("📂 Interpretar e Comandar Arquivos"):60        file_input = gr.File(61            label="Arraste ou selecione arquivos (.py, .txt, .zip)",62            file_types=[".py", ".txt", ".zip"],63            file_count="multiple"64        )65 66        file_summary = gr.Textbox(67            label="📄 Resumo dos arquivos enviados",68            lines=15,69            max_lines=1000,70            interactive=False,71            show_copy_button=True72        )73 74        command_file = gr.Textbox(75            label="💬 O que deseja fazer com os arquivos?",76            placeholder="Ex: Corrija todos e transforme em uma API com FastAPI"77        )78 79        command_output = gr.Code(label="⚙️ Resultado gerado pela IA")80        btn_command = gr.Button("Executar comando")81 82        # Resumo automático dos arquivos83        file_input.change(84            fn=lambda files: "\n\n".join(read_uploaded_file(f)[:1000] for f in files),85            inputs=file_input,86            outputs=file_summary87        )88 89        # Executa comando com base nos arquivos carregados90        btn_command.click(91            fn=process_file_command,92            inputs=[file_input, command_file],93            outputs=command_output94        )95 96demo.launch()97