sun00009's picture
Update app.py
33a985a verified
raw
history blame contribute delete
No virus
793 Bytes
import os
import gradio as gr
from pymongo import MongoClient
access_token = os.environ.get('HF_TOKEN')
ui = gr.load("sun000090/gradio_space", hf_token=access_token, src="spaces")
### Fetch MongoDB Authentication
uri = os.environ.get('MONGODB_URI')
client = MongoClient(uri, connect=True)
db = client['LoginDetails']
userLogin = db['LoginDetails']
def check_credentials(email, password):
user = userLogin.find_one({'email': email})
if not user or user['password'] != password:
return 'Incorrect'
return 'Correct'
### Authenticate through space
def environ_auth(username, password):
auth_ = check_credentials(username, password)
if auth_=='Correct':
return True
else:
return False
ui.launch(share=True, auth=environ_auth, show_api=False)