Update app.py

app.py

@@ -15,35 +15,65 @@ import os
 from PIL import Image
 
 article_text = """
-<div style="text-align: center; padding: 20px; border-top: 1px solid #e0e0e0; margin-top: 30px;">
-    <h3>Leer zelf Generative AI-tools te bouwen en te begrijpen! 🚀</h3>
-    <a href="https://www.glossybranding.com/masterclass" target="_blank" style="
-        background: linear-gradient(45deg, #7B1FA2, #4A148C);
-        color: white;
-        padding: 12px 24px;
-        border-radius: 8px;
-        text-decoration: none;
-        font-weight: bold;
-        box-shadow: 0 4px 6px rgba(0,0,0,0.1);
-    ">Schrijf je in voor de AI Masterclass</a>
+<div style="text-align: center;">
+    <p>Leer zelf Generative AI-tools te bouwen en te begrijpen!</p>
+    <a href="https://www.glossybranding.com/masterclass">Schrijf je in voor de AI Masterclass</a>
 </div>
 """
 
-def swap_face(source_file, target_file, doFaceEnhancer):
-    # ... [jouw bestaande swap_face functie blijft identiek] ...
+
+def swap_face(source_file, target_file,doFaceEnhancer):
+
+    source_path = "input.jpg"
+    target_path = "target.jpg"
+
+    source_image = Image.fromarray(source_file)
+    source_image.save(source_path)
+    target_image = Image.fromarray(target_file)
+    target_image.save(target_path)
+
+    print("source_path: ", source_path)
+    print("target_path: ", target_path)
+
+    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()
+
+    print(
+        "start process",
+        roop.globals.source_path,
+        roop.globals.target_path,
+        roop.globals.output_path,
+    )
+
+    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(label="Upload jouw gezicht"), 
-        gr.Image(label="Kies een template afbeelding"),
-        gr.Checkbox(label="Gezicht verbeteren?", info="Aanvinken voor HD-kwaliteit")
-    ], 
-    outputs=gr.Image(label="Resultaat", show_share_button=True),
-    title="🎭 Glossy Branding AI Face Swap",
-    description="Transformeer jezelf in een AI-kunstwerk! Upload je foto en kies een template.",
-    article=article_text,  # 👈 Dit is de cruciale toevoeging
-    theme=gr.themes.Soft(primary_hue="purple")
+    fn=swap_face, inputs=[gr.Image(), gr.Image(),gr.Checkbox(label="face_enhancer?", info="do face enhancer?")], outputs="image", article=article_text,
 )
-
 app.launch()