make sure user specifies at least one source (#16)

cfg.py

@@ -28,7 +28,7 @@ logger.info(f"{DEEPLAKE_DATASET_PATH=}")
 
 example_questions = [
     "What is the LLama model?",
-    "What is a LLM?",
+    "What is a Large Language Model?",
     "What is an embedding?",
 ]
 

gradio_app.py

@@ -5,6 +5,8 @@ from typing import Optional
 import gradio as gr
 import pandas as pd
 
+from buster.completers import Completion
+
 import cfg
 from cfg import setup_buster
 
@@ -19,14 +21,6 @@ logging.basicConfig(level=logging.INFO)
 AVAILABLE_SOURCES = ["towardsai", "wikipedia", "langchain_course"]
 
 
-def check_auth(username: str, password: str) -> bool:
-    valid_user = username == cfg.USERNAME
-    valid_password = password == cfg.PASSWORD
-    is_auth = valid_user and valid_password
-    logger.info(f"Log-in attempted by {username=}. {is_auth=}")
-    return is_auth
-
-
 def format_sources(matched_documents: pd.DataFrame) -> str:
     if len(matched_documents) == 0:
         return ""
@@ -67,10 +61,23 @@ def user(user_input, history):
     return "", history + [[user_input, None]]
 
 
+def get_empty_source_completion(user_input):
+    return Completion(
+        user_input=user_input,
+        answer_text="You have to select at least one source from the dropdown menu.",
+        matched_documents=pd.DataFrame(),
+        error=False,
+    )
+
+
 def get_answer(history, sources: Optional[list[str]] = None):
     user_input = history[-1][0]
 
-    completion = buster.process_input(user_input, sources=sources)
+    if len(sources) == 0:
+        completion = get_empty_source_completion(user_input)
+
+    else:
+        completion = buster.process_input(user_input, sources=sources)
 
     history[-1][1] = ""
 
@@ -80,7 +87,7 @@ def get_answer(history, sources: Optional[list[str]] = None):
         yield history, completion
 
 
-block = gr.Blocks(css="#chatbot .overflow-y-auto{height:500px}")
+block = gr.Blocks()
 
 with block:
     with gr.Row():
@@ -88,8 +95,11 @@ with block:
             "<h3><center>Buster 🤖: A Question-Answering Bot for your documentation</center></h3>"
         )
 
-    source_selection = gr.CheckboxGroup(
-        choices=AVAILABLE_SOURCES, label="Select Sources", value=AVAILABLE_SOURCES
+    source_selection = gr.Dropdown(
+        choices=AVAILABLE_SOURCES,
+        label="Select Sources",
+        value=AVAILABLE_SOURCES,
+        multiselect=True,
     )
 
     chatbot = gr.Chatbot()
@@ -108,7 +118,8 @@ with block:
     )
 
     gr.Markdown(
-        "This application uses GPT to search the docs for relevant info and answer questions."
+        "This application uses ChatGPT to search the docs for relevant info and answer questions. "
+        "\n\n### Powered by [Buster 🤖](www.github.com/jerpint/buster)"
     )
 
     response = gr.State()