Update app.py

app.py

@@ -1,29 +1,59 @@
 import gradio as gr
+import bcrypt
 
-# Base de datos de usuarios y contraseñas (en este caso, solo un ejemplo simple)
-user_db = {"admin": "admin", "foo": "bar"}
+# Simulate a database with a dictionary
+user_db = {}
 
-# Función para obtener el nombre de usuario desde la cabecera
-def get_user(request):
-    return request.headers.get("user")
-
-# Función de saludo
-def greet(name):
-    return f"Hello {name}!"
-
-# Definimos una interfaz simple de Gradio
-def demo_function(username):
+# Function to register a new user
+def register(username, password):
     if username in user_db:
-        return greet(username)
+        return "Username already exists!"
+    
+    # Hash the password using bcrypt
+    hashed_password = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt())
+    
+    # Store username and hashed password
+    user_db[username] = hashed_password
+    
+    return "Registration successful!"
+
+# Function to login with a username and password
+def login(username, password):
+    if username not in user_db:
+        return "Username not found!"
+    
+    hashed_password = user_db[username]
+    
+    # Check if the entered password matches the stored hashed password
+    if bcrypt.checkpw(password.encode('utf-8'), hashed_password):
+        return f"Welcome back, {username}!"
     else:
-        return "User not authenticated"
-
-# Creamos la interfaz de Gradio
-demo = gr.Interface(fn=demo_function, inputs="textbox", outputs="textbox")
-
-# Añadimos autenticación al inicio del demo
-demo.auth(auth_function=get_user, auth_message="Please provide your user header.")
+        return "Incorrect password!"
+
+# Create Gradio interfaces
+def create_gradio_ui():
+    with gr.Blocks() as demo:
+        with gr.Tab("Register"):
+            gr.Markdown("### User Registration")
+            username = gr.Textbox(label="Username")
+            password = gr.Password(label="Password")
+            register_button = gr.Button("Register")
+            register_output = gr.Textbox(label="Registration Status")
+            register_button.click(register, inputs=[username, password], outputs=register_output)
+        
+        with gr.Tab("Login"):
+            gr.Markdown("### User Login")
+            login_username = gr.Textbox(label="Username")
+            login_password = gr.Password(label="Password")
+            login_button = gr.Button("Login")
+            login_output = gr.Textbox(label="Login Status")
+            login_button.click(login, inputs=[login_username, login_password], outputs=login_output)
+    
+    return demo
+
+# Launch Gradio app
+if __name__ == "__main__":
+    demo = create_gradio_ui()
+    demo.launch()
 
-# Lanzamos la interfaz
-demo.launch()