|
|
|
|
|
""" |
|
|
Unified launcher for YouTube MCP application. |
|
|
Provides options to run API server, Gradio UI, or both. |
|
|
""" |
|
|
|
|
|
import argparse |
|
|
import asyncio |
|
|
import uvicorn |
|
|
import threading |
|
|
import time |
|
|
from dotenv import load_dotenv |
|
|
|
|
|
|
|
|
load_dotenv() |
|
|
|
|
|
def start_api_server(host="127.0.0.1", port=8080): |
|
|
"""Start FastAPI server.""" |
|
|
from api_server import app |
|
|
print(f"Starting API server on http://{host}:{port}") |
|
|
uvicorn.run(app, host=host, port=port) |
|
|
|
|
|
def start_gradio_ui(host="127.0.0.1", port=8081): |
|
|
"""Start Gradio UI.""" |
|
|
import gradio_app |
|
|
print(f"Starting Gradio UI on http://{host}:{port}") |
|
|
gradio_app.demo.launch(server_name=host, server_port=port, share=False) |
|
|
|
|
|
def start_both(api_host="127.0.0.1", api_port=8080, ui_host="127.0.0.1", ui_port=8081): |
|
|
"""Start both API server and Gradio UI.""" |
|
|
print(f"Starting API server on http://{api_host}:{api_port}") |
|
|
print(f"Starting Gradio UI on http://{ui_host}:{ui_port}") |
|
|
|
|
|
|
|
|
api_thread = threading.Thread( |
|
|
target=start_api_server, |
|
|
args=(api_host, api_port), |
|
|
daemon=True |
|
|
) |
|
|
api_thread.start() |
|
|
|
|
|
|
|
|
time.sleep(2) |
|
|
|
|
|
|
|
|
start_gradio_ui(ui_host, ui_port) |
|
|
|
|
|
def main(): |
|
|
parser = argparse.ArgumentParser(description="YouTube MCP Application Launcher") |
|
|
parser.add_argument( |
|
|
"--mode", |
|
|
choices=["api", "ui", "both"], |
|
|
default="both", |
|
|
help="Launch mode: 'api' for FastAPI server only, 'ui' for Gradio UI only, 'both' for both services" |
|
|
) |
|
|
parser.add_argument( |
|
|
"--host", |
|
|
default="127.0.0.1", |
|
|
help="Host address (default: 127.0.0.1)" |
|
|
) |
|
|
parser.add_argument( |
|
|
"--port", |
|
|
type=int, |
|
|
default=8080, |
|
|
help="Port number (default: 8080 for API, 8081 for UI in 'both' mode)" |
|
|
) |
|
|
|
|
|
args = parser.parse_args() |
|
|
|
|
|
try: |
|
|
if args.mode == "api": |
|
|
start_api_server(args.host, args.port) |
|
|
elif args.mode == "ui": |
|
|
start_gradio_ui(args.host, args.port) |
|
|
elif args.mode == "both": |
|
|
start_both(args.host, args.port, args.host, args.port + 1) |
|
|
except KeyboardInterrupt: |
|
|
print("\nKeyboard interruption in main thread... closing server.") |
|
|
except Exception as e: |
|
|
print(f"Error starting application: {e}") |
|
|
|
|
|
if __name__ == "__main__": |
|
|
main() |
|
|
|
