Update app.py

app.py

@@ -2,15 +2,7 @@ import os
 from threading import Thread
 from typing import Iterator
 import os
-from huggingface_hub import HfApi
-
-# Retrieve the token
-hf_token = os.getenv("HF_AUTH_TOKEN")
-
-# Example: Using the token with Hugging Face Hub
 from huggingface_hub import login
-print(hf_token)
-login(token = hf_token)
 import gradio as gr
 import spaces
 import torch
@@ -117,6 +109,18 @@ chat_interface = gr.ChatInterface(
 
 with gr.Blocks(css_paths="style.css", fill_height=True) as demo:
     gr.DuplicateButton(value="Duplicate Space for private use", elem_id="duplicate-button")
+    def authenticate_token(token):
+        try:
+            # Validate token using Hugging Face Hub API
+            login(token = hf_token)
+            return f"Authenticated successfully"
+        except HfHubHTTPError:
+            return "Invalid token. Please try again."
+
+    # Components
+    token_input = gr.Textbox(label="Hugging Face Access Token", type="password", placeholder="Enter your token here...")
+    auth_button = gr.Button("Authenticate")
+    output = gr.Textbox(label="Output")
     chat_interface.render()
 
 if __name__ == "__main__":