added more questions

database/mock_qna.sqlite

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:66df2080ffb3456c39b4bf554effc49266e957e63792ce12a4f11cb991f369fd
-size 20480
+oid sha256:d2a27be7e05ba8545ed59d9f364028d86130cfb2460c74fa71fa16dd3b214e30
+size 40960

database/mock_qna_source.csv

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c80d88333c3b9fb2a700d49113d2ba3fef7cc671c11b640168c389bef411bc05
-size 7624
+oid sha256:0794f608da3a328f2d089cdcfa3c69a9e4b9fb60911dd16492de014fd45c4671
+size 25530

notebooks/003_create_mock_qna.ipynb

@@ -77,6 +77,16 @@
     "qna_data[:3]"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "3f804c2a-6f64-4af2-a030-7fac1248835b",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "qna_data[\"chapter\"].value_counts()"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,

raw_documents/qna.txt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:59cc1b620ccad1393fc4311e91e538774ac76149a42bd2391af7c855895d80bc
-size 56746
+oid sha256:da72ca56312ecb78d7cf6c9288b16a520baa2286136b4677cf09f36ee4f07b36
+size 56792

streamlit_app.py

@@ -5,6 +5,7 @@ import os
 import pandas as pd
 import base64
 from io import BytesIO
+import sqlite3
 
 import chromadb
 from llama_index.core import (
@@ -28,7 +29,7 @@ import nest_asyncio
 nest_asyncio.apply()
 
 # App title
-st.set_page_config(page_title="🐻🍯 Study Bear")
+st.set_page_config(page_title="🐻📚 Study Bear 🍯")
 openai_api = os.getenv("OPENAI_API_KEY")
 
 # "./raw_documents/HI_Knowledge_Base.pdf"
@@ -38,6 +39,7 @@ input_files = ["./raw_documents/HI Chapter Summary Version 1.3.pdf",
 embedding_model = "BAAI/bge-small-en-v1.5"
 persisted_vector_db = "./models/chroma_db"
 fine_tuned_path = "local:models/fine-tuned-embeddings"
+questionaire_db_path = "./database/mock_qna.sqlite"
 system_content = (
                     "You are a helpful study assistant. "
                     "You do not respond as 'User' or pretend to be 'User'. "
@@ -88,9 +90,10 @@ introduction_line = (
                     "- Etc ... \n"
                     "  \n"
                     "For examples, you could ask me \n"
+                    "- *How many modules I have to take to become an insurance agent in Singapore?*  \n"
                     "- *How many chapters are there in textbook 'Health Insurance 7th Edition'?*  \n"
                     "- *Can you list all the chapters by name and its number for me?*  \n"
-                    "- *Please extract the important key concept from chapter 1 into 10 bullet points*  \n"
+                    "- *Please extract the important key concept from chapter 1 - overview of healthcare environment in singapore, into 10 bullet points*  \n"
                     "- *Please ask me a question so that I can tell if I have enough understanding about Chapter 2*  \n"
 )
 
@@ -158,6 +161,7 @@ if "qna_answer" not in st.session_state:
     st.session_state.qna_answer = None
 
 def clear_chat_history():
+
     st.session_state.messages = [{"role": "assistant", 
                                   "content": introduction_line,
                                   "type": "text"}]
@@ -169,8 +173,20 @@ def clear_chat_history():
                                    system_content=system_content,
                                    persisted_vector_db=persisted_vector_db)
     chat_engine.reset()
+    st.toast("yumyum, what was I saying again? 🐻💬", icon="🍯")
+
+def clear_question_history():
+
+    con = sqlite3.connect(questionaire_db_path)
+    cur = con.cursor()
+    sql_string = "DELETE FROM answer_tbl"
+    res = cur.execute(sql_string)
+    con.commit()
+    con.close()
+    st.toast("the tale of one thousand and one questions, reset! 🧨🧨", icon="📝")
 
 st.sidebar.button("Clear Chat History", on_click=clear_chat_history)
+st.sidebar.button("Clear Question History", on_click=clear_question_history)
 if st.sidebar.button("I want to submit a feedback!"):
     st.session_state.feedback = True
     st.session_state.feedback_key += 1  # overwrite feedback component