Changed login

app.py

@@ -20,14 +20,6 @@ with demo:
     ## Title
     gr.Markdown('# Chat Quest!', elem_id='gradio_center')
 
-    input_openai_key = gr.Textbox(
-        label='Para jugar necesitás una API key de OpenAI.',
-        placeholder='sk-********************',
-        type='password',
-        interactive=True,
-        autofocus=True
-    )
-
     with gr.Column(variant='panel'):
         ## Header
         with gr.Row(variant='default'):
@@ -49,7 +41,8 @@ with demo:
             )
 
             situation_goal = gr.Markdown(format_goal_messsage(sample_conversation.goal_message))
-            choice_buttons = [gr.Button(e) for e in sample_conversation.starting_choices]
+
+            choice_buttons = [gr.Button(e, min_width=10) for e in sample_conversation.starting_choices]
             
 
         with gr.Column(variant='panel', visible=False) as outcome_won:
@@ -87,14 +80,10 @@ with demo:
         queue=False
     )
 
-    input_openai_key.submit(
-        update_openai_key,
-        inputs=[input_openai_key],
-        outputs=[input_openai_key]
-    )
 
 if __name__ == "__main__":
     demo.launch(
         server_port=None,
+        auth=validate_auth,
         max_threads=4, show_api=False, share=False
     )

utils.py

@@ -1,6 +1,5 @@
 import gradio as gr
 
-import time
 from chat_quest import sample_conversation, sample_conversation_christmas, generate_chat_response, generate_last_response, ConversationSituation, N_CHOICES
 
 
@@ -69,14 +68,12 @@ def load_next_level():
 
     opponent_name = format_chat_header(new_situation.character)
     goal_message = format_goal_messsage(new_situation.goal_message)
-
     
     return choices_update + [new_situation, new_situation.starting_choices, outcome_update, chat_history_update, chat_history_update] + [new_situation.character_emoji, opponent_name, goal_message, False]
 
 
 def process_key(openai_key: str):
     import os
-
     result = openai_key
 
     if openai_key.lower() == os.environ.get('USER'):
@@ -89,18 +86,16 @@ def validate_openai_key(password: str):
     return password.startswith("sk-")
 
 
-def update_openai_key(input_key: str):
+def validate_auth(user: str, pw: str) -> bool:
     import openai
 
-    processed_pw = process_key(input_key)
+    processed_pw = process_key(pw)
     is_valid_login = validate_openai_key(processed_pw)
 
     if is_valid_login:
         openai.api_key = processed_pw
-    else:
-        gr.Warning("Not a valid password")
 
-    return gr.update(value='', visible=not is_valid_login)
+    return is_valid_login
 
 
 def _format_dialogue(dialogue_list, opponent):
@@ -111,4 +106,4 @@ def _format_dialogue(dialogue_list, opponent):
             formatted_dialogue.append(f"{opponent}: {opponent_line}")
         if player_line:
             formatted_dialogue.append(f"Player: {player_line}")
-    return '\n'.join(formatted_dialogue)
+    return '\n'.join(formatted_dialogue)