Update app.py

app.py

@@ -1,32 +1,7 @@
-
 import gradio as gr
 import pandas as pd
 from functools import partial
-
-def save_chatbot_dialogue(chat_tutor, save_type):
-
-    formatted_convo = pd.DataFrame(chat_tutor.conversation_memory, columns=['user', 'chatbot'])
-
-    output_fname = f'tutoring_conversation.{save_type}'
-
-    if save_type == 'csv':
-        formatted_convo.to_csv(output_fname, index=False)
-    elif save_type == 'json':
-        formatted_convo.to_json(output_fname, orient='records')
-    elif save_type == 'txt':
-        temp = formatted_convo.apply(lambda x: 'User: {0}\nAI: {1}'.format(x[0], x[1]), axis=1)
-        temp = '\n\n'.join(temp.tolist())
-        with open(output_fname, 'w') as f:
-            f.write(temp)
-    else:
-        gr.update(value=None, visible=False)
-    
-    return gr.update(value=output_fname, visible=True)
-
-save_json = partial(save_chatbot_dialogue, save_type='json')
-save_txt = partial(save_chatbot_dialogue, save_type='txt')
-save_csv = partial(save_chatbot_dialogue, save_type='csv')
-
+from ai_classroom_suite.UIBaseComponents import *
 
 # history is a list of list  
 # [[user_input_str, bot_response_str], ...]