import argparse
import os
from open_webui import app


def main():
    # Set up Open WebUI environment variables if they're provided
    # These will be set by the Tauri app for proper data persistence
    data_dir = os.environ.get("DATA_DIR")
    if data_dir:
        print(f"Using data directory: {data_dir}")
        # Ensure the data directory exists
        os.makedirs(data_dir, exist_ok=True)

        # Set additional Open WebUI environment variables for persistence
        os.environ.setdefault("WEBUI_DATA_DIR", data_dir)
        os.environ.setdefault("WEBUI_UPLOAD_DIR", os.path.join(data_dir, "uploads"))
        os.environ.setdefault("WEBUI_CACHE_DIR", os.path.join(data_dir, "cache"))

        # Create subdirectories
        os.makedirs(os.environ["WEBUI_UPLOAD_DIR"], exist_ok=True)
        os.makedirs(os.environ["WEBUI_CACHE_DIR"], exist_ok=True)

    # Run the FastAPI app - this will handle the command line args including "serve"
    app()


if __name__ == "__main__":
    main()