Update app.py

app.py

@@ -1,66 +1,66 @@
 import gradio as gr
-import fitz  # PyMuPDF
-from docx import Document
-from reportlab.pdfgen import canvas
 from reportlab.lib.pagesizes import letter
-import io
+from reportlab.pdfgen import canvas
+from docx import Document
+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}"
 
-# PDF to Word conversion function
 def pdf_to_word(pdf_file):
-    # Read PDF file using PyMuPDF
-    doc = fitz.open(stream=pdf_file.read(), filetype="pdf")
-    text = ""
-    for page in doc:
-        text += page.get_text()
-    
-    # Create a Word document using python-docx
-    docx = Document()
-    docx.add_paragraph(text)
-    
-    # Save the Word document to a bytes buffer
-    buffer = io.BytesIO()
-    docx.save(buffer)
-    buffer.seek(0)
-    
-    return buffer, "converted.docx"
+    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")
+
+        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")
+
+        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")
 
-# Word to PDF conversion function
-def word_to_pdf(docx_file):
-    # Read Word document using python-docx
-    doc = Document(docx_file)
-    text = ""
-    for para in doc.paragraphs:
-        text += para.text + "\n"
-    
-    # Create a PDF using reportlab
-    buffer = io.BytesIO()
-    c = canvas.Canvas(buffer, pagesize=letter)
-    textobject = c.beginText()
-    textobject.setTextOrigin(50, 750)
-    lines = text.split('\n')
-    for line in lines:
-        textobject.textLine(line)
-    c.drawText(textobject)
-    c.showPage()
-    c.save()
-    buffer.seek(0)
-    
-    return buffer, "converted.pdf"
+        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)
 
-# Gradio interface
-with gr.Blocks() as demo:
-    gr.Markdown("<h1>PDF to Word and Word to PDF Converter</h1>")
-    
-    with gr.Tab("PDF to Word"):
-        pdf_input = gr.File(label="Upload PDF File", type="file")
-        pdf_convert_btn = gr.Button("Convert to Word")
-        word_output = gr.File(label="Download Word File")
-        pdf_convert_btn.click(pdf_to_word, inputs=pdf_input, outputs=word_output)
-    
-    with gr.Tab("Word to PDF"):
-        word_input = gr.File(label="Upload Word File", type="file")
-        word_convert_btn = gr.Button("Convert to PDF")
-        pdf_output = gr.File(label="Download PDF File")
-        word_convert_btn.click(word_to_pdf, inputs=word_input, outputs=pdf_output)
+    return app
 
-demo.launch()
+app = interface()
+app.launch()