feat: ai logo

.gitignore

@@ -1 +1,2 @@
-venv
+venv
+.env

app.ipynb

@@ -0,0 +1,152 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 26,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/Users/ny0/workspace/ailogo/venv/lib/python3.11/site-packages/tqdm/auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html\n",
+      "  from .autonotebook import tqdm as notebook_tqdm\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Loaded as API: https://passant-labs-ailogo.hf.space/ ✔\n",
+      "Hello Howdy!!!\n"
+     ]
+    }
+   ],
+   "source": [
+    "from gradio_client import Client\n",
+    "\n",
+    "client = Client(\"https://passant-labs-ailogo.hf.space/\")\n",
+    "result = client.predict(\n",
+    "\t\t\t\t\"Howdy!\",\n",
+    "\t\t\t\tapi_name=\"/predict\"\n",
+    ")\n",
+    "print(result)\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 9,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Running on local URL:  http://127.0.0.1:7867\n",
+      "\n",
+      "To create a public link, set `share=True` in `launch()`.\n"
+     ]
+    },
+    {
+     "data": {
+      "text/html": [
+       "<div><iframe src=\"http://127.0.0.1:7867/\" 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"
+    },
+    {
+     "data": {
+      "text/plain": []
+     },
+     "execution_count": 9,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "import os\n",
+    "import gradio as gr\n",
+    "import requests\n",
+    "%env OPENAI_KEY\n",
+    "openai_key = os.getenv(\"OPENAI_KEY\")\n",
+    "def prompt_builder(logo_name, slogan, industry, color_scheme, font_style):\n",
+    "    n = 1\n",
+    "    prompt = f\"\"\"Please generate a logo which name is {logo_name} {(\", slogan is \"+slogan) if slogan else \"\"}, industry is {industry}, color scheme is {color_scheme}, font style is {font_style}\"\"\"\n",
+    "    size=\"256x256\"\n",
+    "    data = {\"n\": n, \"prompt\": prompt, \"size\": size}\n",
+    "    headers = {\"Content-Type\": \"application/json\",\"Authorization\": f\"\"\"Bearer {openai_key}\"\"\"}\n",
+    "    result = requests.post(\"https://api.openai.com/v1/images/generations\",json=data,headers=headers)\n",
+    "    return result.json()[\"data\"][0][\"url\"]\n",
+    "\n",
+    "iface = gr.Interface(\n",
+    "    fn=prompt_builder, \n",
+    "    inputs=[\n",
+    "        gr.Textbox(placeholder=\"Logo Name\", label=\"Logo name\"),\n",
+    "        gr.Textbox(placeholder=\"Slogan (optional)\", label=\"Slogan\"),\n",
+    "        gr.Radio([\"Travel\", \"Sport Fitness\", \"Retail\", \"Religious\", \"Real Estate\", \n",
+    "                  \"Legal\", \"Internet\", \"Technology\", \"Home Family\", \"Events\"\n",
+    "                  \"Medical Dental\", \"Restaurant\", \"Finance\", \"Nonprofit\", \n",
+    "                  \"Entertainment\", \"Construction\", \"Education\", \"Beauty Spa\", \"Automotive\", \"Animals Pets\", \"Others\"], label=\"Industry\", info=\"Please Select A Industry\"),\n",
+    "        gr.Radio([\"Warm\", \"Cold\", \"Contrast\", \"Pastel\", \"Greyscale\", \"Gradient\"], label=\"Color scheme\", info=\"Select color schemes that matches your brand\"),\n",
+    "        gr.Radio([\"Modern\", \"Elegant\", \"Slab\", \"Handwritten\", \"Playful\", \"Futuristic\"],label=\"Font style\", info=\"Select font styles that you like\")\n",
+    "    ], \n",
+    "    outputs=\"image\")\n",
+    "iface.launch()\n",
+    "\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "'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'"
+      ]
+     },
+     "execution_count": 2,
+     "metadata": {},
+     "output_type": "execute_result"
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "key  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\n",
+      "result,  <Response [200]>\n"
+     ]
+    }
+   ],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3",
+   "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.11.2"
+  },
+  "orig_nbformat": 4
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

app.py

@@ -1,7 +1,27 @@
+import os
 import gradio as gr
+import requests
+openai_key = os.getenv("OPENAI_KEY")
+def prompt_builder(logo_name, slogan, industry, color_scheme, font_style):
+    n = 1
+    prompt = f"""Please generate a logo which name is {logo_name} {(", slogan is "+slogan) if slogan else ""}, industry is {industry}, color scheme is {color_scheme}, font style is {font_style}"""
+    size="256x256"
+    data = {"n": n, "prompt": prompt, "size": size}
+    headers = {"Content-Type": "application/json","Authorization": f"""Bearer {openai_key}"""}
+    result = requests.post("https://api.openai.com/v1/images/generations",json=data,headers=headers)
+    return result.json()["data"][0]["url"]
 
-def greet(name):
-    return "Hello " + name + "!!"
-
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+iface = gr.Interface(
+    fn=prompt_builder, 
+    inputs=[
+        gr.Textbox(placeholder="Logo Name", label="Logo name"),
+        gr.Textbox(placeholder="Slogan (optional)", label="Slogan"),
+        gr.Radio(["Travel", "Sport Fitness", "Retail", "Religious", "Real Estate", 
+                  "Legal", "Internet", "Technology", "Home Family", "Events"
+                  "Medical Dental", "Restaurant", "Finance", "Nonprofit", 
+                  "Entertainment", "Construction", "Education", "Beauty Spa", "Automotive", "Animals Pets", "Others"], label="Industry", info="Please Select A Industry"),
+        gr.Radio(["Warm", "Cold", "Contrast", "Pastel", "Greyscale", "Gradient"], label="Color scheme", info="Select color schemes that matches your brand"),
+        gr.Radio(["Modern", "Elegant", "Slab", "Handwritten", "Playful", "Futuristic"],label="Font style", info="Select font styles that you like")
+    ], 
+    outputs="image")
 iface.launch()