Update app.py

app.py

@@ -1,50 +1,57 @@
 import gradio as gr
+from docx import Document
 from PyPDF2 import PdfFileReader, PdfFileWriter
-import docx
-import io
-
-def pdf_to_word(pdf_file):
-    pdfReader = PdfFileReader(pdf_file)
-    doc = docx.Document()
-    for page_num in range(pdfReader.numPages):
-        page = pdfReader.getPage(page_num)
-        text = page.extractText()
-        doc.add_paragraph(text)
-    
-    doc_io = io.BytesIO()  # Criar um buffer de memória
-    doc.save(doc_io)  # Salvar o documento no buffer
-    doc_io.seek(0)  # Voltar ao início do buffer
-    return doc_io.getvalue()  # Retornar o conteúdo binário
-
-
-def word_to_pdf(docx_file):
-    doc = docx.Document(docx_file)
-    pdf_writer = PdfFileWriter()
-    packet = io.BytesIO()
-
-    for para in doc.paragraphs:
-        packet.write(para.text.encode('utf-8'))
-
-    pdf_writer.addBlankPage()
-    packet.seek(0)
-    return packet.getvalue()
-
-with gr.Blocks() as demo:
-    gr.Markdown("PDF <--> Word Converter")
-    with gr.Row():
-        with gr.Column():
-            gr.Markdown("### PDF to Word")
-            pdf_input = gr.File(label="Upload PDF")
-            word_output = gr.File(label="Download Word", type="binary")
-            pdf_to_word_btn = gr.Button("Convert")
-        with gr.Column():
-            gr.Markdown("### Word to PDF")
-            word_input = gr.File(label="Upload Word")
-            pdf_output = gr.File(label="Download PDF", type="binary")
-            word_to_pdf_btn = gr.Button("Convert")
-
-    pdf_to_word_btn.click(pdf_to_word, inputs=[pdf_input], outputs=[word_output])
-    word_to_pdf_btn.click(word_to_pdf, inputs=[word_input], outputs=[pdf_output])
-
-demo.launch()
 
+def converter_pdf_para_docx(caminho_pdf):
+  """Converte um arquivo PDF para DOCX."""
+  leitor_pdf = PdfFileReader(caminho_pdf)
+  escritor_docx = Document()
+  for pagina in leitor_pdf.pages:
+    texto_pagina = pagina.extractText()
+    escritor_docx.add_paragraph(texto_pagina)
+  return escritor_docx
+
+def converter_docx_para_pdf(caminho_docx):
+  """Converte um arquivo DOCX para PDF."""
+  escritor_pdf = PdfFileWriter()
+  leitor_docx = Document(caminho_docx)
+  for paragrafo in leitor_docx.paragraphs:
+    escritor_pdf.addPage(PdfPage().createBlankPage())
+    escritor_pdf.getPage(0).drawText(paragrafo.text)
+  return escritor_pdf
+
+def salvar_arquivo(escritor, formato, nome_arquivo):
+  """Salva o documento no formato especificado."""
+  if formato == "docx":
+    escritor.save(f"{nome_arquivo}.docx")
+  elif formato == "pdf":
+    escritor.write(f"{nome_arquivo}.pdf")
+
+# Interface Gradio
+# Entrada
+caminho_arquivo = gr.File(label="Selecione o arquivo")
+
+# Opções de conversão
+opcoes_conversao = gr.Radio(choices=["PDF para Word", "Word para PDF"], label="Conversão")
+
+# Saída
+nome_arquivo = gr.Text(label="Nome do arquivo")
+
+# Botão para converter
+botao_converter = gr.Button(value="Converter")
+
+# Função para executar a conversão
+def converter(caminho, tipo_conversao, nome):
+  if tipo_conversao == "PDF para Word":
+    escritor = converter_pdf_para_docx(caminho)
+    formato = "docx"
+  else:
+    escritor = converter_docx_para_pdf(caminho)
+    formato = "pdf"
+  salvar_arquivo(escritor, formato, nome)
+
+# Interface
+interface = gr.Interface(fn=converter, inputs=[caminho_arquivo, opcoes_conversao], outputs=[nome_arquivo], title="Conversor de PDF/Word", theme="default")
+
+# Exibir a interface
+interface.launch()