🎉 | Project released

main.py

@@ -0,0 +1,90 @@
+#!/usr/bin/env python3
+"""FOBG — AI Background Remover — Entry Point.
+
+Usage:
+    python main.py                          # Launch Gradio UI (default)
+    python main.py --host 0.0.0.0 --port 7860
+    python main.py --share                  # Create public Gradio share link
+    python main.py --cli --help             # CLI mode (scripted use)
+"""
+
+import argparse
+import sys
+import os
+
+# Ensure the project root is on sys.path
+_PROJECT_ROOT = os.path.dirname(os.path.abspath(__file__))
+sys.path.insert(0, _PROJECT_ROOT)
+
+from src.ui.app import launch_app
+
+
+def cli_demo():
+    """Simple CLI demo for batch image processing."""
+    import argparse as ap2
+    parser = ap2.ArgumentParser(description="FOBG CLI — process images/videos")
+    parser.add_argument("input", help="Input image or video file path")
+    parser.add_argument("-o", "--output", help="Output file path (auto-generated if omitted)")
+    parser.add_argument("--model", default="u2net", choices=["u2net", "u2netp", "u2net_human_seg", "isnet-general-use", "isnet-anime"], help="rembg model")
+    parser.add_argument("--alpha-matting", action="store_true", default=True, help="Use alpha matting")
+    parser.add_argument("--no-alpha-matting", action="store_true", help="Disable alpha matting")
+    parser.add_argument("--video", action="store_true", help="Treat input as video (frame-by-frame processing)")
+    args = parser.parse_args()
+
+    input_path = args.input
+    if not os.path.exists(input_path):
+        print(f"Error: File not found: {input_path}")
+        sys.exit(1)
+
+    is_video = args.video or input_path.lower().endswith((".mp4", ".avi", ".webm", ".mov", ".mkv"))
+    output = args.output or f"output_{os.path.splitext(os.path.basename(input_path))[0]}_{os.urandom(4).hex()}"
+    if is_video:
+        output += ".mp4"
+    else:
+        output += ".png"
+
+    from src.core.bg_remove import remove_background_image, remove_background_bytes
+    from src.core.video_bg_remove import process_video_frame_by_frame
+
+    if is_video:
+        print(f"Processing video: {input_path}")
+        result = process_video_frame_by_frame(
+            input_path, output,
+            model=args.model,
+            alpha_matting=not args.no_alpha_matting,
+        )
+        print(f"Video saved to: {result}")
+    else:
+        print(f"Processing image: {input_path}")
+        from PIL import Image
+        input_img = Image.open(input_path).convert("RGB")
+        result = remove_background_image(input_img, model=args.model, force=True, alpha_matting=not args.no_alpha_matting)
+        result.save(output, format="PNG")
+        print(f"Image saved to: {output}")
+
+
+def main():
+    parser = argparse.ArgumentParser(description="FOBG — AI Background Remover")
+    parser.add_argument("--cli", action="store_true", help="Run in CLI mode for batch processing")
+    parser.add_argument("--host", default="0.0.0.0", help="Gradio server host")
+    parser.add_argument("--port", type=int, default=7860, help="Gradio server port")
+    parser.add_argument("--share", action="store_true", help="Create a public share link")
+    parser.add_argument("--no-browser", action="store_true", help="Do not open browser")
+
+    args = parser.parse_args()
+
+    if args.cli:
+        # Parse remaining args for CLI mode
+        sys.argv = ["fobg"] + [arg for arg in sys.argv[1:] if arg not in ("--cli",)]
+        cli_demo()
+    else:
+        launch_app(
+            host=args.host,
+            port=args.port,
+            share=args.share,
+            inbrowser=not args.no_browser,
+        )
+
+
+if __name__ == "__main__":
+    main()