Tune the Docker image (#52)

* Docker guide

* docker flag

* Running in Docker with user and password

* 谁把import openai 注释了？

* Update requirements.txt

* Deal with Docker api conflict

* Update README.md

* # import openai

* # import openai

ChuanhuChatbot.py

@@ -9,15 +9,28 @@ import requests
 
 my_api_key = ""    # 在这里输入你的 API 密钥
 initial_prompt = "You are a helpful assistant."
-
 API_URL = "https://api.openai.com/v1/chat/completions"
 
-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)
+
+#if we are running in Docker
+if os.environ.get('dockerrun') == 'yes':
+    dockerflag = True
+else:
+    dockerflag = False
+    
+if dockerflag:
+    my_api_key = os.environ.get('my_api_key')
+    if my_api_key == "empty":
+        print("Please give a api key!")
+        sys.exit(1)
+    #auth
+    username = os.environ.get('USERNAME')
+    password = os.environ.get('PASSWORD')
+    if isinstance(username, type(None)) or isinstance(password, type(None)):
+        authflag = False
+    else:
+        authflag = True
 
 
 def parse_text(text):
@@ -266,6 +279,15 @@ with gr.Blocks() as demo:
 print("川虎的温馨提示：访问 http://localhost:7860 查看界面")
 # 默认开启本地服务器，默认可以直接从IP访问，默认不创建公开分享链接
 demo.title = "川虎ChatGPT 🚀"
-demo.queue().launch(server_name = "0.0.0.0", share=False) # 改为 share=True 可以创建公开分享链接
-# demo.queue().launch(server_name="0.0.0.0", server_port=7860, share=False) # 可自定义端口
-# demo.queue().launch(server_name="0.0.0.0", server_port=7860,auth=("在这里填写用户名", "在这里填写密码")) # 可设置用户名与密码
+
+#if running in Docker
+if dockerflag:
+    if authflag:
+        demo.queue().launch(server_name="0.0.0.0", server_port=7860,auth=(username, password))
+    else:
+        demo.queue().launch(server_name="0.0.0.0", server_port=7860, share=False)
+#if not running in Docker
+else:
+    demo.queue().launch(server_name = "0.0.0.0", share=False) # 改为 share=True 可以创建公开分享链接
+    #demo.queue().launch(server_name="0.0.0.0", server_port=7860, share=False) # 可自定义端口
+    #demo.queue().launch(server_name="0.0.0.0", server_port=7860,auth=("在这里填写用户名", "在这里填写密码")) # 可设置用户名与密码

Dockerfile

@@ -10,4 +10,5 @@ ENV PATH=/root/.local/bin:$PATH
 COPY . /app
 WORKDIR /app
 ENV my_api_key empty
+ENV dockerrun yes
 CMD ["python3", "-u", "ChuanhuChatbot.py", "2>&1", "|", "tee", "/var/log/application.log"]

README.md

@@ -83,29 +83,32 @@ python3 ChuanhuChatbot.py
 
 如果还是不行，请先[安装Python](https://www.runoob.com/python/python-install.html)。
 
-## 或者，使用Docker 安装与运行
+## 或者，使用Docker 运行
 
-### 从本项目的Packages页面拉取
-
-从本项目的[Packages](https://github.com/GaiZhenbiao/ChuanhuChatGPT/pkgs/container/chuanhuchatgpt)页面拉取Docker镜像，使用Github Actions自动创建。也可以去本项目的[Dockerhub页面](https://hub.docker.com/r/tuchuanhuhuhu/chuanhuchatgpt)拉取。
-
-### 手动构建镜像
+### 拉取镜像
 
 ```
-docker build -t chuanhuchatgpt:latest .
+docker pull tuchuanhuhuhu/chuanhuchatgpt:latest
 ```
 
-运行
+### 运行
 
 ```
-docker run -d --name chatgpt -e my_api_key="替换成API"  --network host chuanhuchatgpt:latest
+docker run -d --name chatgpt -e my_api_key="替换成API"  -p 7860:7860 tuchuanhuhuhu/chuanhuchatgpt:latest
 ```
 
-### 查看本地访问地址
+### 查看运行状态
 ```
 docker logs chatgpt
 ```
 
+### 也可修改脚本后手动构建镜像
+
+```
+docker build -t chuanhuchatgpt:latest .
+```
+
+
 ## 部署相关
 
 ### 部署到公网服务器