from flask import Flask, request
import mlflow
import os


# For local testing, uses sandbox, for production, uses fansifter
DATABASE_NAME = os.getenv('DATABASE_NAME')

mlflow_client = mlflow.tracking.MlflowClient(tracking_uri='https://mlflow.fansifter.cloud')
mlflow.tracking.set_tracking_uri('https://mlflow.fansifter.cloud')

# TODO! add proper logging
app = Flask(__name__)

import logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger('ml_engine')

# Modules using app need to be imported
from . import app_controller
from . import tasks_controller
from .tasks import *