Update app.py

app.py

@@ -1,69 +1,76 @@
+# -* coding:UTF-8 -*
+# !/usr/bin/env python
+import numpy as np
 import gradio as gr
-from pdf2docx import Converter
-from docx import Document
-from fpdf import FPDF
+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
 import os
+from PIL import Image
 
-title_and_description = """
-# PDF to Word & Word to PDF
-"""
 
-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
+def swap_face(source_file, target_file, doFaceEnhancer):
 
-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_path = "input.jpg"
+    target_path = "target.jpg"
 
-    for para in doc.paragraphs:
-        text = para.text.strip()
-        if not text:
-            continue
+    source_image = Image.fromarray(source_file)
+    source_image.save(source_path)
+    target_image = Image.fromarray(target_file)
+    target_image.save(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)
+    print("source_path: ", source_path)
+    print("target_path: ", target_path)
 
-    pdf.output(pdf_filename)
-    return pdf_filename
+    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()
 
-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])
+    print(
+        "start process",
+        roop.globals.source_path,
+        roop.globals.target_path,
+        roop.globals.output_path,
+    )
 
-app.launch()
+    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"
+)
+
+# Launch the Gradio interface with the specified theme
+app.launch(theme="xiaobaiyuan/theme_brief")