Add temp frame component to UI

roop/typing.py

@@ -11,4 +11,4 @@
 FaceAnalyserGender = Literal['male', 'female']
 TempFrameFormat = Literal['jpg', 'png']
 OutputVideoEncoder = Literal['libx264', 'libx265', 'libvpx-vp9', 'h264_nvenc', 'hevc_nvenc']
-LogLevel = Literal['error']
+LogLevel = Literal['error']

roop/uis/__components__/output.py

@@ -22,14 +22,15 @@ def render() -> None:
  with gradio.Row():
  START_BUTTON = gradio.Button('Start')
  CLEAR_BUTTON = gradio.Button('Clear')
- OUTPUT_IMAGE = gradio.Image(
- label='OUTPUT',
- visible=False
- )
- OUTPUT_VIDEO = gradio.Video(
- label='OUTPUT',
- visible=False
- )
+ with gradio.Row():
+ with gradio.Box():
+ OUTPUT_IMAGE = gradio.Image(
+ label='OUTPUT'
+ )
+ OUTPUT_VIDEO = gradio.Video(
+ label='OUTPUT',
+ visible=False
+ )
 
 
 def listen() -> None:

roop/uis/__components__/temp_frame.py

@@ -0,0 +1,43 @@
+from typing import List, Optional
+import gradio
+
+import roop.globals
+from roop.processors.frame.core import list_frame_processors_names, load_frame_processor_module, clear_frame_processors_modules
+from roop.typing import TempFrameFormat
+from roop.uis import core as ui
+from roop.uis.typing import Update
+
+TEMP_FRAME_FORMAT_DROPDOWN: Optional[gradio.Dropdown] = None
+TEMP_FRAME_QUALITY_SLIDER: Optional[gradio.Slider] = None
+
+
+def render() -> None:
+ global TEMP_FRAME_FORMAT_DROPDOWN
+ global TEMP_FRAME_QUALITY_SLIDER
+
+ with gradio.Box():
+ TEMP_FRAME_FORMAT_DROPDOWN = gradio.Dropdown(
+ label='TEMP FRAME FORMAT',
+ choices=['jpg', 'png'],
+ value=roop.globals.temp_frame_format
+ )
+ TEMP_FRAME_QUALITY_SLIDER = gradio.Slider(
+ label='TEMP FRAME QUALITY',
+ value=roop.globals.temp_frame_quality,
+ step=1
+ )
+
+
+def listen() -> None:
+ TEMP_FRAME_FORMAT_DROPDOWN.select(update_temp_frame_format, inputs=TEMP_FRAME_FORMAT_DROPDOWN, outputs=TEMP_FRAME_FORMAT_DROPDOWN)
+ TEMP_FRAME_QUALITY_SLIDER.change(update_temp_frame_quality, inputs=TEMP_FRAME_QUALITY_SLIDER, outputs=TEMP_FRAME_QUALITY_SLIDER)
+
+
+def update_temp_frame_format(temp_frame_format: TempFrameFormat) -> Update:
+ roop.globals.temp_frame_format = temp_frame_format
+ return gradio.update(value=temp_frame_format)
+
+
+def update_temp_frame_quality(temp_frame_quality: int) -> Update:
+ roop.globals.temp_frame_quality = temp_frame_quality
+ return gradio.update(value=temp_frame_quality)

roop/uis/__layouts__/default.py

@@ -1,32 +1,33 @@
 import gradio
 
-from roop.uis.__components__ import processors, execution, settings, source, target, preview, trim_frame, face_analyser, face_selector, output
+from roop.uis.__components__ import processor, execution, temp_frame, settings, source, target, preview, trim_frame, face_analyser, face_selector, output
 
 
 def render() -> gradio.Blocks:
  with gradio.Blocks() as layout:
  with gradio.Row():
  with gradio.Column(scale=2):
- processors.render()
+ processor.render()
  execution.render()
+ temp_frame.render()
  settings.render()
  with gradio.Column(scale=1):
  source.render()
  target.render()
+ output.render()
  with gradio.Column(scale=3):
  preview.render()
  trim_frame.render()
  face_selector.render()
  face_analyser.render()
- with gradio.Row():
- output.render()
  return layout
 
 
 def listen() -> None:
- processors.listen()
+ processor.listen()
  execution.listen()
  settings.listen()
+ temp_frame.listen()
  source.listen()
  target.listen()
  preview.listen()

roop/uis/core.py

@@ -19,7 +19,7 @@
 
 
 def init() -> None:
- with gradio.Blocks(theme=get_theme(), title=roop.metadata.name) as ui:
+ with gradio.Blocks(theme=get_theme(), title=roop.metadata.name + ' ' + roop.metadata.version) as ui:
  for ui_layout in roop.globals.ui_layouts:
  ui_layout_module = load_ui_layout_module(ui_layout)
  ui_layout_module.render()