Merge branch 'master' of https://git.yongyuancv.cn/heimoshuiyu/matrix-chain

.dockerignore

@@ -2,3 +2,4 @@ __pycache__
 /venv
 /.env
 .git
+/cache

.gitignore

@@ -2,3 +2,4 @@ __pycache__
 /venv
 /.env
 .git
+/cache

Dockerfile

@@ -4,7 +4,7 @@ FROM python:3.11-slim-buster
 RUN apt-get update -y
 
 # Install LibreOffice, Firefox ESR and pip
-RUN apt-get install -y libreoffice firefox-esr
+RUN apt-get install -y libreoffice firefox-esr ffmpeg
 
 # Set the working directory in the container to /app
 WORKDIR /app

bot_db.py

@@ -20,6 +20,11 @@ from langchain.embeddings import OpenAIEmbeddings
 
 from selenium import webdriver
 
+if not os.path.exists("./cache"):
+    os.mkdir("./cache")
+if not os.path.exists("./cache/yt-dlp"):
+    os.mkdir("./cache/yt-dlp")
+
 print("lanuching driver")
 options = webdriver.FirefoxOptions()
 options.add_argument("-headless")

docker-compose.yaml

@@ -1,25 +1,25 @@
 version: '3'
 services:
   bot-chatgpt:
-    image: matrix-chain
+    image: heimoshuiyu/matrix-chain:latest
     env_file:
       - ./.env
     command: python3 bot_chatgpt.py
 
   bot-db:
-    image: matrix-chain
+    image: heimoshuiyu/matrix-chain:latest
     env_file:
       - ./.env
     command: python3 bot_db.py
 
   bot-whisper:
-    image: matrix-chain
+    image: heimoshuiyu/matrix-chain:latest
     env_file:
       - ./.env
     command: python3 bot_whisper.py
 
   bot-tts:
-    image: matrix-chain
+    image: heimoshuiyu/matrix-chain:latest
     env_file:
       - ./.env
     command: python3 bot_tts.py