diff --git a/main.py b/main.py index 56dec2f..cd67e6e 100644 --- a/main.py +++ b/main.py @@ -1,5 +1,6 @@ import json import logging +from bashful_gcp_functions.models.domain_config import DomainConfig import functions_framework import requests @@ -252,4 +253,19 @@ def get_domain_configs(request): except Exception as e: logger.error(e) return ({"error": str(e)}, 500, response_headers) + +def generate_brand_domain(request): + try: + brand_config_id = request.args.get("brandID") + domain_config_id = request.args.get("domainID") + brand_config = airtable_service.get_brand_config(brand_config_id) + if brand_config: + brand_config = BrandConfig(brand_config["id"], brand_config) + domain_config = airtable_service.get_domain_config(domain_config_id) + if domain_config: + domain_config = DomainConfig(brand_config["id"], brand_config) + return (ai_service.generate_brand_image(brand_config, domain_config), 200, response_headers) + except Exception as e: + logger.error(e) + return ({"error": str(e)}, 500, response_headers)