testing_4_amm / app.py
markerz's picture
Update app.py
1adc7c1 verified
import os
import sys
import warnings
from pathlib import Path
# Setup environment
os.environ['GRADIO_ANALYTICS_ENABLED'] = '0'
os.environ['SYSTEM_VERSION_COMPAT'] = '0'
os.environ['OMP_NUM_THREADS'] = '1'
# Initialize paths
REPO_DIR = Path(__file__).parent
sys.path.append(str(REPO_DIR))
import gradio as gr
from facefusion import core, metadata, state_manager, wording
from facefusion.uis.layouts import default
from facefusion.uis.components import (
source, preview, output_options, webcam_options
)
# Suppress warnings
warnings.filterwarnings('ignore', category=UserWarning, module='gradio')
def create_facefusion_app():
"""Creates the FaceFusion Gradio application with all features"""
# Set initial state
state_manager.set_item('ui_layouts', ['default'])
# Configure gradio temp directory
os.environ['GRADIO_TEMP_DIR'] = os.path.join(
state_manager.get_item('temp_path'),
'gradio'
)
# Create interface using the theme from core
with gr.Blocks(
theme=core.get_theme(),
css=core.get_css(),
title=f"{metadata.get('name')} {metadata.get('version')}",
analytics_enabled=False
) as app:
# Pre-check and render the default layout
default.pre_check()
default.render()
default.listen()
# Run layout setup
default.run(app)
return app
# Create the app
app = create_facefusion_app()
# For local testing
if __name__ == "__main__":
app.launch()