Update app.py

app.py

@@ -4,10 +4,10 @@ from openai import OpenAI
 from typing import List, Tuple
 
 ENDPOINT_URL = "https://api.hyperbolic.xyz/v1"
-
 OAI_API_KEY = os.getenv('HYPERBOLIC_XYZ_KEY')
+PASSWORD = os.getenv("PASSWD")  # Store the password in an environment variable
 
-client = OpenAI(base_url=ENDPOINT_URL,api_key=OAI_API_KEY)
+client = OpenAI(base_url=ENDPOINT_URL, api_key=OAI_API_KEY)
 
 def respond(
     message: str,
@@ -17,19 +17,14 @@ def respond(
     temperature: float,
     top_p: float,
 ):
-    # Prepare the conversation history
     messages = [{"role": "system", "content": system_message}]
-
     for user_msg, assistant_msg in history:
         if user_msg:
             messages.append({"role": "user", "content": user_msg})
         if assistant_msg:
             messages.append({"role": "assistant", "content": assistant_msg})
-
-    # Add the latest user message
     messages.append({"role": "user", "content": message})
 
-    # Stream the response from OpenAI
     response = ""
     for chunk in client.chat.completions.create(
         model="deepseek-ai/DeepSeek-V3",
@@ -43,24 +38,34 @@ def respond(
         response += token
         yield response
 
+def check_password(input_password):
+    if input_password == PASSWORD:
+        return gr.update(visible=False), gr.update(visible=True)
+    else:
+        return gr.update(value="", interactive=True), gr.update(visible=False)
+
+with gr.Blocks() as demo:
+    with gr.Column():
+        password_input = gr.Textbox(
+            type="password", label="Enter Password", interactive=True
+        )
+        submit_button = gr.Button("Submit")
+        error_message = gr.Textbox(
+            label="Error", visible=False, interactive=False
+        )
 
-# Gradio ChatInterface with additional inputs for customization
-demo = gr.ChatInterface(
-    respond,
-    additional_inputs=[
-        gr.Textbox(value="You are a helpful assistant.", 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)",
-        ),
-    ],
-)
+    with gr.Column(visible=False) as chat_interface:
+        chat = gr.ChatInterface(
+            respond,
+            additional_inputs=[
+                gr.Textbox(value="You are a helpful assistant.", 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)"),
+            ],
+        )
 
+    submit_button.click(check_password, inputs=password_input, outputs=[password_input, chat_interface])
 
 if __name__ == "__main__":
-    demo.launch(share=True,auth=("user",os.getenv("PASSWD")))
+    demo.launch(share=True)