#ifndef BIERGARTEN_PIPELINE_DATA_GENERATION_DATA_GENERATOR_H_
#define BIERGARTEN_PIPELINE_DATA_GENERATION_DATA_GENERATOR_H_

#include <string>

struct BreweryResult {
   std::string name;
   std::string description;
};

struct UserResult {
   std::string username;
   std::string bio;
};

class DataGenerator {
  public:
   virtual ~DataGenerator() = default;

   virtual void Load(const std::string& model_path) = 0;

   virtual BreweryResult GenerateBrewery(const std::string& city_name,
                                         const std::string& country_name,
                                         const std::string& region_context) = 0;

   virtual UserResult GenerateUser(const std::string& locale) = 0;
};

#endif  // BIERGARTEN_PIPELINE_DATA_GENERATION_DATA_GENERATOR_H_