1 files changed, 13 insertions, 1 deletions
diff --git a/src/main.cpp b/src/main.cpp
index b5afe53..31d1047 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -44,12 +44,16 @@ void loop()
     uint16_t pressDuration = de::uvok::badge::gpio_poll();
     if (pressDuration < 64)
     {
+        uint8_t newNum = UINT8_MAX;
         if (pressDuration == NEXT_KEY)
             de::uvok::badge::display_next();
         if (pressDuration == PRV_KEY)
             de::uvok::badge::display_prev();
         if (pressDuration == OK_KEY)
-            de::uvok::badge::display_refresh();
+            newNum = de::uvok::badge::display_refresh();
+
+        if (newNum != UINT8_MAX)
+            de::uvok::badge::ble_set_image(newNum);
     }
     else if (pressDuration > PRESS_TIME_BLE_ADV)
     {
@@ -61,6 +65,14 @@ void loop()
         de::uvok::badge::display_demo();
     }
 
+    {
+        de::uvok::badge::ble_poll_result_t pollres = de::uvok::badge::ble_poll();
+        if (pollres.has_data)
+        {
+            de::uvok::badge::display_direct(pollres.new_template);
+        }
+    }
+
     // if (millis() - releasedTime > SLEEP_TIME && !pAdvertising->isAdvertising())
     // {
     //   Serial.println("Go to sleep...");