Added job count, fix video quality, tweak logging

app.py

@@ -12,6 +12,15 @@ app = FastAPI()
 # This acts as a queue: only 1 task can hold the "permit" at a time.
 cpu_semaphore = asyncio.Semaphore(1)
 
+def get_queue_status():
+    # If value is 1, nobody is using it. If 0, it's busy.
+    is_busy = cpu_semaphore.locked()
+    # Number of tasks currently waiting for the lock
+    waiting_count = len(cpu_semaphore._waiters) if cpu_semaphore._waiters else 0
+    if not is_busy:
+        return "Idle"
+    return f"Processing (plus {waiting_count} in queue)"
+
 # Configuration
 SOURCE_FACE = "source.jpg" 
 OUTPUT_DIR = Path("outputs")
@@ -26,7 +35,7 @@ def run_facefusion(target_path: Path, output_path: Path):
         "--target-path", str(target_path),
         "--output-path", str(output_path)
     ]
-    print(f"Executing: {' '.join(cmd)}")
+    print(f"###### Executing: {' '.join(cmd)}")
     try:
         # We use subprocess.run here because it's called inside to_thread
         subprocess.run(cmd, check=True)
@@ -34,13 +43,13 @@ def run_facefusion(target_path: Path, output_path: Path):
         # if target_path.exists():
         #     target_path.unlink()
     except Exception as e:
-        print(f"Error during processing: {e}")
+        print(f"###### Error during processing: {e}")
 
 async def background_worker(url: str, target_path: Path, output_path: Path, headers: dict):
     """Handles queuing, downloading, and processing"""
 
     async with cpu_semaphore:
-        print(f"Starting processing for {url}")
+        print(f"###### Starting processing for {url}")
 
         try:
             response = requests.get(url, headers=headers, timeout=30, stream=True)
@@ -49,12 +58,13 @@ async def background_worker(url: str, target_path: Path, output_path: Path, head
                 for chunk in response.iter_content(chunk_size=8192):
                     f.write(chunk)
         except Exception as e:
-            print(f"Download failed in background: {e}")
+            print(f"###### Download failed in background: {e}")
             return
         await asyncio.to_thread(run_facefusion, target_path, output_path)
 
 @app.get("/", response_class=HTMLResponse)
 async def index():
+    status = get_queue_status()
     files = []
     for ext in EXTENSIONS:
         files.extend(OUTPUT_DIR.glob(ext))
@@ -74,6 +84,7 @@ async def index():
                 <input type="text" name="url" placeholder="Paste Image/Video URL..." style="width:300px;" required>
                 <button type="submit">Start Swapping</button>
             </form>
+            <p><strong>System Status:</strong> {status}</p>
             <hr>
             <h2>Processed Files</h2>
             <p><button onclick="location.reload()">Refresh List</button></p>
@@ -100,6 +111,8 @@ async def process_swap(background_tasks: BackgroundTasks, url: str = Form(...)):
         "Connection": "keep-alive",
     }
 
+    print(f"###### Queueing job {url}")
+
     background_tasks.add_task(background_worker, url, target_path, output_path, headers)
 
     return HTMLResponse(content=f"""
@@ -112,7 +125,7 @@ async def process_swap(background_tasks: BackgroundTasks, url: str = Form(...)):
 @app.get("/download/{filename}")
 async def download_file(filename: str):
     file_path = OUTPUT_DIR / filename
-    print(f"Searching for file at: {file_path.absolute()}")
+    print(f"###### Searching for file at: {file_path.absolute()}")
     if file_path.exists():
         return FileResponse(path=file_path)
     raise HTTPException(status_code=404, detail="File not found.")

patches/facefusion.ini

@@ -58,7 +58,7 @@ output_audio_quality =
 output_audio_volume =
 output_video_encoder =
 output_video_preset =
-output_video_quality = 100
+output_video_quality = 80
 output_video_scale =
 output_video_fps =