Update app.py

app.py

@@ -22,6 +22,7 @@ PER_USER_DAILY = int(os.environ.get("PER_USER_DAILY", 3))
 GLOBAL_DAILY = int(os.environ.get("GLOBAL_DAILY", 200))
 ADMIN_USERS = {u.strip().lower() for u in os.environ.get("ADMIN_USERS", "").split(",") if u.strip()}
 FLOW = os.environ.get("FLOW", "oneshot").strip().lower()  # "oneshot" or "twostep"
+SKIP_FIND_FACE = os.environ.get("SKIP_FIND_FACE", "1").strip()  # value sent in one-step mode
 POLL_EVERY = 5      # seconds between status checks
 JOB_TIMEOUT = 900   # give up after 15 minutes
 FONT = "/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf"
@@ -198,7 +199,8 @@ def startup_check():
         else:
             print(f"[startup] WARNING: no route found for {name}; keeping {PATHS[name]}")
     print(f"[startup] using paths: {PATHS}")
-    print(f"[startup] flow: {FLOW} | admins (no limits): {sorted(ADMIN_USERS) or 'none'}")
+    print(f"[startup] flow: {FLOW} | skip_find_face={SKIP_FIND_FACE} (one-step only)"
+          f" | admins (no limits): {sorted(ADMIN_USERS) or 'none'}")
 
 
 def build_drama(mood, excited, angry):
@@ -232,7 +234,7 @@ def submit_job(video, audio, drama, find_face_id=None, face_id=None):
     if find_face_id:
         data.update({"find_face_id": find_face_id, "global_face_id": str(face_id)})
     else:
-        data["skip_find_face"] = "1"
+        data["skip_find_face"] = SKIP_FIND_FACE
     with open(video, "rb") as v, open(audio, "rb") as a:
         files = {"audio": (os.path.basename(audio), a)}
         if not find_face_id:
@@ -453,4 +455,4 @@ with gr.Blocks(title="DRAMA: facial expression editing") as demo:
     )
 
 startup_check()
-demo.queue(max_size=20).launch()
+demo.queue(max_size=20).launch()