Initial commit.

Author: githubChenchi

LICENSE.txt

@@ -0,0 +1,13 @@
+Copyright (c) Quectel Wireless Solution, Co., Ltd.All Rights Reserved.
+ 
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+ 
+    http://www.apache.org/licenses/LICENSE-2.0
+ 
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.

code/ai_main.py

@@ -0,0 +1,146 @@
+import sim
+from usr.ui import SelectWindow, ChatWindow
+from usr.jobs import scheduler
+from usr import pypubsub as pub
+
+import dataCall
+import utime as time
+import TiktokRTC
+import atcmd
+import _thread
+
+from machine import Pin
+
+PA = Pin.GPIO39
+
+from machine import ExtInt
+from queue import Queue
+
+def key1(args):
+    global rtc_queue
+    rtc_queue.put(1)
+
+def key2(args):
+    global rtc_queue
+    rtc_queue.put(2)
+
+def enable_pid2():
+    resp = bytearray(50)
+    atcmd.sendSync('AT+qicsgp=2,1,3gnet,"","",0\r\n',resp,'',20)
+    atcmd.sendSync('at+cgact=1,2\r\n',resp,'',20)
+
+def ai_callback(args):
+    global GPIO39
+    event = args[0]
+    msg = args[1]
+    global chat_win
+    if event == 1:
+        print('TIKTOK_RTC_EVENT_START')
+        GPIO39.write(1)
+        chat_win.update_status("Please speak to me")
+    elif event == 2:
+        print('TIKTOK_RTC_EVENT_STOP')
+        GPIO39.write(0)
+    elif event == 3:
+        #chat_win.update_status("AI speaking . . .")
+        print('TIKTOK_RTC_EVENT_TTS_TEXT {}'.format(msg))
+        #call.stopAudioService()
+    elif event == 4:
+        #chat_win.update_status("AI listening . . .")
+        print('TIKTOK_RTC_EVENT_ASR_TEXT {}'.format(msg))
+        #call.stopAudioService()
+    elif event == 5:
+        print('TIKTOK_RTC_EVENT_ERROR {}'.format(msg))
+    else:
+        print('TIKTOK_RTC_EVENT UNKNOWN {}'.format(event))
+
+def update_status_with_animation(chat_win, base_message, steps=3, delay_ms=400, final_wait=2):
+    # 更新动画
+    for i in range(steps + 1):
+        chat_win.update_status(base_message + " " + " ." * i)
+        time.sleep_ms(delay_ms)
+    time.sleep(final_wait)
+
+def perform_initialization(chat_win, tiktok):
+    # 初始化动画加载状态
+    print('start rtc') 
+    chat_win.show()
+    tiktok.active(True)
+
+    # 需要展示的状态列表
+    status_list = [
+        "Connecting to the server",
+        "Building the AI engine",
+        "Joining the AI room",
+        "Loading AI personality",
+        "Creating AI characters"
+    ]
+
+    # 依次遍历状态并显示动画
+    for status in status_list:
+        update_status_with_animation(chat_win, status)
+def ai_task():
+    global rtc_queue
+    global extint1
+    global extint2
+    global tiktok
+    global chat_win
+    global selsct_win
+    while True:
+        lte = dataCall.getInfo(1, 0)
+        if lte[2][0] == 1:
+            print('lte network normal')
+            #pub.publish('update_status', status="ready")
+            break
+        print('wait lte network normal...')
+        pub.publish('update_status', status="connect network")
+        time.sleep(3)
+
+    extint1.enable()
+    extint2.enable()
+    print('ai task running')
+    while True:
+        data = rtc_queue.get()
+        print('rtc_queue key event {}'.format(data))
+        if data == 1:
+            perform_initialization(chat_win, tiktok)
+        elif data == 2:
+            print('stop rtc')
+            selsct_win.show()
+            tiktok.active(False)
+            
+
+if __name__ == "__main__":
+
+    enable_pid2()
+    
+    # 使能sim卡热插拔
+    sim.setSimDet(1, 1)
+
+    # 设置按键中断
+    extint1 = ExtInt(ExtInt.GPIO13, ExtInt.IRQ_FALLING, ExtInt.PULL_PU, key1, filter_time=50)
+    extint2 = ExtInt(ExtInt.GPIO12, ExtInt.IRQ_FALLING, ExtInt.PULL_PU, key2, filter_time=50)
+
+    rtc_queue = Queue()
+
+    # 初始化界面
+    selsct_win = SelectWindow()
+    selsct_win.show()
+    
+    chat_win = ChatWindow()
+    
+    # 启动后台任务调度器
+    scheduler.start()
+    
+    print('window show over')
+
+    tiktok = TiktokRTC(300000, ai_callback)
+    GPIO39 = Pin(PA, Pin.OUT, Pin.PULL_DISABLE, 0)
+    tiktok.config(volume=6)
+    print('volume: {}'.format(tiktok.config('volume')))
+
+    _thread.start_new_thread(ai_task, ())
+
+
+    
+