diff --git a/foodie_config.py b/foodie_config.py index efa4ab4..e1f7f8d 100644 --- a/foodie_config.py +++ b/foodie_config.py @@ -248,8 +248,7 @@ HOME_KEYWORDS = ["home", "house", "household", "appliance", "kitchen", "gadget"] PRODUCT_KEYWORDS = ["best", "buy", "storage", "organizer", "shop", "price", "container", "product", "deal", "sale", "discount"] CATEGORIES = [ - "People", "Trends", "Travel", - "Lifestyle", "Buzz", "Culture", "Health", "Drink", "Food", "Eats" + "Buzz", "Trends", "Lifestyle", "Culture", "Health", "Drink", "Food", "Eats" ] REDDIT_CLIENT_ID = os.getenv("REDDIT_CLIENT_ID") diff --git a/foodie_utils.py b/foodie_utils.py index 5a2462b..5afe36e 100644 --- a/foodie_utils.py +++ b/foodie_utils.py @@ -663,7 +663,7 @@ def generate_category_from_summary(summary): model=LIGHT_TASK_MODEL, messages=[ {"role": "system", "content": ( - "Based on this summary, select the most relevant category from: Food, Culture, Trends, Health, Lifestyle, Drink, Eats. " + "Based on this summary, select the most relevant category from: Buzz, Trends, Lifestyle, Culture, Health, Drink, Food, Eats. " "Return only the category name." )}, {"role": "user", "content": summary} @@ -672,7 +672,7 @@ def generate_category_from_summary(summary): ) category = response.choices[0].message.content.strip() logging.info(f"Generated category: {category}") - return category if category in ["Food", "Culture", "Trends", "Health", "Lifestyle", "Drink", "Eats"] else "Trends" + return category if category in ["Buzz", "Trends", "Lifestyle", "Culture", "Health", "Drink", "Food", "Eats"] else "Trends" except Exception as e: logging.error(f"Category generation failed: {e}") return "Trends" @@ -1637,7 +1637,7 @@ def prepare_post_data(summary, title, main_topic=None): # Select a full author dictionary from AUTHORS (already imported from foodie_config) author = random.choice(AUTHORS) - categories = ["Food", "Trends", "Eats", "Culture"] + categories = ["Buzz", "Trends", "Lifestyle", "Culture", "Health", "Drink", "Food", "Eats"] category = random.choice(categories) post_data = {