chunzhi/chatgpt_academic
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

unifying code

Browse Source
This commit is contained in:
Your Name 2023-04-22 17:24:22 +08:00
parent 6f7e8076c7
commit 7beea951c6
3 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config.py
View File

@ -62,4 +62,4 @@ AUTHENTICATION = []
API_URL_REDIRECT = {} API_URL_REDIRECT = {}
# 如果你需要把网址放在二级地址下（常规情况下，不要修改！！） # 如果你需要把网址放在二级地址下（常规情况下，不要修改！！）
CUSTOM_PATH = "/" CUSTOM_PATH = "/gra"

8
main.py
View File

@ -176,18 +176,18 @@ def main():
def auto_opentab_delay(): def auto_opentab_delay():
import threading, webbrowser, time import threading, webbrowser, time
print(f"如果浏览器没有自动打开，请复制并转到以下URL：") print(f"如果浏览器没有自动打开，请复制并转到以下URL：")
print(f"\t（亮色主题）: http://localhost:{PORT}") print(f"\t（亮色主题）: http://localhost:{PORT}" + f"{CUSTOM_PATH}".replace('//','/'))
print(f"\t（暗色主题）: http://localhost:{PORT}/?__dark-theme=true") print(f"\t（暗色主题）: http://localhost:{PORT}" + f"{CUSTOM_PATH}/?__dark-theme=true".replace('//','/'))
def open(): def open():
time.sleep(2) # 打开浏览器 time.sleep(2) # 打开浏览器
webbrowser.open_new_tab(f"http://localhost:{PORT}/?__dark-theme=true") webbrowser.open_new_tab(f"http://localhost:{PORT}" + f"{CUSTOM_PATH}/?__dark-theme=true".replace('//','/'))
threading.Thread(target=open, name="open-browser", daemon=True).start() threading.Thread(target=open, name="open-browser", daemon=True).start()
threading.Thread(target=auto_update, name="self-upgrade", daemon=True).start() threading.Thread(target=auto_update, name="self-upgrade", daemon=True).start()
threading.Thread(target=warm_up_modules, name="warm-up", daemon=True).start() threading.Thread(target=warm_up_modules, name="warm-up", daemon=True).start()
auto_opentab_delay() auto_opentab_delay()
demo.queue(concurrency_count=CONCURRENT_COUNT) demo.queue(concurrency_count=CONCURRENT_COUNT)
run_gradio(demo, auth=AUTHENTICATION, favicon_path="docs/logo.png", port=PORT, custom_path=CUSTOM_PATH) run_gradio(demo, auth=AUTHENTICATION, port=PORT, custom_path=CUSTOM_PATH)
if __name__ == "__main__": if __name__ == "__main__":
main() main()

9
toolbox.py
View File

@ -521,13 +521,14 @@ class DummyWith():
def __exit__(self, exc_type, exc_value, traceback): def __exit__(self, exc_type, exc_value, traceback):
return return
def run_gradio(demo, auth, favicon_path, port, custom_path): def run_gradio(demo, auth, port, custom_path):
import uvicorn import uvicorn
import gradio as gr import gradio as gr
from fastapi import FastAPI from fastapi import FastAPI
app = FastAPI() app = FastAPI()
if custom_path != "/":
@app.get("/") @app.get("/")
def read_main(): def read_main():
return {"message": "NULL"} return {"message": f"Gradio is running at: {custom_path}"}
app = gr.mount_gradio_app(app, demo, path=custom_path, auth=auth, favicon_path=favicon_path) app = gr.mount_gradio_app(app, demo, path=custom_path)
uvicorn.run(app, host="0.0.0.0", port=port) uvicorn.run(app, host="0.0.0.0", port=port) # , auth=auth