{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "4b48236b-3e6f-4dec-ac45-5bb8c5f7ff45",
   "metadata": {},
   "source": [
    "<div style=\"font-size: 50px; text-align: center\" >Deadpool™️ Detector</div>\n",
    "\n"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "731ffedc-3131-4d20-bed6-ecfcf0681d73",
   "metadata": {},
   "source": [
    "# Description\n",
    "First development app using deep learning techniques. Stepping stone in my journey.\n",
    "\n",
    "Given an image infer if it contains Deadpool in it."
   ]
  },
  {
   "cell_type": "markdown",
   "id": "1afbc8dd-8b83-4603-8ab8-3724fca03fa3",
   "metadata": {},
   "source": [
    "# Train a model"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "3e2cfb0c-6ac4-4cd3-b66e-1eed809bb88d",
   "metadata": {},
   "source": [
    "## Training Data\n",
    "Using [SH2022 Dataset](https://www.kaggle.com/datasets/muhammadhananasghar/sh2022-dataset) there is a labeled set of Deadpool images."
   ]
  },
  {
   "cell_type": "markdown",
   "id": "2549da81-e133-4f5d-93e2-e754885d9d99",
   "metadata": {},
   "source": [
    "Add data to a `DataLoaders` object and display a batch"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d9fa3c5f-c90f-44bb-bf95-cfbc9698c97e",
   "metadata": {},
   "outputs": [],
   "source": [
    "from fastai.vision.all import *\n",
    "\n",
    "data_path = Path('/home/zeus/data/SH2022')\n",
    "\n",
    "dls = ImageDataLoaders.from_folder(\n",
    "    path=data_path,\n",
    "    train='train',\n",
    "    valid='test/Deadpool',\n",
    "    item_tfms=Resize(224),\n",
    "    batch_tfms=aug_transforms()\n",
    ")\n",
    "dls.show_batch()"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "f4b6eb81-25b6-449c-8f48-f418c6e1cb7a",
   "metadata": {},
   "source": [
    "## Training"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "54353c83-2bf3-419d-a7f3-24eb6feef718",
   "metadata": {},
   "outputs": [],
   "source": [
    "learn = vision_learner(dls, resnet34, metrics=error_rate)\n",
    "learn.fine_tune(3)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "1f5afdb5-998b-4b12-b56f-0c89077ae5c2",
   "metadata": {},
   "source": [
    "## Export Trained Model"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "22ff7ee5-efc0-4c71-88e6-c1a43b94edba",
   "metadata": {},
   "outputs": [],
   "source": [
    "learn.export('/home/zeus/projects/Deadpool-Detector/deadpool-detection-model.pkl')"
   ]
  },
  {
   "attachments": {},
   "cell_type": "markdown",
   "id": "792d8a70",
   "metadata": {},
   "source": [
    "## Share Learner with the world!"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "27376bb5",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "14796a3e",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "491daa3086dd4a109f036d62bdecf5fa",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "Upload 1 LFS files:   0%|          | 0/1 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "46bae672442548c18d955a6d8c7ed5ad",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "model.pkl:   0%|          | 0.00/88.7M [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "text/plain": [
       "'https://huggingface.co/jeffreymjohnson/Deadpool-Detector/tree/main/'"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "from huggingface_hub import push_to_hub_fastai\n",
    "\n",
    "learn = load_learner('/home/zeus/projects/Deadpool-Detector/deadpool-detection-model.pkl')\n",
    "repo_id = \"jeffreymjohnson/Deadpool-Detector\"\n",
    "push_to_hub_fastai(learner=learn, repo_id=repo_id)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "61b3019d-cd37-49b4-bced-503499796d6c",
   "metadata": {},
   "source": [
    "# Create App"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "f678e4ce-6b99-45b2-ae12-3e3a6978fe0a",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "a5c9312bd4be458e8694df43a510adcc",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "Fetching 4 files:   0%|          | 0/4 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "86445259808342c7895ef51c3c30a27b",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "Downloading (…)3b40e/.gitattributes:   0%|          | 0.00/1.48k [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "0f5c6206271041c2b3d978afff8af373",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "Downloading (…)4d02d3b40e/README.md:   0%|          | 0.00/768 [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "576a45913eff4216bac8db9e6a3bd8f5",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "Downloading (…)3b40e/pyproject.toml:   0%|          | 0.00/162 [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "c847d69e1eab41b6bc51f1e65124ed5f",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "Downloading model.pkl:   0%|          | 0.00/88.7M [00:00<?, ?B/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "#|export\n",
    "from huggingface_hub import from_pretrained_fastai\n",
    "\n",
    "learner = from_pretrained_fastai(\"jeffreymjohnson/Deadpool-Detector\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "cc0fa8a2-51fe-4591-9056-e84e0fec9af7",
   "metadata": {},
   "outputs": [],
   "source": [
    "#|export\n",
    "labels = learn.dls.vocab\n",
    "def predict(img):\n",
    "    yes_reaction = \"reactions/yes.jfif\"\n",
    "    no_reaction = \"reactions/no.jfif\"\n",
    "    what_reaction = \"reactions/what.jpg\"\n",
    "    \n",
    "    img = PILImage.create(img)\n",
    "    pred,pred_idx,probs = learn.predict(img)\n",
    "    index = pred_idx.item()\n",
    "    probability = probs[index].item()\n",
    "    isDeadpool = pred == \"Deadpool\"\n",
    "    \n",
    "    if isDeadpool:\n",
    "        x = \"is\"\n",
    "    else:\n",
    "        x = \"is not\"\n",
    "    \n",
    "    #print(\"The probability is: [\" + str(probability) + \"] that this \" + x + \" Deadpool\")\n",
    "    \n",
    "    if probability < .75:\n",
    "        return PILImage.create(what_reaction)\n",
    "    elif isDeadpool:\n",
    "        return PILImage.create(yes_reaction)\n",
    "    else:\n",
    "        return PILImage.create(no_reaction)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "3cec4455-8159-457b-9563-7e41aa202dc1",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Running on local URL:  http://127.0.0.1:7860\n",
      "Running on public URL: https://ed025bd9d3a5e5798f.gradio.live\n",
      "\n",
      "This share link expires in 72 hours. For free permanent hosting and GPU upgrades (NEW!), check out Spaces: https://huggingface.co/spaces\n"
     ]
    },
    {
     "data": {
      "text/html": [
       "<div><iframe src=\"https://ed025bd9d3a5e5798f.gradio.live\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
      ],
      "text/plain": [
       "<IPython.core.display.HTML object>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "The probability is: [0.9999980926513672] that this is Deadpool\n"
     ]
    },
    {
     "data": {
      "text/html": [
       "\n",
       "<style>\n",
       "    /* Turns off some styling */\n",
       "    progress {\n",
       "        /* gets rid of default border in Firefox and Opera. */\n",
       "        border: none;\n",
       "        /* Needs to be in here for Safari polyfill so background images work as expected. */\n",
       "        background-size: auto;\n",
       "    }\n",
       "    progress:not([value]), progress:not([value])::-webkit-progress-bar {\n",
       "        background: repeating-linear-gradient(45deg, #7e7e7e, #7e7e7e 10px, #5c5c5c 10px, #5c5c5c 20px);\n",
       "    }\n",
       "    .progress-bar-interrupted, .progress-bar-interrupted::-webkit-progress-bar {\n",
       "        background: #F44336;\n",
       "    }\n",
       "</style>\n"
      ],
      "text/plain": [
       "<IPython.core.display.HTML object>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "The probability is: [0.9999998807907104] that this is not Deadpool\n"
     ]
    }
   ],
   "source": [
    "#|export\n",
    "import gradio as gr\n",
    "\n",
    "demo = gr.Interface(\n",
    "    title=\"Deadpool™️ Detector\",\n",
    "    fn=predict,\n",
    "    inputs=gr.Image(shape=(200,200)),\n",
    "    outputs=gr.Image(shape=(200,200)), \n",
    "    examples=[\"examples/deadpool_example1.jpg\", \"examples/deadpool_example2.jpg\", \"examples/deadpool_example3.jpg\", \"examples/spiderman.jpg\",]\n",
    ").launch(share=true)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "0c703c3c-f9a9-4c44-a1d5-44dd1d6189ae",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.9.16"
  },
  "widgets": {
   "application/vnd.jupyter.widget-state+json": {
    "state": {},
    "version_major": 2,
    "version_minor": 0
   }
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}