Update app.py

app.py

@@ -1,72 +1,69 @@
-# -* coding:UTF-8 -*
-# !/usr/bin/env python
-import numpy as np
 import gradio as gr
-import roop.globals
-from roop.core import (
-    start,
-    decode_execution_providers,
-    suggest_max_memory,
-    suggest_execution_threads,
-)
-from roop.processors.frame.core import get_frame_processors_modules
-from roop.utilities import normalize_output_path
+from pdf2docx import Converter
+from docx import Document
+from fpdf import FPDF
 import os
-from PIL import Image
 
+title_and_description = """
+# PDF to Word and Word to PDF
+"""
 
-def swap_face(source_file, target_file,doFaceEnhancer):
+def pdf_to_word(pdf_file):
+    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
 
-    source_path = "input.jpg"
-    target_path = "target.jpg"
+def word_to_pdf(docx_file):
+    pdf_filename = "output.pdf"
+    
+    doc = Document(docx_file.name)
+    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)
 
-    source_image = Image.fromarray(source_file)
-    source_image.save(source_path)
-    target_image = Image.fromarray(target_file)
-    target_image.save(target_path)
+    for para in doc.paragraphs:
+        text = para.text.strip()
+        if not text:
+            continue
 
-    print("source_path: ", source_path)
-    print("target_path: ", target_path)
+        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)
 
-    roop.globals.source_path = source_path
-    roop.globals.target_path = target_path
-    output_path = "output.jpg"
-    roop.globals.output_path = normalize_output_path(
-        roop.globals.source_path, roop.globals.target_path, output_path
-    )
-    if doFaceEnhancer == True:
-        roop.globals.frame_processors = ["face_swapper","face_enhancer"]
-    else:
-        roop.globals.frame_processors = ["face_swapper"]
-    roop.globals.headless = True
-    roop.globals.keep_fps = True
-    roop.globals.keep_audio = True
-    roop.globals.keep_frames = False
-    roop.globals.many_faces = False
-    roop.globals.video_encoder = "libx264"
-    roop.globals.video_quality = 18
-    roop.globals.max_memory = suggest_max_memory()
-    roop.globals.execution_providers = decode_execution_providers(["cuda"])
-    roop.globals.execution_threads = suggest_execution_threads()
+    pdf.output(pdf_filename)
+    return pdf_filename
 
-    print(
-        "start process",
-        roop.globals.source_path,
-        roop.globals.target_path,
-        roop.globals.output_path,
-    )
+with gr.Blocks(theme="xiaobaiyuan/theme_brief") 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])
 
-    for frame_processor in get_frame_processors_modules(
-        roop.globals.frame_processors
-    ):
-        if not frame_processor.pre_check():
-            return
-
-    start()
-    return output_path
-
-
-app = gr.Interface(
-    fn=swap_face, inputs=[gr.Image(), gr.Image(),gr.Checkbox(label="face_enhancer?", info="do face enhancer?")], outputs="image"
-)
-app.launch()
+app.launch()