pr2

demo.py

@@ -2,6 +2,7 @@ from openai import OpenAI, AssistantEventHandler
 from dotenv import load_dotenv
 import os
 import gradio as gr
+import hashlib
 
 # Load environment variables from .env file
 load_dotenv()
@@ -9,6 +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
 
 client = OpenAI(api_key=OPENAI_API_KEY)
 
@@ -16,9 +18,6 @@ def create_thread():
     return client.beta.threads.create()
 
 def predict(user_message, history, thread):
-    # Append the new user message to the history
-    # history.append({"role": "user", "content": user_message})
-
     # Send the user message to the OpenAI API
     client.beta.threads.messages.create(
         thread_id=thread.id,
@@ -42,15 +41,62 @@ def predict(user_message, history, thread):
     history.append({"role": "assistant", "content": ''.join(response)})
     yield history, thread
 
-# Launch the Gradio chat interface
-with gr.Blocks(title="Hair Library Shopping Assistant Demo") as demo:
-    chatbot = gr.Chatbot(type='messages', label="Hair Library Shopping Assistant Demo")
-    msg = gr.Textbox(placeholder="Type your message here...")
-    user_content = gr.State("")
-    thread = gr.State(create_thread)
-    history = gr.State([])
-    submit = gr.Button("Submit")
+def verify_password(password):
+    """Verify password against the stored hash"""
+    if not PASSWORD_HASH:
+        return False, "Password not configured in environment variables."
+    
+    # Hash the input password
+    hashed_input = hashlib.sha256(password.encode()).hexdigest()
+    
+    # Compare with stored hash
+    if hashed_input == PASSWORD_HASH:
+        return True, "Authentication successful"
+    else:
+        return False, "Incorrect password. Please try again."
 
+# Launch the Gradio chat interface
+with gr.Blocks(title="TM Tender Assistant") as demo:
+    # Login state
+    is_authenticated = gr.State(False)
+    
+    # Login component
+    with gr.Group(visible=True) as login_group:
+        gr.Markdown("## TM Tender Assistant - Login")
+        password_input = gr.Textbox(type="password", label="Password", placeholder="Enter password")
+        login_button = gr.Button("Login")
+        login_message = gr.Markdown("")
+    
+    # Chat interface component
+    with gr.Group(visible=False) as chat_group:
+        gr.Markdown("## TM Tender Assistant")
+        chatbot = gr.Chatbot(type='messages', label="TM Tender Assistant")
+        msg = gr.Textbox(placeholder="Type your message here...")
+        user_content = gr.State("")
+        thread = gr.State(create_thread)
+        history = gr.State([])
+        submit = gr.Button("Submit")
+    
+    def login(password):
+        success, message = verify_password(password)
+        if success:
+            return {
+                login_group: gr.update(visible=False),
+                chat_group: gr.update(visible=True),
+                login_message: ""
+            }
+        else:
+            return {
+                login_message: message,
+                password_input: ""
+            }
+    
+    login_button.click(
+        login,
+        inputs=[password_input],
+        outputs=[login_group, chat_group, login_message]
+    )
+    
     def user_input(user_message, history):
         user_content = user_message
         return "", history + [{"role": "user", "content": user_message}], user_content
@@ -62,7 +108,4 @@ with gr.Blocks(title="Hair Library Shopping Assistant Demo") as demo:
         predict, [user_content, chatbot, thread], [chatbot, thread]
     )
 
-    # msg.submit(predict, [msg, chatbot, thread], [chatbot, thread], queue=False)
-    # clear.click(lambda: None, None, chatbot, queue=False)
-
 demo.launch(share=False)

password_generator.py

@@ -0,0 +1,16 @@
+import hashlib
+import getpass
+
+def generate_password_hash():
+    """Generate a SHA-256 hash for a password"""
+    password = getpass.getpass("Enter the password you want to hash: ")
+    if not password:
+        print("Password cannot be empty")
+        return
+    
+    hashed = hashlib.sha256(password.encode()).hexdigest()
+    print("\nAdd the following line to your .env file:")
+    print(f"PASSWORD={hashed}")
+
+if __name__ == "__main__":
+    generate_password_hash()

requirements.txt

@@ -1,3 +1,3 @@
 gradio==5.3.0
-openai==1.52.1
+openai
 python-dotenv==1.0.1