cevheri commited on
Commit
e73254b
·
1 Parent(s): 25a073c

chore: gradio basic authentication

Browse files
Files changed (2) hide show
  1. gradio_chatbot.py +9 -0
  2. 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("/")