Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import pandas as pd
 from functools import partial
 from ai_classroom_suite.UIBaseComponents import *
-from ai_classroom_suite.MediaVectorStores import get_document_segments
+from ai_classroom_suite.MediaVectorStores import *
 
 # Testing purpose
 class EchoingTutor(SlightlyDelusionalTutor):
@@ -38,14 +38,13 @@ 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, upload_files, openai_auth):
+def create_prompt_store(chat_tutor, vs_button, text_cp, openai_auth):
     text_segs = []
     upload_segs = []
     
-    if upload_files:
-        print(upload_files)
-        upload_fnames = [f.name for f in upload_files]
-        upload_segs = get_document_segments(upload_fnames, 'file', chunk_size=700, chunk_overlap=100)
+    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
@@ -163,7 +162,7 @@ with gr.Blocks() as demo:
                 fn=embed_prompt, inputs=[instructor_prompt], outputs=[test_secret]
             ).then(
                 fn=create_prompt_store, 
-                inputs=[study_tutor, prompt_submit_btn, file_input, api_input],
+                inputs=[study_tutor, prompt_submit_btn, instructor_prompt, api_input],
                 outputs=[study_tutor, prompt_submit_btn]
             )
             # TODO: may need a way to set the secret prompt permanently in settings/secret