add auth

app.py

@@ -299,4 +299,5 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue='orange', secondary_hue='gray
     msg.submit(**chat_btn_args)
 
 demo.queue(default_concurrency_limit=10)
-demo.launch(show_error=True, auth=(os.getenv("USERNAME",''), os.getenv("PASSWORD",'')))
+gradio_auth=None # comment this if you want to use Authentication
+demo.launch(show_error=True, auth=gradio_auth)

ttyd_functions.py

@@ -52,7 +52,7 @@ def getOaiCreds(key):
 def getBamCreds(key):
     key = key if key else 'Null'
     return {'service': 'bam',
-                'bam_creds' : Credentials(key, api_endpoint='https://bam-api.res.ibm.com/v1')
+                'bam_creds' : Credentials(key, api_endpoint=os.getenv("BAM_ENDPOINT",'https://bam-api.res.ibm.com/v1'))
             }
 
 
@@ -373,4 +373,10 @@ def changeModel(oldModel, newModel):
     return newModel
 
 def getModelChoices(openAi_models, wml_models, bam_models):
-    return [model +' (openai)' for model in openAi_models] + [model.value +' (watsonx)' for model in wml_models] + [model + ' (bam)' for model in bam_models]
+    return [model +' (openai)' for model in openAi_models] + [model.value +' (watsonx)' for model in wml_models] + [model + ' (bam)' for model in bam_models]
+
+def gradio_auth(userInput, passInput):
+    if userInput == os.getenv("USERNAME",'') and passInput == os.getenv("PASSWORD",''):
+        return True
+    else:
+        return False