updated main with auth

.env

@@ -1,4 +1,5 @@
 OPENAI_API_KEY=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
 ELEVEN_API_KEY=5db7eb7ac620c89ec32e72de98128417
 REDIS_HOST=localhost
-REDIS_PORT=6379
+REDIS_PORT=6379
+GRADIO_PASSWORD=oscepal

main.py

@@ -10,10 +10,9 @@ from queue import Empty, Queue
 from threading import Thread
 import os
 import gradio as gr
-from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
 from langchain.chat_models import ChatOpenAI
 from langchain.prompts import HumanMessagePromptTemplate, SystemMessagePromptTemplate
-from langchain.schema import AIMessage, BaseMessage, HumanMessage, SystemMessage
+from langchain.schema import AIMessage, BaseMessage, HumanMessage
 from js import get_window_url_params
 from callback import QueueCallback
 from fastapi import FastAPI, File, UploadFile, Request
@@ -35,7 +34,7 @@ DEFAULT_TEMPERATURE = 0.7
 ChatHistory = List[str]
 
 logging.basicConfig(
-    format="[%(asctime)s %(levelname)s]: %(message)s", level=logging.DEBUG
+    format="[%(asctime)s %(levelname)s]: %(message)s", level=logging.INFO
 )
 # load redis client
 client = get_client()
@@ -142,8 +141,8 @@ def on_drop_down_change(selected_item, messages):
     return patient, patient, [], messages
 
 
-def on_demo_load(url_params):
-    username = url_params.get("username", "test")
+def on_demo_load(url_params, request: gr.Request):
+    username = request.username or url_params.get("username", "test")
     logging.info(f"Getting user for username={username}")
     create_user(client, User(username=username, uid=None))
     user = get_user_by_username(client, username)
@@ -247,9 +246,14 @@ with gr.Blocks(
     # async def home(request: Request):
     #     return templates.TemplateResponse(
     #         "home.html", {"request": request, "videos": []})
-
-    demo.queue()
-    demo.launch()
+def auth_handler(username: str, password: str) -> bool:
+    if password != os.environ["GRADIO_PASSWORD"]:
+        return False
+    
+    return True
+
+demo.queue()
+demo.launch(auth=auth_handler)
 
     # gradio_app = gr.routes.App.create_app(demo)
     # app.mount("/gradio", gradio_app)