Healthy.ai/main.py

from Healthyai import logger
from Healthyai.pipeline.stage_01_diet_recomd_data_ingestion import DietRecomendationDataIngestionPipeline
from Healthyai.pipeline.stage_02_diet_recomd_data_validation import DietRecomendationDataValidatePipeline
from Healthyai.pipeline.stage_03_diet_recomd_model_trainer import DietRecomendationModelTrainingPipeline
from Healthyai.pipeline.stage_04_food_classifi_data_ingestion import FoodDataIngeestionPipeLine
from Healthyai.pipeline.stage_05_food_classifi_prebasemodel import PrepareBaseModelTrainingPipeline
from Healthyai.pipeline.stage_06_food_classifi_training import ModelTrainingPipeline
from Healthyai.pipeline.stage_07_food_classifi_evaluation import ModelEvaluationPipeline
# Data Ingestion for the Diet Recomendation Pipeline

STAGE_NAME = "Diet Recomendation Data Ingestion Stage"

try:
    logger.info(f">>>>>>>>>> Stage {STAGE_NAME} Started <<<<<<<<<<<")
    obj = DietRecomendationDataIngestionPipeline()
    obj.main()
    logger.info(f">>>>>>>>>> Stage {STAGE_NAME} Completed <<<<<<<<\n\nx============x")
except Exception as e:
    logger.exception(e)
    raise e 

# Data Validation for the Diet Recomendation Pipeline

STAGE_NAME = "Diet Recomendation Data Validation Stage"

try: 
    logger.info(f">>>>>>>>>> Stage {STAGE_NAME} Started <<<<<<<<<<<")
    obj = DietRecomendationDataValidatePipeline()
    obj.main()
    logger.info(f">>>>>>>>>> Stage {STAGE_NAME} Completed <<<<<<<<\n\nx============x")
    
except Exception as e:
    logger.exception(e)
    raise e

# Model Trainer for the Diet Recomendation Pipeline

STAGE_NAME = "Diet Recomendation Model Trainer Stage"


try: 
    logger.info(f">>>>>>>>>> Stage {STAGE_NAME} Started <<<<<<<<<<<")
    obj = DietRecomendationModelTrainingPipeline()
    obj.main()
    logger.info(f">>>>>>>>>> Stage {STAGE_NAME} Completed <<<<<<<<\n\nx============x")
    
except Exception as e:
    logger.exception(e)
    raise e
    
    
# Data Ingestion for  Food Image Classification 

STAGE_NAME = "Food Classification Data Ingestion"

try:      
    logger.info(f">>>>>>>>>> Stage {STAGE_NAME} Started <<<<<<<<<<<")
    obj = FoodDataIngeestionPipeLine()
    obj.main()
    logger.info(f">>>>>>>>>> Stage {STAGE_NAME} Completed <<<<<<<<\n\nx============x")
    
    
except Exception as e:
    raise e

# PrepareBaseModel for  Food Image Classification 

STAGE_NAME = "Food Classification PrepareBase Model"


try:      
    logger.info(f">>>>>>>>>> Stage {STAGE_NAME} Started <<<<<<<<<<<")
    obj = PrepareBaseModelTrainingPipeline()
    obj.main()
    logger.info(f">>>>>>>>>> Stage {STAGE_NAME} Completed <<<<<<<<\n\nx============x")
    
    
except Exception as e:
    raise e



# ModelTraining for  Food Image Classification 

STAGE_NAME = "Food Classification  Model Trainig"

try:      
    logger.info(f">>>>>>>>>> Stage {STAGE_NAME} Started <<<<<<<<<<<")
    obj = ModelTrainingPipeline()
    obj.main()
    logger.info(f">>>>>>>>>> Stage {STAGE_NAME} Completed <<<<<<<<\n\nx============x")
    
    
except Exception as e:
    raise e

STAGE_NAME = "Food Classification Model Evaluation"

try:      
    logger.info(f">>>>>>>>>> Stage {STAGE_NAME} Started <<<<<<<<<<<")
    obj = ModelEvaluationPipeline()
    obj.main()
    logger.info(f">>>>>>>>>> Stage {STAGE_NAME} Completed <<<<<<<<\n\nx============x")
except Exception as e:
    raise e