optimize: improve prompt

app.ipynb

@@ -35,14 +35,14 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 15,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Running on local URL:  http://127.0.0.1:7867\n",
+      "Running on local URL:  http://127.0.0.1:7873\n",
       "\n",
       "To create a public link, set `share=True` in `launch()`.\n"
      ]
@@ -50,7 +50,7 @@
     {
      "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>"
+       "<div><iframe src=\"http://127.0.0.1:7873/\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
       ],
       "text/plain": [
        "<IPython.core.display.HTML object>"
@@ -63,9 +63,31 @@
      "data": {
       "text/plain": []
      },
-     "execution_count": 9,
+     "execution_count": 15,
      "metadata": {},
      "output_type": "execute_result"
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "prompt  \n",
+      "        I would like you to act as an SVG designer. I will ask you to create images, and you will come up with SVG code for the image,\n",
+      "        Create a logo for a company called 'Shareverse' \n",
+      "        The company specializes in Technology research and development. \n",
+      "        The logo should convey a sense of Contrast, Elegant, \n",
+      "        and the microscopic world of Technology. \n",
+      "        Use a color scheme that represents Contrast. \n",
+      "    \n",
+      "prompt  \n",
+      "        I would like you to act as an SVG designer. I will ask you to create images, and you will come up with SVG code for the image,\n",
+      "        Create a logo for a company called 'Shareverse' \n",
+      "        The company specializes in Beauty Spa research and development. \n",
+      "        The logo should convey a sense of Cold, Elegant, \n",
+      "        and the microscopic world of Beauty Spa. \n",
+      "        Use a color scheme that represents Cold. \n",
+      "    \n"
+     ]
     }
    ],
    "source": [
@@ -74,9 +96,29 @@
     "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",
+    "class Logo:\n",
+    "    def __init__(self,name,slogan, industry, color_scheme, font_style) -> None:\n",
+    "        self.name = name\n",
+    "        self.slogan = slogan\n",
+    "        self.industry = industry\n",
+    "        self.color_scheme = color_scheme\n",
+    "        self.font_style = font_style\n",
+    "\n",
+    "def build_prompt(desc) -> str:\n",
+    "    return f\"\"\"\n",
+    "        I would like you to act as an SVG designer. I will ask you to create images, and you will come up with SVG code for the image,\n",
+    "        Create a logo for a company called '{desc.name}' \n",
+    "        The company specializes in {desc.industry} research and development. \n",
+    "        The logo should convey a sense of {desc.color_scheme}, {desc.font_style}, \n",
+    "        and the microscopic world of {desc.industry}. \n",
+    "        Use a color scheme that represents {desc.color_scheme}. \n",
+    "    \"\"\"\n",
+    "\n",
+    "def ganerate(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",
+    "    logo = Logo(logo_name, slogan, industry, color_scheme, font_style)\n",
+    "    prompt = build_prompt(logo)\n",
     "    size=\"256x256\"\n",
     "    data = {\"n\": n, \"prompt\": prompt, \"size\": size}\n",
     "    headers = {\"Content-Type\": \"application/json\",\"Authorization\": f\"\"\"Bearer {openai_key}\"\"\"}\n",
@@ -84,7 +126,7 @@
     "    return result.json()[\"data\"][0][\"url\"]\n",
     "\n",
     "iface = gr.Interface(\n",
-    "    fn=prompt_builder, \n",
+    "    fn=ganerate, \n",
     "    inputs=[\n",
     "        gr.Textbox(placeholder=\"Logo Name\", label=\"Logo name\"),\n",
     "        gr.Textbox(placeholder=\"Slogan (optional)\", label=\"Slogan\"),\n",

app.py

@@ -2,9 +2,29 @@ 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):
+
+class Logo:
+    def __init__(self,name,slogan, industry, color_scheme, font_style) -> None:
+        self.name = name
+        self.slogan = slogan
+        self.industry = industry
+        self.color_scheme = color_scheme
+        self.font_style = font_style
+
+def build_prompt(desc) -> str:
+    return f"""
+        I would like you to act as an SVG designer. I will ask you to create images, and you will come up with SVG code for the image,
+        Create a logo for a company called '{desc.name}' 
+        The company specializes in {desc.industry} research and development. 
+        The logo should convey a sense of {desc.color_scheme}, {desc.font_style}, 
+        and the microscopic world of {desc.industry}. 
+        Use a color scheme that represents {desc.color_scheme}. 
+    """
+
+def ganerate(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}"""
+    logo = Logo(logo_name, slogan, industry, color_scheme, font_style)
+    prompt = build_prompt(logo)
     size="256x256"
     data = {"n": n, "prompt": prompt, "size": size}
     headers = {"Content-Type": "application/json","Authorization": f"""Bearer {openai_key}"""}
@@ -12,7 +32,7 @@ def prompt_builder(logo_name, slogan, industry, color_scheme, font_style):
     return result.json()["data"][0]["url"]
 
 iface = gr.Interface(
-    fn=prompt_builder, 
+    fn=ganerate, 
     inputs=[
         gr.Textbox(placeholder="Logo Name", label="Logo name"),
         gr.Textbox(placeholder="Slogan (optional)", label="Slogan"),

flagged/log.csv

@@ -0,0 +1,2 @@
+Logo name,Slogan,Industry,Color scheme,Font style,output,flag,username,timestamp
+Shareverse,,Technology,Cold,Elegant,/Users/ny0/workspace/ailogo/flagged/output/1516953fb06afb7de691ab6f2625e7b75c0081ec/tmpr238pvma.txt,,,2023-09-12 01:32:19.153945