import argparse
import uvicorn
import os

def main():
    parser = argparse.ArgumentParser(description="Executor do Aduc-Sdr")
    parser.add_argument("mode", choices=["gradio", "api"], help="Modo de execução: 'gradio' para a UI, 'api' para o servidor FastAPI.")
    args = parser.parse_args()

    if args.mode == "gradio":
        print("Iniciando a interface Gradio...")
        # Importa e executa a lógica de lançamento que está no final de app_gradio.py
        from app_gradio import demo
        demo.queue().launch()
    elif args.mode == "api":
        print("Iniciando o servidor FastAPI em http://127.0.0.1:8000")
        uvicorn.run("app_api:app", host="127.0.0.1", port=8000, reload=True)

if __name__ == "__main__":
    main()