Update space

app.py

@@ -1,6 +1,11 @@
 import gradio as gr
+import os
 from huggingface_hub import InferenceClient
 
+# Password authentication setup
+PASSWORD = os.environ.get("APP_PASSWORD", "default_password")  # Get from Space secrets
+authenticated = False  # Track authentication state
+
 """
 For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
 """
@@ -40,25 +45,53 @@ def respond(
         yield response
 
 
-"""
-For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
-"""
-demo = gr.ChatInterface(
-    respond,
-    additional_inputs=[
-        gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-        gr.Slider(
-            minimum=0.1,
-            maximum=1.0,
-            value=0.95,
-            step=0.05,
-            label="Top-p (nucleus sampling)",
-        ),
-    ],
-)
+def check_password(password_attempt):
+    if password_attempt == PASSWORD:
+        return True, "Login successful! Redirecting to chat..."
+    else:
+        return False, "Incorrect password. Please try again."
+
+
+# Authentication interface
+with gr.Blocks() as login_interface:
+    gr.Markdown("## Authentication Required")
+    gr.Markdown("Please enter the password to access this application.")
+    password_input = gr.Textbox(type="password", label="Password")
+    login_button = gr.Button("Login")
+    result = gr.Markdown()
+    
+    # Chat interface (initially hidden)
+    with gr.Accordion("Chat", visible=False) as chat_accordion:
+        chat_interface = gr.ChatInterface(
+            respond,
+            additional_inputs=[
+                gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
+                gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
+                gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
+                gr.Slider(
+                    minimum=0.1,
+                    maximum=1.0,
+                    value=0.95,
+                    step=0.05,
+                    label="Top-p (nucleus sampling)",
+                ),
+            ],
+        )
+    
+    # Login logic
+    def handle_login(password):
+        success, message = check_password(password)
+        if success:
+            return message, gr.update(visible=True), gr.update(visible=False)
+        else:
+            return message, gr.update(visible=False), gr.update(visible=True)
+    
+    login_button.click(
+        handle_login,
+        inputs=[password_input],
+        outputs=[result, chat_accordion, password_input],
+    )
 
 
 if __name__ == "__main__":
-    demo.launch()
+    login_interface.launch()