Update app.py

app.py

@@ -1,66 +1,69 @@
 import gradio as gr
-from reportlab.lib.pagesizes import letter
-from reportlab.pdfgen import canvas
+from pdf2docx import Converter
 from docx import Document
+from fpdf import FPDF
 import os
 
-def word_to_pdf(word_file):
-    try:
-        doc = Document(word_file.name)
-        pdf_file = "output.pdf"
-        c = canvas.Canvas(pdf_file, pagesize=letter)
-        width, height = letter
-
-        for paragraph in doc.paragraphs:
-            text = paragraph.text
-            if text.strip():
-                c.drawString(72, height - 72, text)
-                height -= 14
-                if height < 72:  # Create a new page if space runs out
-                    c.showPage()
-                    height = letter[1]
-
-        c.save()
-        return pdf_file
-    except Exception as e:
-        return f"Error: {e}"
+title_and_description = """
+# PDF to Word and Word to PDF converter
+"""
 
 def pdf_to_word(pdf_file):
-    try:
-        from PyPDF2 import PdfReader
-
-        reader = PdfReader(pdf_file.name)
-        doc = Document()
-
-        for page in reader.pages:
-            text = page.extract_text()
-            if text.strip():
-                doc.add_paragraph(text)
-
-        word_file = "output.docx"
-        doc.save(word_file)
-        return word_file
-    except Exception as e:
-        return f"Error: {e}"
-
-def interface():
-    with gr.Blocks() as app:
-        gr.Markdown("## PDF to Word and Word to PDF Converter")
+    docx_filename = pdf_file.name.replace('.pdf', '.docx')
+    
+    cv = Converter(pdf_file.name)
+    cv.convert(docx_filename, multi_processing=True, start=0, end=None)
+    cv.close()
+    
+    return docx_filename
 
-        with gr.Tab("Word to PDF"):
-            word_input = gr.File(label="Upload Word File", file_types=[".docx"])
-            pdf_output = gr.File(label="Download PDF File")
-            word_to_pdf_button = gr.Button("Convert Word to PDF")
+def word_to_pdf(docx_file):
+    pdf_filename = "output.pdf"
+    
+    doc = Document(docx_file)
+    pdf = FPDF(format='A4')
+    pdf.set_auto_page_break(auto=True, margin=15)
+    pdf.add_page()
+    pdf.add_font('Arial', '', 'Arial.ttf', uni=True)
+    pdf.set_font('Arial', size=12)
 
-        with gr.Tab("PDF to Word"):
-            pdf_input = gr.File(label="Upload PDF File", file_types=[".pdf"])
-            word_output = gr.File(label="Download Word File")
-            pdf_to_word_button = gr.Button("Convert PDF to Word")
+    for para in doc.paragraphs:
+        text = para.text.strip()
+        if not text:  # Ignorar linhas vazias
+            continue
+        # Quebrar o texto em várias linhas se necessário
+        words = text.split()
+        line = ''
+        for word in words:
+            if pdf.get_string_width(line + word) < (pdf.w - 2 * pdf.l_margin):
+                line += word + ' '
+            else:
+                pdf.cell(0, 10, line, ln=True)
+                line = word + ' '
+        if line:
+            pdf.cell(0, 10, line, ln=True)
 
-        word_to_pdf_button.click(word_to_pdf, inputs=word_input, outputs=pdf_output)
-        pdf_to_word_button.click(pdf_to_word, inputs=pdf_input, outputs=word_output)
+    pdf.output(pdf_filename)
+    return pdf_filename
 
-    return app
+with gr.Blocks() as app:
+    gr.Markdown(title_and_description)
+    
+    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="filepath")
+                
+                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="filepath")
+                
+                convert_word_to_pdf.click(word_to_pdf, inputs=[word_input], outputs=[pdf_output])
 
-app = interface()
 app.launch()