Update app.py

app.py

@@ -184,86 +184,95 @@ def swap_face(source_file, target_file, doFaceEnhancer):
             shutil.rmtree(folder_path)
         raise gr.Error(f"Face swap failed: {str(e)}")
 
-# Create custom style
-custom_css = """
-.container {
-    max-width: 1200px;
-    margin: auto;
-    padding: 20px;
-}
-.output-image {
-    min-height: 400px;
-    border: 1px solid #ccc;
-    border-radius: 8px;
-    padding: 10px;
-}
-"""
+def create_interface():
+    # Create custom style
+    custom_css = """
+    .container {
+        max-width: 1200px;
+        margin: auto;
+        padding: 20px;
+    }
+    .output-image {
+        min-height: 400px;
+        border: 1px solid #ccc;
+        border-radius: 8px;
+        padding: 10px;
+    }
+    """
 
-# Gradio interface setup
-title = "Face - Интегратор"
-description = r"""
-The application will save the image history to Hugging Face dataset using the environment variable token.
-Please upload source and target images to begin the face swap process.
-"""
-article = r"""
-<div style="text-align: center; max-width: 650px; margin: 40px auto;">
-    <p>
-        This tool performs face swapping with optional enhancement.
-        The processed images are automatically saved to the Hugging Face dataset.
-    </p>
-</div>
-"""
+    # Gradio interface setup
+    title = "Face - Интегратор"
+    description = r"""
+    The application will save the image history to Hugging Face dataset using the environment variable token.
+    Please upload source and target images to begin the face swap process.
+    """
+    article = r"""
+    <div style="text-align: center; max-width: 650px; margin: 40px auto;">
+        <p>
+            This tool performs face swapping with optional enhancement.
+            The processed images are automatically saved to the Hugging Face dataset.
+        </p>
+    </div>
+    """
 
-# Create Gradio interface with improved layout
-with gr.Blocks(title=title, css=custom_css) as app:
-    gr.Markdown(f"<h1 style='text-align: center;'>{title}</h1>")
-    gr.Markdown(description)
-    
-    with gr.Row():
-        with gr.Column(scale=1):
-            source_image = gr.Image(
-                label="Source Image",
-                type="numpy",
-                tool="upload"
-            )
-            
-        with gr.Column(scale=1):
-            target_image = gr.Image(
-                label="Target Image",
-                type="numpy",
-                tool="upload"
+    # Create Gradio interface with improved layout
+    with gr.Blocks(title=title, css=custom_css) as app:
+        gr.Markdown(f"<h1 style='text-align: center;'>{title}</h1>")
+        gr.Markdown(description)
+        
+        with gr.Row():
+            with gr.Column(scale=1):
+                source_image = gr.Image(
+                    label="Source Image",
+                    type="numpy",
+                    sources=["upload"]
+                )
+                
+            with gr.Column(scale=1):
+                target_image = gr.Image(
+                    label="Target Image",
+                    type="numpy",
+                    sources=["upload"]
+                )
+                
+            with gr.Column(scale=1):
+                output_image = gr.Image(
+                    label="Output Image",
+                    type="numpy",
+                    interactive=False,
+                    elem_classes="output-image"
+                )
+        
+        with gr.Row():
+            enhance_checkbox = gr.Checkbox(
+                label="Применить алгоритм?",
+                info="Улучшение качества изображения",
+                value=False
             )
             
-        with gr.Column(scale=1):
-            output_image = gr.Image(
-                label="Output Image",
-                type="numpy",
-                elem_classes="output-image"
+        with gr.Row():
+            process_btn = gr.Button(
+                "Process Face Swap",
+                variant="primary",
+                size="lg"
             )
-    
-    with gr.Row():
-        enhance_checkbox = gr.Checkbox(
-            label="Применить алгоритм?",
-            info="Улучшение качества изображения",
-            value=False
-        )
         
-    with gr.Row():
-        process_btn = gr.Button(
-            "Process Face Swap",
-            variant="primary",
-            size="lg"
+        # Set up the processing event
+        process_btn.click(
+            fn=swap_face,
+            inputs=[source_image, target_image, enhance_checkbox],
+            outputs=output_image,
+            api_name="swap_face"
         )
+        
+        gr.Markdown(article)
     
-    # Set up the processing event
-    process_btn.click(
-        fn=swap_face,
-        inputs=[source_image, target_image, enhance_checkbox],
-        outputs=output_image,
-        api_name="swap_face"
-    )
-    
-    gr.Markdown(article)
+    return app
+
+def main():
+    # Create and launch the interface
+    app = create_interface()
+    app.launch(share=False)
 
-# Launch the application
-app.launch(share=False)
+if __name__ == "__main__":
+    main()