Merge pull request #7 from iskoldt-X/main

ChuanhuChatbot.py

@@ -1,12 +1,20 @@
 import gradio as gr
 import openai
+import os
+import sys
 # import markdown
 
-
-我的API密钥 = "sk-XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX"    # 在这里输入你的 API 密钥
+my_api_key = ""    # 在这里输入你的 API 密钥
 initial_prompt = "You are a helpful assistant."
 
-openai.api_key = 我的API密钥
+if my_api_key == "":
+    my_api_key = os.environ.get('my_api_key')
+
+if my_api_key == "empty":
+    print("Please give a api key!")
+    sys.exit(1)
+
+openai.api_key = my_api_key
 
 def parse_text(text):
     lines = text.split("\n")

Dockerfile

@@ -0,0 +1,13 @@
+FROM python:3.9 as builder
+RUN apt-get update && apt-get install -y build-essential
+COPY requirements.txt .
+RUN pip install --user -r requirements.txt
+
+FROM python:3.9
+MAINTAINER iskoldt
+COPY --from=builder /root/.local /root/.local
+ENV PATH=/root/.local/bin:$PATH
+COPY . /app
+WORKDIR /app
+ENV my_api_key empty
+CMD ["python3", "-u", "ChuanhuChatbot.py", "2>&1", "|", "tee", "/var/log/application.log"]

README.md

@@ -42,6 +42,26 @@ python3 ChuanhuChatbot.py
 
 如果还是不行，请先[安装Python](https://www.runoob.com/python/python-install.html)。
 
+## 使用Docker 安装与运行
+
+构建镜像
+```
+docker build -t chuanhuchatgpt:latest .
+```
+
+一键运行
+```
+docker run -d --name chatgpt -e my_api_key="替换成API"  --network host chuanhuchatgpt:latest
+```
+
+查看本地访问地址
+```
+docker logs chatgpt
+```
+
+
+
+
 ## 使用技巧
 
 - 使用System Prompt可以很有效地设定前提条件