Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import gradio as gr
 import pandas as pd
 from functools import partial
@@ -6,12 +7,7 @@ from ai_classroom_suite.UIBaseComponents import *
 # history is a list of list  
 # [[user_input_str, bot_response_str], ...]
 
-class EchoingTutor:
-    # create basic initialization function
-    def __init__(self):
-        self.conversation_memory = []
-        self.flattened_conversation = ''
-
+class EchoingTutor(SlightlyDelusionalTutor):
     def add_user_message(self, user_message):
         self.conversation_memory.append([user_message, None])
         self.flattened_conversation = self.flattened_conversation + '\n\n' + 'User: ' + user_message
@@ -40,38 +36,16 @@ def get_tutor_reply(user_message, chat_tutor):
 
 def get_conversation_history(chat_tutor):
     return chat_tutor.conversation_memory, chat_tutor
+        
 
-
-css="""
-#sources-container {
-  overflow: scroll !important; /* Needs to override default formatting */
-  /*max-height: 20em; */ /* Arbitrary value */
-}
-#sources-container > div { padding-bottom: 1em !important; /* Arbitrary value */ }
-.short-height > * > * { min-height: 0 !important; }
-.translucent { opacity: 0.5; }
-.textbox_label { padding-bottom: .5em; }
-"""
-
-with gr.Blocks(css=css) as demo:
+with gr.Blocks() as demo:
     #initialize tutor (with state)
     study_tutor = gr.State(EchoingTutor())
 
-    # Title
-    gr.Markdown("# Basic Interaction Tutor")
-
-    # API Authentication functionality
-    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)
-        
+    # API Authentication
+    # Instead of ask students to provide key, the key is now directly provided by the instructor
+    embed_key(os.environ.get("OPENAI_API_KEY"), study_tutor)
+    
     # Chatbot interface
     gr.Markdown("""
     ## Chat with the Model