Update app.py

app.py

@@ -23,7 +23,6 @@ class EchoingTutor(SlightlyDelusionalTutor):
         self.conversation_memory = []
         self.flattened_conversation = ''
 
-
 ### Chatbot Functions ###
 def add_user_message(user_message, chat_tutor):
   """Display user message and update chat history to include it."""
@@ -37,6 +36,8 @@ def get_tutor_reply(user_message, chat_tutor):
 def get_conversation_history(chat_tutor):
     return chat_tutor.conversation_memory, chat_tutor
 
+
+### Instructor Interface Helper Functions ###
 def get_instructor_prompt(fileobj):
     file_path = fileobj.name
     f = open(file_path, "r")
@@ -47,14 +48,14 @@ def embed_prompt(instructor_prompt):
     os.environ["SECRET_PROMPT"] = instructor_prompt
     return os.environ.get("SECRET_PROMPT")
 
+    
+### User Interfaces ###
 with gr.Blocks() as demo:
+    
     #initialize tutor (with state)
     study_tutor = gr.State(EchoingTutor())
 
-    # Instead of ask students to provide key, the key is now directly provided by the instructor
-    api_input = os.environ.get("OPENAI_API_KEY")
-    embed_key(api_input, study_tutor)
-
+    # Student interface
     with gr.Tab("For Students"): 
         # Chatbot interface
         gr.Markdown("""
@@ -94,8 +95,32 @@ with gr.Blocks() as demo:
         export_dialogue_button_json.click(save_json, study_tutor, file_download, show_progress=True)
         export_dialogue_button_txt.click(save_txt, study_tutor, file_download, show_progress=True)
         export_dialogue_button_csv.click(save_csv, study_tutor, file_download, show_progress=True)
-        
+
+    # Instructor interface
     with gr.Tab("Instructor Only"):
+        # API Authentication functionality
+        # Instead of ask students to provide key, the key is now provided by the instructor
+        with gr.Box():
+            
+            gr.Markdown("### OpenAI API Key ")
+            gr.HTML("""<span>Embed your OpenAI API key below; if you haven't created one already, visit
+            <a href="https://platform.openai.com/account/api-keys">platform.openai.com/account/api-keys</a>
+            to sign up for an account and get your personal API key</span>""",
+                    elem_classes="textbox_label")
+            api_input = gr.Textbox(show_label=False, type="password", container=False, autofocus=True,
+                          placeholder="●●●●●●●●●●●●●●●●●", value='')
+            api_input.submit(fn=embed_key, inputs=[api_input, study_tutor], outputs=study_tutor)
+            api_input.blur(fn=embed_key, inputs=[api_input, study_tutor], outputs=study_tutor)
+
+            """
+            Another way to permanently set the key is to directly go to 
+                Settings -> Variables and secrets -> Secrets
+            Then replace OPENAI_API_KEY value with whatever openai key of the instructor.
+            """
+            # api_input = os.environ.get("OPENAI_API_KEY")
+            # embed_key(api_input, study_tutor)
+
+        # Upload secret prompt functionality
         with gr.Blocks():
             # testing purpose, change visible to False at deployment
             test_secret = gr.Textbox(label="Current secret prompt", value=os.environ.get("SECRET_PROMPT"), visible=True)
@@ -105,8 +130,11 @@ with gr.Blocks() as demo:
                             elem_classes="short-height")
             instructor_prompt = gr.Textbox(label="Verify your prompt content", visible=True)
             file_input.upload(fn=get_instructor_prompt, inputs=file_input, outputs=instructor_prompt)
-            
+
+            # Set the secret prompt in this session
             prompt_submit_btn = gr.Button("Submit")
             prompt_submit_btn.click(fn=embed_prompt, inputs=instructor_prompt, outputs=test_secret)
+
+            # TODO: may need a way to set the secret prompt permanently in settings/secret
     
 demo.queue().launch(server_name='0.0.0.0', server_port=7860)