Spaces:
Running
Running
File size: 2,872 Bytes
6ccea25 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
import gradio as gr
import roop.globals
import ui.globals
camera_frame = None
def livecam_tab():
with gr.Tab("🎥 Live Cam"):
with gr.Row(variant='panel'):
gr.Markdown("""
This feature will allow you to use your physical webcam and apply the selected faces to the stream.
You can also forward the stream to a virtual camera, which can be used in video calls or streaming software.<br />
Supported are: v4l2loopback (linux), OBS Virtual Camera (macOS/Windows) and unitycapture (Windows).<br />
**Please note:** to change the face or any other settings you need to stop and restart a running live cam.
""")
with gr.Row(variant='panel'):
with gr.Column():
bt_start = gr.Button("▶ Start", variant='primary')
with gr.Column():
bt_stop = gr.Button("⏹ Stop", variant='secondary', interactive=False)
with gr.Column():
camera_num = gr.Slider(0, 8, value=0, label="Camera Number", step=1.0, interactive=True)
cb_obs = gr.Checkbox(label="Forward stream to virtual camera", interactive=True)
with gr.Column():
dd_reso = gr.Dropdown(choices=["640x480","1280x720", "1920x1080"], value="1280x720", label="Fake Camera Resolution", interactive=True)
cb_xseg = gr.Checkbox(label="Use DFL Xseg masking", interactive=True, value=True)
cb_mouthrestore = gr.Checkbox(label="Restore original mouth area", interactive=True, value=False)
with gr.Row():
fake_cam_image = gr.Image(label='Fake Camera Output', interactive=False, format="jpeg")
start_event = bt_start.click(fn=start_cam, inputs=[ui.globals.ui_selected_swap_model, cb_obs, cb_xseg, cb_mouthrestore, camera_num, dd_reso, ui.globals.ui_selected_enhancer, ui.globals.ui_blend_ratio, ui.globals.ui_upscale],outputs=[bt_start, bt_stop,fake_cam_image])
bt_stop.click(fn=stop_swap, cancels=[start_event], outputs=[bt_start, bt_stop], queue=False)
def start_cam(swap_model, stream_to_obs, use_xseg, use_mouthrestore, cam, reso, enhancer, blend_ratio, upscale):
from roop.virtualcam import start_virtual_cam
from roop.utilities import convert_to_gradio
roop.globals.selected_enhancer = enhancer
roop.globals.blend_ratio = blend_ratio
roop.globals.subsample_size = int(upscale[:3])
start_virtual_cam(swap_model, stream_to_obs, use_xseg, use_mouthrestore, cam, reso)
while True:
yield gr.Button(interactive=False), gr.Button(interactive=True), convert_to_gradio(ui.globals.ui_camera_frame)
def stop_swap():
from roop.virtualcam import stop_virtual_cam
stop_virtual_cam()
return gr.Button(interactive=True), gr.Button(interactive=False)
|