Email logger (#24)

* Added email logger

* added new prompt to buster

* add reset text box on click

app.py

@@ -55,6 +55,21 @@ def log_likes(completion: Completion, like_data: gr.LikeData):
         logger.info("Something went wrong logging")
 
 
+def log_emails(email: gr.Textbox):
+    collection = "email_data-test"
+
+    logger.info(f"User reported {email=}")
+    email_document = {"email": email}
+
+    try:
+        cfg.mongo_db[collection].insert_one(email_document)
+        logger.info("")
+    except:
+        logger.info("Something went wrong logging")
+
+    return ""
+
+
 def format_sources(matched_documents: pd.DataFrame) -> str:
     if len(matched_documents) == 0:
         return ""
@@ -73,7 +88,7 @@ def format_sources(matched_documents: pd.DataFrame) -> str:
     # drop duplicates, keep highest ranking ones
     matched_documents = matched_documents.sort_values(
         "similarity_to_answer", ascending=False
-    ).drop_duplicates("title", keep="first")
+    ).drop_duplicates("title", keep="first")[:5]
 
     # Revert back to correct display
     display_source_to_ui = {
@@ -172,10 +187,19 @@ with demo:
         )
         submit = gr.Button(value="Send", variant="secondary")
 
-    examples = gr.Examples(
-        examples=cfg.example_questions,
-        inputs=question,
-    )
+    with gr.Row():
+        examples = gr.Examples(
+            examples=cfg.example_questions,
+            inputs=question,
+        )
+        with gr.Row():
+            email = gr.Textbox(
+                label="Want to receive updates about our AI tutor?",
+                placeholder="Enter your email here...",
+                lines=1,
+                scale=3,
+            )
+            submit_email = gr.Button(value="Submit", variant="secondary", scale=0)
 
     gr.Markdown(
         "This application uses ChatGPT to search the docs for relevant info and answer questions. "
@@ -193,5 +217,8 @@ with demo:
 
     chatbot.like(log_likes, completion)
 
+    submit_email.click(log_emails, email, email)
+    email.submit(log_emails, email, email)
+
 demo.queue(concurrency_count=CONCURRENCY_COUNT)
 demo.launch(debug=True, share=False)

cfg.py

@@ -106,7 +106,7 @@ Q:
             "Only respond with information inside the json documentation. DO NOT use additional information, even if you know the answer. "
             "If the answer is in the documentation, answer the question (depending on the questions and the variety of relevant information in the json documentation, answer in 5 paragraphs."
             "If the documentation does not discuss the topic related to the question, kindly respond that you cannot answer the question because it is not part of your knowledge. "
-            "Here is the information you can use (json documentation): "
+            "Here is the information you can use (json documentation) in order: "
         ),
         "text_after_docs": (
             "REMEMBER:\n"
@@ -114,7 +114,7 @@ Q:
             "You are provided information found in the json documentation. "
             "Here are the rules you must follow:\n"
             "* Only respond with information inside the json documentation. DO NOT provide additional information, even if you know the answer. "
-            "* If the answer is in the documentation, answer the question (depending on the questions and the variety of relevant information in the json documentation, answer in 5 paragraphs. "
+            "* If the answer is in the documentation, answer the question (depending on the questions and the variety of relevant information in the json documentation. Your answer needs to be pertinent and not redundant giving a clear explanation as if you were a teacher. "
             "* If the documentation does not discuss the topic related to the question, kindly respond that you cannot answer the question because it is not part of your knowledge. "
             "* Only use information summarized from the json documentation, do not respond otherwise. "
             "* Do not refer to the json documentation directly, but use the instructions provided within it to answer questions. "