add new project

CircuitPython_GetSuperpower_PicoW_OpenAI/code.py

@@ -0,0 +1,204 @@
+# SPDX-FileCopyrightText: 2023 Jeff Epler for Adafruit Industries
+# SPDX-License-Identifier: MIT
+import json
+import os
+import ssl
+
+import board
+import displayio
+import digitalio
+import keypad
+import socketpool
+from wifi import radio
+
+import adafruit_requests
+import adafruit_displayio_ssd1306
+from adafruit_bitmap_font.bitmap_font import load_font
+from adafruit_display_text import wrap_text_to_pixels
+from adafruit_display_text.bitmap_label import Label
+from adafruit_ticks import ticks_add, ticks_less, ticks_ms
+
+
+# Choose your own prompt and wait messages, either by changing it below inside
+# the """triple quoted""" string, or by putting it in your settings.toml file,
+# like so:
+#
+# MY_PROMPT="Give me an idea for a plant-based dinner. Write one sentence"
+# PLEASE_WAIT="Cooking something up just for you"
+
+# Here are some prompts you might want to try:
+
+# Give me an idea for a plant-based dinner. Write one sentence
+#
+# Give jepler a description as a comic book supervillain. write one sentence.
+#
+# Invent and describe an alien species. write one sentence
+#
+# Invent a zany "as seen on TV product" that can't possibly work. One sentence
+#
+# Tell a 1-sentence story about a kitten and a funny mishap
+#
+# Make up a 1-sentence fortune for me
+#
+# In first person, write a 1-sentence story about an AI avoiding boredom in a creative way.
+#
+# Pick an everyday object (don't say what it is) and describe it using only the
+# ten hundred most common words.
+#
+# Invent an alien animal or plant, name it, and vividly describe it in 1
+# sentence
+prompt=os.getenv("MY_PROMPT", """
+Write 1 setence starting "you can" about an unconventional but useful superpower 
+""").strip()
+please_wait=os.getenv("PLEASE_WAIT", """
+Finding superpower
+""").strip()
+
+openai_api_key = os.getenv("OPENAI_API_KEY")
+
+nice_font = load_font("helvR08.pcf")
+line_spacing = 0.68
+
+#  i2c display setup
+displayio.release_displays()
+oled_reset = board.GP9
+
+# STEMMA I2C on picowbell
+i2c = board.STEMMA_I2C()
+display_bus = displayio.I2CDisplay(i2c, device_address=0x3D, reset=oled_reset)
+
+WIDTH = 128
+HEIGHT = 64
+offset_y = 5
+
+display = adafruit_displayio_ssd1306.SSD1306(
+    display_bus, width=WIDTH, height=HEIGHT
+)
+if openai_api_key is None:
+    input("Place your\nOPENAI_API_KEY\nin settings.toml")
+display.auto_refresh = False
+main_group = displayio.Group()
+display.root_group = main_group
+
+terminal = Label(
+    font=nice_font,
+    color=0xFFFFFF,
+    background_color=0,
+    line_spacing=line_spacing,
+    anchor_point=(0, 0),
+    anchored_position=(0, 0),
+)
+max_lines = display.height // int(nice_font.get_bounding_box()[1] * terminal.line_spacing)
+main_group.append(terminal)
+
+class WrappedTextDisplay:
+    def __init__(self):
+        self.line_offset = 0
+        self.lines = []
+        self.text = ""
+
+    def add_text(self, new_text):
+        self.set_text(self.text + new_text)
+        self.scroll_to_end()
+
+    def set_text(self, text):
+        print("\033[H\033[2J", end=text)
+        self.text = text
+        self.lines = wrap_text_to_pixels(text, display.width, nice_font)
+        self.line_offset = 0
+
+    def show(self, text):
+        self.set_text(text)
+        self.refresh()
+
+    def add_show(self, new_text):
+        self.add_text(new_text)
+        self.refresh()
+
+    def scroll_to_end(self):
+        self.line_offset = self.max_offset()
+
+    def scroll_next_line(self):
+        max_offset = self.max_offset()
+        if max_offset > 0:
+            line_offset = self.line_offset + 1
+            self.line_offset = line_offset % (max_offset + 1)
+
+    def max_offset(self):
+        return max(0, len(self.lines) - max_lines)
+
+    def on_last_line(self):
+        return self.line_offset == self.max_offset()
+
+    def refresh(self):
+        text = '\n'.join(self.lines[self.line_offset : self.line_offset + max_lines])
+        terminal.text = text
+        display.refresh()
+wrapped_text = WrappedTextDisplay()
+
+def wait_button_scroll_text():
+    led.switch_to_output(True)
+    deadline = ticks_add(ticks_ms(),
+            5000 if wrapped_text.on_last_line() else 1000)
+    while True:
+        if (event := keys.events.get()) and event.pressed:
+            break
+        if wrapped_text.max_offset() > 0 and ticks_less(deadline, ticks_ms()):
+            wrapped_text.scroll_next_line()
+            wrapped_text.refresh()
+            deadline = ticks_add(deadline,
+                    5000 if wrapped_text.on_last_line() else 1000)
+    led.value = False
+
+if radio.ipv4_address is None:
+    wrapped_text.show(f"connecting to {os.getenv('WIFI_SSID')}")
+    radio.connect(os.getenv('WIFI_SSID'), os.getenv('WIFI_PASSWORD'))
+requests = adafruit_requests.Session(socketpool.SocketPool(radio), ssl.create_default_context())
+
+def iter_lines(resp):
+    partial_line = []
+    for c in resp.iter_content():
+        if c == b'\n':
+            yield (b"".join(partial_line)).decode('utf-8')
+            del partial_line[:]
+        else:
+            partial_line.append(c)
+    if partial_line:
+        yield (b"".join(partial_line)).decode('utf-8')
+
+full_prompt = [
+    {"role": "user", "content": prompt},
+]
+
+keys = keypad.Keys((board.GP14,), value_when_pressed=False)
+led = digitalio.DigitalInOut(board.LED)
+
+while True:
+    wrapped_text.show(please_wait)
+
+    with requests.post("https://api.openai.com/v1/chat/completions",
+        json={"model": "gpt-3.5-turbo", "messages": full_prompt, "stream": True},
+        headers={
+            "Authorization": f"Bearer {openai_api_key}",
+        },
+        ) as response:
+
+        wrapped_text.set_text("")
+        if response.status_code != 200:
+            wrapped_text.show(f"Uh oh! {response.status_code}: {response.reason}")
+        else:
+            wrapped_text.show("")
+            for line in iter_lines(response):
+        #        print(line)
+        #        continue
+                if line.startswith("data: [DONE]"):
+                    break
+                if line.startswith("data:"):
+                    content = json.loads(line[5:])
+                    try:
+                        token = content['choices'][0]['delta'].get('content', '')
+                    except (KeyError, IndexError) as e:
+                        token = None
+                    if token:
+                        wrapped_text.add_show(token)
+            wait_button_scroll_text()