From 0fdf46a885e087d88e1649ab9507ea2b647fcc62 Mon Sep 17 00:00:00 2001
From: Kasisnu <5044120+kasisnu@users.noreply.github.com>
Date: Thu, 5 Sep 2024 19:19:13 +0530
Subject: [PATCH] facefusion: minor updates to runner

---
 workflows/comfy/facefusion-runner.sh | 30 ++++++++++++++++++++++------
 1 file changed, 24 insertions(+), 6 deletions(-)

diff --git a/workflows/comfy/facefusion-runner.sh b/workflows/comfy/facefusion-runner.sh
index 0d98681..a5fb829 100755
--- a/workflows/comfy/facefusion-runner.sh
+++ b/workflows/comfy/facefusion-runner.sh
@@ -3,8 +3,8 @@ set -ex
 
 # we're invoked as 
 # ./facefusion-runner.sh --input-audio $input_audio --input-video --output $output
-LONGOPTS=input_audio:,input_video:,output:
-OPTIONS=a:v:o:
+LONGOPTS=input_audio:,input_video:,output:,tmpdir:
+OPTIONS=a:v:o:t
 
 PARSED=$(getopt --options=$OPTIONS --longoptions=$LONGOPTS --name "$0" -- "$@") || exit 2
 # read getopt’s output this way to handle the quoting right:
@@ -25,6 +25,10 @@ while true; do
       output="$2"
       shift 2
       ;;
+    -t|--tmpdir)
+      tmpdir="$2"
+      shift 2
+      ;;
     --)
       shift
       break
@@ -36,13 +40,29 @@ while true; do
   esac
 done
 
+if [ -z "$tmpdir" ]; then
+  tmpdir=$(mktemp -d)
+fi
+TEMPDIR=$tmpdir
+
+
+
+
+# rename /tmp/downloads_long_lived/temp_face_fusion_2.niZOOlINrFkE/audio.bin to /tmp/downloads_long_lived/temp_face_fusion_2.niZOOlINrFkE/audio.wav
+cp $input_audio $TEMPDIR/audio.wav
+# rename /tmp/downloads_long_lived/temp_face_fusion_2.niZOOlINrFkE/image_or_video.bin to /tmp/downloads_long_lived/temp_face_fusion_2.niZOOlINrFkE/image_or_video.mp4
+cp $input_video $TEMPDIR/image_or_video.mp4
+
 echo "input_audio: $input_audio"
 echo "input_video: $input_video"
 echo "output: $output"
+echo "tmpdir: $tmpdir"
 
 export input_audio
 export input_video
 export output
+export TEMPDIR
+
 
 deactivate || true
 
@@ -50,10 +70,8 @@ cd /facefusion
 # Set up the environment
 source venv/bin/activate
 
-TEMPDIR=$(mktemp -d)
 
-
-venv/bin/python3 run.py -s $input_audio -t $input_video --frame-processors lip_syncer --execution-provider cuda -o $TEMPDIR/1.mp4 --headless
-venv/bin/python3 run.py -s $input_audio -t $TEMPDIR/1.mp4 --frame-processors face_enhancer --execution-provider cuda -o $output --headless
+venv/bin/python3 run.py -s $TEMPDIR/audio.wav -t $TEMPDIR/image_or_video.mp4 --frame-processors lip_syncer --execution-provider cuda -o $TEMPDIR/1.mp4 --headless
+venv/bin/python3 run.py -s $TEMPDIR/audio.wav -t $TEMPDIR/1.mp4 --frame-processors face_enhancer --execution-provider cuda -o $output --headless
 
 # rm -rf $TEMPDIR
\ No newline at end of file