Update password handling and improve login flow

demo.py

@@ -10,7 +10,7 @@ load_dotenv()
 # Load env variables
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 ASSISTANT_ID = os.getenv("ASSISTANT_ID")
-PASSWORD_HASH = os.getenv("PASSWORD")  # Stored password hash in .env
+PASSWORD_HASH = os.getenv("PASSWORD_HASH")  # Stored password hash in .env
 
 client = OpenAI(api_key=OPENAI_API_KEY)
 
@@ -48,7 +48,7 @@ def verify_password(password):
     
     # Hash the input password
     hashed_input = hashlib.sha256(password.encode()).hexdigest()
-    
+    print(hashed_input)
     # Compare with stored hash
     if hashed_input == PASSWORD_HASH:
         return True, "Authentication successful"
@@ -81,14 +81,16 @@ with gr.Blocks(title="TM Tender Assistant") as demo:
         success, message = verify_password(password)
         if success:
             return {
-                login_group: gr.update(visible=False),
-                chat_group: gr.update(visible=True),
+                login_group: gr.Group(visible=False),
+                chat_group: gr.Group(visible=True),
                 login_message: ""
             }
         else:
+            print("Fail")
             return {
-                login_message: message,
-                password_input: ""
+                login_group: gr.Group(visible=True),
+                chat_group: gr.Group(visible=False),
+                login_message: ""
             }
     
     login_button.click(
@@ -96,6 +98,12 @@ with gr.Blocks(title="TM Tender Assistant") as demo:
         inputs=[password_input],
         outputs=[login_group, chat_group, login_message]
     )
+
+    password_input.submit(
+        login,
+        inputs=[password_input],
+        outputs=[login_group, chat_group, login_message]
+    )
     
     def user_input(user_message, history):
         user_content = user_message