Update app.py

app.py

@@ -3,8 +3,6 @@ import gradio as gr
 import pandas as pd
 from functools import partial
 from ai_classroom_suite.UIBaseComponents import *
-from ai_classroom_suite.MediaVectorStores import *
-from ai_classroom_suite.PromptInteractionBase import *
 
 # Testing purpose
 class EchoingTutor(SlightlyDelusionalTutor):
@@ -38,33 +36,7 @@ def get_tutor_reply(chat_tutor):
 
 def get_conversation_history(chat_tutor):
     return chat_tutor.conversation_memory, chat_tutor
-
-def create_prompt_store(chat_tutor, vs_button, text_cp, openai_auth):
-    text_segs = []
-    upload_segs = []
-    
-    if text_cp.strip():
-        text_segs = get_document_segments(text_cp, 'text', chunk_size=700, chunk_overlap=100)
-        [doc.metadata.update({'source':'text box'}) for doc in text_segs];
     
-    # get the full list of everything
-    all_segs = text_segs + upload_segs
-    print(all_segs)
-
-    # create the vector store and update tutor
-    vs_db, vs_retriever = create_local_vector_store(all_segs, search_kwargs={"k": 2})
-    chat_tutor.vector_store = vs_db
-    chat_tutor.vs_retriever = vs_retriever
-
-    # create the tutor chain
-    if not chat_tutor.api_key_valid or not chat_tutor.openai_auth:
-        chat_tutor = embed_key(openai_auth, chat_tutor)
-    qa_chain = create_tutor_mdl_chain(kind="retrieval_qa", mdl=chat_tutor.chat_llm, retriever = chat_tutor.vs_retriever, return_source_documents=True)
-    chat_tutor.tutor_chain = qa_chain
-
-    # return the store
-    return chat_tutor, gr.update(interactive=True, value='Tutor Initialized!')
-
 
 ### Instructor Interface Helper Functions ###
 def get_instructor_prompt(fileobj):
@@ -137,6 +109,7 @@ with gr.Blocks() as demo:
     with gr.Tab("Instructor Only"):
         # API Authentication functionality
         # Instead of ask students to provide key, the key is now provided by the instructor
+        
         """
         To permanently set the key, go to Settings -> Variables and secrets -> Secrets, 
         then replace OPENAI_API_KEY value with whatever openai key of the instructor.
@@ -148,10 +121,13 @@ with gr.Blocks() as demo:
         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)
-            
-            file_input = gr.File(label="Load a .txt or .py file",
-                            file_types=['.py', '.txt'], type="file",
-                            elem_classes="short-height")
+
+            text_input = gr.TextArea(visible=False)
+            file_input = gr.File(label="Load a .txt or .py file", 
+                                 file_types=['.py', '.txt'], type="file", 
+                                 elem_classes="short-height")
+            instructor_input = gr.TextArea(visible=False)
+            learning_objectives = gr.Textbox(visible=False)
 
             # Verify prompt content
             instructor_prompt = gr.Textbox(label="Verify your prompt content", visible=True)
@@ -162,8 +138,8 @@ with gr.Blocks() as demo:
             prompt_submit_btn.click(
                 fn=embed_prompt, inputs=[instructor_prompt], outputs=[test_secret]
             ).then(
-                fn=create_prompt_store, 
-                inputs=[study_tutor, prompt_submit_btn, instructor_prompt, api_input],
+                fn=create_reference_store, 
+                inputs=[study_tutor, prompt_submit_btn, text_input, file_input, instructor_input, api_input, learning_objectives],
                 outputs=[study_tutor, prompt_submit_btn]
             )
             # TODO: may need a way to set the secret prompt permanently in settings/secret