2_rec_system.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "import random\n",
    "import os\n",
    "from PIL import Image"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "def create_outfit(wardrobe_path: str) -> dict:\n",
    "    \"\"\"\n",
    "    Create a randomized outfit from the wardrobe, ensuring it includes one apparel\n",
    "    from both the 'topwear' and 'bottomwear' categories, and randomly selecting\n",
    "    from other categories.\n",
    "\n",
    "    Parameters:\n",
    "    - wardrobe_path: str\n",
    "        Path pointing towards the categorized image folder, where each\n",
    "        folder inside represents a clothing category (e.g., 'Shirts', 'Pants').\n",
    "\n",
    "    Returns:\n",
    "    - dict\n",
    "        A dictionary containing one randomly selected apparel from each category,\n",
    "        ensuring at least 'topwear' and 'bottomwear' are present in the outfit.\n",
    "    \"\"\"\n",
    "    \n",
    "    output = {}\n",
    "\n",
    "    # Required categories for any outfit\n",
    "    required_categories = ['Topwear', 'Bottomwear']\n",
    "    \n",
    "    # Get the list of all categories (folders in the wardrobe path)\n",
    "    categories = [category for category in os.listdir(wardrobe_path) if os.path.isdir(f\"{wardrobe_path}/{category}\")]\n",
    "    \n",
    "    # Ensure the required categories exist in the wardrobe\n",
    "    for required in required_categories:\n",
    "        if required not in categories:\n",
    "            raise ValueError(f\"'{required}' category not found in the wardrobe. Outfit cannot be created.\")\n",
    "\n",
    "    # Select one item from both 'topwear' and 'bottomwear' categories\n",
    "    for category in required_categories:\n",
    "        items = os.listdir(f\"{wardrobe_path}/{category}\")\n",
    "        if items:\n",
    "            output[category] = random.choice(items)\n",
    "        else:\n",
    "            raise ValueError(f\"No items found in the '{category}' category.\")\n",
    "\n",
    "    # Randomly select from other categories (optional) while ensuring no duplicates\n",
    "    remaining_categories = [cat for cat in categories if cat not in required_categories]\n",
    "    \n",
    "    # Shuffle remaining categories for randomness\n",
    "    random.shuffle(remaining_categories)\n",
    "    \n",
    "    # Optionally pick additional items from other categories (one item per category)\n",
    "    for category in remaining_categories:\n",
    "        items = os.listdir(f\"{wardrobe_path}/{category}\")\n",
    "        if items:\n",
    "            output[category] = random.choice(items)\n",
    "    \n",
    "    return output"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "def show_outfit(outfit: dict, wardrobe_path: str) -> None:\n",
    "    \"\"\"\n",
    "    Display the outfit generated by the create_outfit function by opening the images\n",
    "    of the selected apparels.\n",
    "\n",
    "    Parameters:\n",
    "    - outfit: dict\n",
    "        A dictionary containing categories as keys and the corresponding selected apparel\n",
    "        as values (output from create_outfit function).\n",
    "    - wardrobe_path: str\n",
    "        Path pointing towards the categorized image folder.\n",
    "    \n",
    "    Returns:\n",
    "    - None\n",
    "    \"\"\"\n",
    "    \n",
    "    for category, apparel in outfit.items():\n",
    "        apparel_path = os.path.join(wardrobe_path, category, apparel)\n",
    "        \n",
    "        if os.path.exists(apparel_path):\n",
    "            print(f\"Displaying {apparel} from {category}\")\n",
    "            image = Image.open(apparel_path)\n",
    "            image.show()\n",
    "        else:\n",
    "            print(f\"Image for {apparel} in {category} not found at {apparel_path}.\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "ename": "ValueError",
     "evalue": "'topwear' category not found in the wardrobe. Outfit cannot be created.",
     "output_type": "error",
     "traceback": [
      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[0;31mValueError\u001b[0m                                Traceback (most recent call last)",
      "Cell \u001b[0;32mIn[5], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m outfit \u001b[38;5;241m=\u001b[39m \u001b[43mcreate_outfit\u001b[49m\u001b[43m(\u001b[49m\u001b[43mwardrobe_path\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[38;5;124;43mcategorized_images\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[43m)\u001b[49m\n\u001b[1;32m      2\u001b[0m show\n",
      "Cell \u001b[0;32mIn[2], line 32\u001b[0m, in \u001b[0;36mcreate_outfit\u001b[0;34m(wardrobe_path)\u001b[0m\n\u001b[1;32m     30\u001b[0m \u001b[38;5;28;01mfor\u001b[39;00m required \u001b[38;5;129;01min\u001b[39;00m required_categories:\n\u001b[1;32m     31\u001b[0m     \u001b[38;5;28;01mif\u001b[39;00m required \u001b[38;5;129;01mnot\u001b[39;00m \u001b[38;5;129;01min\u001b[39;00m categories:\n\u001b[0;32m---> 32\u001b[0m         \u001b[38;5;28;01mraise\u001b[39;00m \u001b[38;5;167;01mValueError\u001b[39;00m(\u001b[38;5;124mf\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124m'\u001b[39m\u001b[38;5;132;01m{\u001b[39;00mrequired\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m'\u001b[39m\u001b[38;5;124m category not found in the wardrobe. Outfit cannot be created.\u001b[39m\u001b[38;5;124m\"\u001b[39m)\n\u001b[1;32m     34\u001b[0m \u001b[38;5;66;03m# Select one item from both 'topwear' and 'bottomwear' categories\u001b[39;00m\n\u001b[1;32m     35\u001b[0m \u001b[38;5;28;01mfor\u001b[39;00m category \u001b[38;5;129;01min\u001b[39;00m required_categories:\n",
      "\u001b[0;31mValueError\u001b[0m: 'topwear' category not found in the wardrobe. Outfit cannot be created."
     ]
    }
   ],
   "source": [
    "WARDROBE_PATH=\"categorized_images\"\n",
    "outfit = create_outfit(wardrobe_path=\"\")\n",
    "show_outfit(outfit=outfit, wardrobe_path=\"\")"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": ".venv",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.12.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}