change image generation to base64

app.ipynb

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 19,
+   "execution_count": 21,
    "id": "6b57ced9-62ee-44a0-a895-6ed288f970ff",
    "metadata": {},
    "outputs": [
@@ -10,7 +10,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Running on local URL:  http://127.0.0.1:7878\n",
+      "Running on local URL:  http://127.0.0.1:7880\n",
       "\n",
       "To create a public link, set `share=True` in `launch()`.\n"
      ]
@@ -18,7 +18,7 @@
     {
      "data": {
       "text/html": [
-       "<div><iframe src=\"http://127.0.0.1:7878/\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
+       "<div><iframe src=\"http://127.0.0.1:7880/\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
       ],
       "text/plain": [
        "<IPython.core.display.HTML object>"
@@ -26,12 +26,35 @@
      },
      "metadata": {},
      "output_type": "display_data"
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "desc='A 18 year old person. They have the following mental health symptoms: inability to concentrate.'\n",
+      "\n",
+      "\n",
+      "You are writing motivational text messages to help people with their mental health. Messages should be friendly and positive, but also professional and super short. You are limited on space. Messages should be written at the reading level of an eighth grader. Word choice should be short and simple so everyone can understand. \n",
+      "\n",
+      "\n",
+      "You will be given some basic information about the person you are addressing. DO NOT reference all of their likes if there are more than two. Be discerning. You should try to use the person's information to give them relevant and actionable tips for improving their mental health symptoms.\n",
+      "\n",
+      "\n",
+      "Write a short inspirational message for the person with the following description:\n",
+      "\n",
+      "A 18 year old person. They have the following mental health symptoms: inability to concentrate.\n",
+      "\n",
+      "Illustrate one simple, inspirational, fun image to help a person with their mental health. Do not include text. The style is cute and illustrative. The person is a 18 year old person. The image should align with the following message:\n",
+      "\n",
+      "Hey there! Remember, you got this! To boost concentration, try breaking tasks into smaller steps. Take regular breaks and stay hydrated. Stay positive and believe in your abilities. You're capable of great things! 💪🌟\n"
+     ]
     }
    ],
    "source": [
+    "import io\n",
+    "import base64\n",
     "import numpy as np\n",
     "import gradio as gr\n",
-    "import requests\n",
     "from PIL import Image\n",
     "from openai import OpenAI\n",
     "\n",
@@ -151,9 +174,8 @@
     "           + f'The image should align with the following message:\\n\\n{nudge_message}'\n",
     "\n",
     "    print(f'\\n{prompt}')\n",
-    "    response = client.images.generate(prompt=prompt, model='dall-e-3')#, response_format='b64_json')\n",
-    "    print(response)\n",
-    "    nudge_image = Image.open(requests.get(response.data[0].url, stream=True).raw)\n",
+    "    response = client.images.generate(prompt=prompt, model='dall-e-3', response_format='b64_json')\n",
+    "    nudge_image = Image.open(io.BytesIO(base64.b64decode(response.data[0].b64_json)))\n",
     "\n",
     "    return nudge_image\n",
     "\n",