import gradio as gr
import transformers

pipe = transformers.pipeline("text2text-generation", model="ejschwartz/slade-x86-O3")

def predict(asm):
    o = pipe(asm, max_length=1024, num_beams=5, num_return_sequences=1)
    return o[0]['generated_text']

demo = gr.Interface(fn=predict,
                    examples=[".globl sysv_calc\n.type sysv_calc, @function\nsysv_calc:\n.LFB0:\n\t.cfi_startproc\n\tendbr64\n\tpushq\t%rbp\n\t.cfi_def_cfa_offset 16\n\t.cfi_offset 6, -16\n\tpushq\t%rbx\n\t.cfi_def_cfa_offset 24\n\t.cfi_offset 3, -24\n\tmovq\t%rdx, %rbx\n\tsubq\t$8, %rsp\n\t.cfi_def_cfa_offset 32\n\tmovq\t24(%rdx), %rdx\n\tmovl\t(%rsi), %esi\n\tcall\telf_strptr@PLT\n\tmovq\t%rax, %rbp\n\tmovq\t(%rbx), %rax\n\tcmpq\tSHT_SYMTAB(%rip), %rax\n\tje\t.L2\n\tcmpq\tSHT_STRTAB(%rip), %rax\n\tje\t.L2\n\tcmpq\tSHT_RELA(%rip), %rax\n\tje\t.L2\n\tcmpq\tSHT_REL(%rip), %rax\n\tjne\t.L3\n\t.p2align 4,,10\n\t.p2align 3\n.L2:\n\tcmpq\t$0, 8(%rbx)\n\tje\t.L1\n.L3:\n\txorl\t%eax, %eax\n\tcall\ttbl_append@PLT\n\txorl\t%esi, %esi\n\tmovq\t%rbp, %rdi\n\tcall\ttbl_print@PLT\n\tmovq\t16(%rbx), %rdi\n\tmovl\tradix(%rip), %esi\n\tmovl\t$1, %edx\n\tcall\ttbl_print_num@PLT\n\tmovq\t8(%rbx), %rdi\n\tmovl\tradix(%rip), %esi\n\tmovl\t$2, %edx\n\tcall\ttbl_print_num@PLT\n\tmovq\t16(%rbx), %rax\n\taddl\t%eax, text_size_total(%rip)\n.L1:\n\taddq\t$8, %rsp\n\t.cfi_def_cfa_offset 24\n\tpopq\t%rbx\n\t.cfi_def_cfa_offset 16\n\tpopq\t%rbp\n\t.cfi_def_cfa_offset 8\n\tret\n\t.cfi_endproc\n\n"],
                    inputs="text", outputs="text")
demo.queue()
demo.launch()