Develop (#73)

* Cosmetic changes

* Cosmetic changes

* Run single warm up for the benchmark suite

* Use latest version of Gradio

* More testing

* Introduce basic installer

* Fix typo

* Move more to installer file

* Fix the installer with the uninstall all trick

* Adjust wording

* Fix coreml in installer

* Allow Pyhton 3.9

* Add VENV to installer

* Just some cosmetics

* Just some cosmetics

* Dedicated headless mode, Refine API of UI layouts

* Use --headless for pytest

* Fix testing for Windows

* Normalize output path that lacks extension

* Fix CI for Windows

* Fix CI for Windows

* UI to change output path

* Add conda support for the installer

* Improve installer quite a bit

* Drop conda support

* Install community wheels for coreml silicon

* Improve output video component

* Fix silicon wheel downloading

* Remove venv from installer as we cannot activate via subprocess

* Use join to create wheel name

* Refine the output path normalization

* Refine the output path normalization

* Introduce ProcessMode and rename some methods

* Introduce ProcessMode and rename some methods

* Basic webcam integration and open_ffmpeg()

* Basic webcam integration part2

* Benchmark resolutions now selectable

* Rename benchmark resolution back to benchmark runs

* Fix repeating output path in UI

* Keep output_path untouched if not resolvable

* Add more cases to normalize output path

* None for those tests that don't take source path into account

* Finish basic webcam integration, UI layout now with custom run()

* Fix CI and hide link in webcam UI

* Cosmetics on webcam UI

* Move get_device to utilities

* Fix CI

* Introduce output-image-quality, Show and hide UI according to target media type

* Benchmark with partial result updates

* fix: trim frame sliders not appearing after draggin video

* fix: output and temp frame setting inputs not appearing

* Fix: set increased update delay to 250ms to let Gradio update conditional inputs properly

* Reverted .gitignore

* Adjust timings

* Remove timeout hacks and get fully event driven

* Update dependencies

* Update dependencies

* Revert NSFW library, Conditional unset trim args

* Face selector works better on preview slider release

* Add limit resources to UI

* Introduce vision.py for all CV2 operations, Rename some methods

* Add restoring audio failed

* Decouple updates for preview image and preview frame slider, Move reduce_preview_frame to vision

* Refactor detect_fps based on JSON output

* Only webcam when open

* More conditions to vision.py

* Add udp and v4l2 streaming to webcam UI

* Detect v4l2 device to be used

* Refactor code a bit

* Use static max memory for UI

* Fix CI

* Looks stable to me

* Update preview

* Update preview

---------

Co-authored-by: Sumit <vizsumit@gmail.com>

facefusion/installer.py

@@ -0,0 +1,49 @@
+from typing import Dict, Tuple
+import os
+import sys
+import subprocess
+import tempfile
+
+subprocess.call([ 'pip', 'install' , 'inquirer', '-q' ])
+
+import inquirer
+
+from facefusion import wording
+
+ONNXRUNTIMES : Dict[str, Tuple[str, str]] =\
+{
+	'cpu': ('onnxruntime', '1.15.1'),
+	'cuda': ('onnxruntime-gpu', '1.15.1'),
+	'coreml-legacy': ('onnxruntime-coreml', '1.13.1'),
+	'coreml-silicon': ('onnxruntime-silicon', '1.14.2'),
+	'directml': ('onnxruntime-directml', '1.15.1'),
+	'openvino': ('onnxruntime-openvino', '1.15.0')
+}
+
+
+def run() -> None:
+	answers : Dict[str, str] = inquirer.prompt(
+	[
+		inquirer.List(
+			'onnxruntime_key',
+			message = wording.get('select_onnxruntime_install'),
+			choices = list(ONNXRUNTIMES.keys())
+		)
+	])
+
+	if answers is not None:
+		onnxruntime_key = answers['onnxruntime_key']
+		onnxruntime_name, onnxruntime_version = ONNXRUNTIMES[onnxruntime_key]
+		python_id = 'cp' + str(sys.version_info.major) + str(sys.version_info.minor)
+		subprocess.call([ 'pip', 'install', '-r', 'requirements.txt' ])
+		if onnxruntime_key != 'cpu':
+			subprocess.call([ 'pip', 'uninstall', 'onnxruntime', onnxruntime_name, '-y' ])
+		if onnxruntime_key != 'coreml-silicon':
+			subprocess.call([ 'pip', 'install', onnxruntime_name + '==' + onnxruntime_version ])
+		elif python_id in [ 'cp39', 'cp310', 'cp311' ]:
+			wheel_name = '-'.join([ 'onnxruntime_silicon', onnxruntime_version, python_id, python_id, 'macosx_12_0_arm64.whl' ])
+			wheel_path = os.path.join(tempfile.gettempdir(), wheel_name)
+			wheel_url = 'https://github.com/cansik/onnxruntime-silicon/releases/download/v' + onnxruntime_version + '/' + wheel_name
+			subprocess.call([ 'curl', wheel_url, '-o', wheel_path, '-L' ])
+			subprocess.call([ 'pip', 'install', wheel_path ])
+			os.remove(wheel_path)