Update app.py

app.py

@@ -1,57 +1,50 @@
 import gradio as gr
+from pdf2docx import Converter
+from fpdf import FPDF
 from docx import Document
-from PyPDF2 import PdfFileReader, PdfFileWriter
-
-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()
+import os
+
+def pdf_to_word(pdf_file):
+    docx_filename = pdf_file.name.replace('.pdf', '.docx')
+    
+    cv = Converter(pdf_file.name)
+    cv.convert(docx_filename, start=0, end=None)
+    cv.close()
+    
+    return docx_filename
+
+def word_to_pdf(docx_file):
+    pdf_filename = docx_file.name.replace('.docx', '.pdf')
+    
+    document = Document(docx_file.name)
+    pdf = FPDF()
+
+    pdf.add_page()
+    
+    for paragraph in document.paragraphs:
+        pdf.set_font("Arial", size = 12)
+        pdf.multi_cell(0, 10, paragraph.text)
+    
+    pdf.output(pdf_filename)
+    
+    return pdf_filename
+
+with gr.Blocks() as app:
+    with gr.Row():
+        with gr.Column():
+            with gr.Accordion("PDF to Word"):
+                pdf_input = gr.File(label="Upload PDF")
+                convert_pdf_to_word = gr.Button("Convert to Word")
+                word_output = gr.File(label="Download Word File", type="file")
+                
+                convert_pdf_to_word.click(pdf_to_word, inputs=[pdf_input], outputs=[word_output])
+                
+        with gr.Column():
+            with gr.Accordion("Word to PDF"):
+                word_input = gr.File(label="Upload Word")
+                convert_word_to_pdf = gr.Button("Convert to PDF")
+                pdf_output = gr.File(label="Download PDF File", type="file")
+                
+                convert_word_to_pdf.click(word_to_pdf, inputs=[word_input], outputs=[pdf_output])
+
+app.launch()