| import os
|
| import pickle
|
| import streamlit as st
|
| from streamlit_option_menu import option_menu
|
| import tempfile
|
| from tqdm import tqdm
|
| import time
|
| import cv2
|
| import moviepy.editor as mp
|
| from PIL import Image
|
| from streamlit_image_select import image_select
|
| import glob
|
| import requests
|
| from streamlit_lottie import st_lottie
|
| from pytube import YouTube
|
| from FinalFaceSwapCode import run
|
|
|
| def load_lottieurl(url: str):
|
| r = requests.get(url)
|
| if r.status_code != 200:
|
| return None
|
| return r.json()
|
|
|
| def download_video(link):
|
| yt = YouTube(link)
|
| video = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first().download(filename="video.mp4")
|
| return video
|
|
|
| st.set_page_config(page_title="Face Swapper",
|
| layout="wide",
|
| page_icon="🧑⚕️")
|
|
|
| with st.sidebar:
|
| selected = option_menu('Image Options',
|
| ['Use Available Images',
|
| 'Upload Custom Image'],
|
| menu_icon='list',
|
| icons=['person', 'upload'],
|
| default_index=0)
|
| VidOptSel = option_menu('Video Options',
|
| ['Upload Custom Video',
|
| 'Input Youtube Url'],
|
| menu_icon='list',
|
| icons=['upload', 'person'],
|
| default_index=0)
|
|
|
| st.title('Face Swap using Inswapper')
|
| col1, col2 = st.columns([1, 3])
|
| url = "https://lottie.host/dd7d9b7b-7713-423a-b3a9-8c78c21a1644/yVdrKmczvO.json"
|
| with col1:
|
| lottie = load_lottieurl(url)
|
| st_lottie(lottie)
|
| with col2:
|
| st.write("""
|
| ## Face Swapper
|
| ##### Upload a video and replace the face in the video with available faces or upload a custom image for face swap.
|
| """)
|
|
|
| working_dir = os.path.dirname(os.path.abspath(__file__))
|
| male_images = os.path.join(working_dir, "male")
|
| female_images = os.path.join(working_dir, "female")
|
| output_vid = os.path.join(working_dir, "source-target.mp4")
|
|
|
| if selected == 'Use Available Images':
|
| col1, col2 = st.columns([0.5, 0.5])
|
| with col1:
|
| gender = option_menu('Select Gender',
|
| ['Male',
|
| 'Female'],
|
| menu_icon='person',
|
| icons=['gender-male', 'gender-female'],
|
| default_index=0)
|
| with col2:
|
| quality = option_menu('Select Output Quality',
|
| ['High',
|
| 'Normal'],
|
| menu_icon='person',
|
| default_index=1)
|
| if gender == 'Male':
|
| imgs = glob.glob((os.path.join(glob.escape(male_images), '*.' + 'jpg')))
|
| image = image_select(
|
| label="Male",
|
| images=imgs,
|
| captions=[os.path.splitext(os.path.basename(image))[0] for image in imgs],
|
| )
|
| if gender == 'Female':
|
| imgs = glob.glob((os.path.join(glob.escape(female_images), '*.' + 'jpeg')))
|
| image = image_select(
|
| label="Female",
|
| images=imgs,
|
| captions=[os.path.splitext(os.path.basename(image))[0] for image in imgs],
|
| )
|
| img = Image.open(image)
|
|
|
| if selected == 'Upload Custom Image':
|
| uploaded_file = st.file_uploader("Upload your images here...", type=['jpg', 'png', 'jpeg'])
|
| if uploaded_file is not None:
|
| file_extension = os.path.splitext(uploaded_file.name)[1].lower()
|
| if file_extension in ['.jpg', '.png', '.jpeg']:
|
| img = Image.open(uploaded_file)
|
| col1, col2 = st.columns([0.5, 0.5])
|
| with col1:
|
| st.image(img)
|
| with col2:
|
| quality = option_menu('Select Output Quality',
|
| ['High',
|
| 'Normal'],
|
| menu_icon='person',
|
| default_index=1)
|
|
|
| link = None
|
| uploaded_file = None
|
|
|
| if VidOptSel == 'Upload Custom Video':
|
| uploaded_file = st.file_uploader("Upload your video here...", type=['mp4', 'mov', 'avi', 'mkv'])
|
| if VidOptSel == 'Input Youtube Url':
|
| link = st.text_input("YouTube Link (The longer the video, the longer the processing time)")
|
|
|
| if st.button("Swap"):
|
| if link:
|
| yt_video = download_video(link)
|
| video = cv2.VideoCapture("video.mp4")
|
| col1, col2 = st.columns([0.5, 0.5])
|
| with col1:
|
| st.markdown('<p style="text-align: center;">Before</p>', unsafe_allow_html=True)
|
| st.video(yt_video)
|
| if uploaded_file is not None:
|
| file_extension = os.path.splitext(uploaded_file.name)[1].lower()
|
| if file_extension in ['.mp4', '.avi', '.mov', '.mkv']:
|
| audio = mp.AudioFileClip(uploaded_file.name)
|
| video = cv2.VideoCapture(uploaded_file.name)
|
| col1, col2 = st.columns([0.5, 0.5])
|
| with col1:
|
| st.markdown('<p style="text-align: center;">Before</p>', unsafe_allow_html=True)
|
| st.video(temp_file.name)
|
| if uploaded_file is not None or link is not None:
|
| with col2:
|
| st.markdown('<p style="text-align: center;">After</p>', unsafe_allow_html=True)
|
| total_frames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
|
| fps = video.get(cv2.CAP_PROP_FPS)
|
| width = int(video.get(cv2.CAP_PROP_FRAME_WIDTH))
|
| height = int(video.get(cv2.CAP_PROP_FRAME_HEIGHT))
|
| fourcc = cv2.VideoWriter_fourcc(*'DIVX')
|
| output_path = os.path.join(working_dir, "target.mp4")
|
| out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
|
| for _ in tqdm(range(total_frames), unit='frame', desc="Progress"):
|
| ret, frame = video.read()
|
| out.write(frame)
|
| if not ret:
|
| break
|
| source_img = os.path.join(working_dir, "source.png")
|
| img.save(source_img)
|
| out.release()
|
| if quality == 'High':
|
| boolq = True
|
| else:
|
| boolq = False
|
| run(boolq)
|
| video.release()
|
| st.download_button(
|
| label="Download Swapped Video",
|
| data=open(output_vid, "rb").read(),
|
| file_name="swapped_video.mp4"
|
| )
|
|
|
|
|
|
|
|
|
