chore: gradio basic authentication
Browse files- gradio_chatbot.py +9 -0
- main.py +2 -2
gradio_chatbot.py
CHANGED
|
@@ -34,6 +34,15 @@ AVATAR_DIR = os.path.join(STATIC_DIR, "avatars")
|
|
| 34 |
USER_AVATAR = os.path.join(AVATAR_DIR, "user.png")
|
| 35 |
BOT_AVATAR = os.path.join(AVATAR_DIR, "bot.png")
|
| 36 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
# Custom CSS for fonts
|
| 38 |
CUSTOM_CSS = """
|
| 39 |
@font-face {{
|
|
|
|
| 34 |
USER_AVATAR = os.path.join(AVATAR_DIR, "user.png")
|
| 35 |
BOT_AVATAR = os.path.join(AVATAR_DIR, "bot.png")
|
| 36 |
|
| 37 |
+
AUTH_USERNAME = env.str("AUTH_USERNAME", "admin")
|
| 38 |
+
AUTH_PASSWORD = env.str("AUTH_PASSWORD", "admin")
|
| 39 |
+
|
| 40 |
+
|
| 41 |
+
def app_auth(username: str, password: str) -> bool:
|
| 42 |
+
logger.debug(f"Entering app_auth: Username: {username}")
|
| 43 |
+
logger.debug(f"AUTH_USERNAME: {AUTH_USERNAME}")
|
| 44 |
+
return username == AUTH_USERNAME and password == AUTH_PASSWORD
|
| 45 |
+
|
| 46 |
# Custom CSS for fonts
|
| 47 |
CUSTOM_CSS = """
|
| 48 |
@font-face {{
|
main.py
CHANGED
|
@@ -8,7 +8,7 @@ from loguru import logger
|
|
| 8 |
from environs import Env
|
| 9 |
from contextlib import asynccontextmanager
|
| 10 |
from app.db.client import mongodb
|
| 11 |
-
from gradio_chatbot import build_gradio_app
|
| 12 |
import gradio as gr
|
| 13 |
import os
|
| 14 |
|
|
@@ -118,7 +118,7 @@ app.include_router(chat_api.router)
|
|
| 118 |
|
| 119 |
# Build and mount Gradio app
|
| 120 |
demo = build_gradio_app()
|
| 121 |
-
app = gr.mount_gradio_app(app, demo, path="/ui")
|
| 122 |
|
| 123 |
|
| 124 |
@app.get("/")
|
|
|
|
| 8 |
from environs import Env
|
| 9 |
from contextlib import asynccontextmanager
|
| 10 |
from app.db.client import mongodb
|
| 11 |
+
from gradio_chatbot import build_gradio_app, app_auth
|
| 12 |
import gradio as gr
|
| 13 |
import os
|
| 14 |
|
|
|
|
| 118 |
|
| 119 |
# Build and mount Gradio app
|
| 120 |
demo = build_gradio_app()
|
| 121 |
+
app = gr.mount_gradio_app(app, demo, path="/ui", auth=app_auth)
|
| 122 |
|
| 123 |
|
| 124 |
@app.get("/")
|