Use OpenAI models

Author: mxcl

.github/scripts/gen-imgs.py

@@ -96,28 +96,21 @@ def get_painting_feature(name, description):
         prompt += f" which is described as “{description}”"
 
     prompt += f""".
-From that name and description extract some language that can be featured in the image.
-Eg. if the description is “foo is a bridge to bar” then output BRIDGE.
-Keep it physical. If the description is "foo is a dynamic runtime bridge to bar” then output BRIDGE, not DYNAMIC RUNTIME BRIDGE.
-Remember I need to paint this!
-If the name is a play on words then find the play on words. Eg “deno” is a play on dino ie. a dinosaur.
-If you know the logo for a project and the logo is a good theme, then say that. Eg. “deno”’s logo is a dinosaur.
-If the name contains a word I can paint, use it! Eg. from `libcap` you can say `cap`.
-Don’t be generic. I have ten million open source packages. eg. don’t pick `lib` or `code` or `syntax`, but eg. database is ok.
-I need the output as JSON "{{theme: ""}}.
-I am feeding the output to a script, so if you don’t output json in the above form my script will break.
+From that name and description extract a word or two that can be the *feature* of the painting.
+Remember it needs to be painted! So stick to objects, animals, plants, or other things that can be painted.
 Thank you sir.
 """
     prompt = re.sub(r'\s+', ' ', prompt).strip()
 
     print(prompt)
 
-    proc = subprocess.run(['pkgx', 'ollama', 'run', 'deepseek-r1', f"Instruct: {prompt}\nOutput: "], capture_output=True, text=True)
+    proc = subprocess.run(['pkgx', 'llm', 'prompt', f"Instruct: {prompt}\nOutput: "], capture_output=True, text=True)
     result = proc.stdout
 
     console.print(result, style="yellow")
 
-    result = result[result.index("</think>") + len("</think>"):].strip()
+    if "</think>" in result:
+        result = result[result.index("</think>") + len("</think>"):].strip()
 
     output = {}
 
@@ -151,12 +144,13 @@ def get_painting_feature(name, description):
 
     prompt = f"Instruct: {prompt}\nOutput: "
 
-    proc = subprocess.run(['pkgx', 'ollama', 'run', 'deepseek-r1', prompt], capture_output=True, text=True)
+    proc = subprocess.run(['pkgx', 'llm', 'prompt', prompt], capture_output=True, text=True)
     result = proc.stdout
 
     console.print(result, style="blue")
 
-    result = result[result.index("</think>") + len("</think>"):].strip()
+    if "</think>" in result:
+        result = result[result.index("</think>") + len("</think>"):].strip()
 
     match = re.search(r'{\s*"?colors"?:(.*?)}', result, re.DOTALL)
     if match:
@@ -167,7 +161,8 @@ def get_painting_feature(name, description):
 
     prompt = f"""
 I have a theme: {output['theme']} and colors: {output['colors']}.
-Now pick from following list of imagery options to best fit that theme and color selection:
+Now I need surrounding imagery. Here’s a list of possible imagery *collections*.
+But these are just examples, I would prefer you come up with your own based on the theme and colors.
 
 ```markdown
 - oceans, waves, tides, coral reefs
@@ -182,22 +177,22 @@ def get_painting_feature(name, description):
 - twilight, dawn, dusk, moonlight
 ```
 
-If none fit and you are inspired please provide your own imagery or adapt the options.
+Don’t go overboard, I only want three or four items.
 I need the output as JSON "{{imagery: "FOO, BAR, BAZ, ETC"}}.
-Output the complete list item of imagery items, not just selected items.
 I am feeding the output to a script, so if you don’t output json in the above form my script will break.
 Thank you sir.
 """
     print(prompt)
 
     prompt = f"Instruct: {prompt}\nOutput: "
 
-    proc = subprocess.run(['pkgx', 'ollama', 'run', 'deepseek-r1', prompt], capture_output=True, text=True)
+    proc = subprocess.run(['pkgx', 'llm', 'prompt', prompt], capture_output=True, text=True)
     result = proc.stdout
 
     console.print(result, style="green")
 
-    result = result[result.index("</think>") + len("</think>"):].strip()
+    if "</think>" in result:
+        result = result[result.index("</think>") + len("</think>"):].strip()
 
     match = re.search(r'{\s*"?imagery"?:(.*?)}', result, re.DOTALL)
     if match: