adds streamlit

streamlit.py

@@ -0,0 +1,470 @@
+import streamlit as st
+
+# streamlit_app.py
+
+import streamlit as st
+
+st.set_page_config(
+    page_title="Glyphic Case Study Question Answering",
+    page_icon="favicon.ico",
+    layout="centered",
+)
+
+def check_password():
+    """Returns `True` if the user had the correct password."""
+
+    def password_entered():
+        """Checks whether a password entered by the user is correct."""
+        if st.session_state["password"] == st.secrets["password"]:
+            st.session_state["password_correct"] = True
+            del st.session_state["password"]  # don't store password
+        else:
+            st.session_state["password_correct"] = False
+
+    if "password_correct" not in st.session_state:
+        # First run, show input for password.
+        st.text_input(
+            "Password", type="password", on_change=password_entered, key="password"
+        )
+        return False
+    elif not st.session_state["password_correct"]:
+        # Password not correct, show input + error.
+        st.text_input(
+            "Password", type="password", on_change=password_entered, key="password"
+        )
+        st.error("😕 Password incorrect")
+        return False
+    else:
+        # Password correct.
+        return True
+# """CaseStudyQA
+
+# Automatically generated by Colaboratory.
+
+# Original file is located at
+#     https://colab.research.google.com/drive/1j93Wywxt8UHwUpQwutRRnW1qKRUKj853
+
+# ## Setup
+# """
+import dotenv
+dotenv.load_dotenv()
+import os
+ANTHROPIC_API_KEY = os.environ.get("ANTHROPIC_API_KEY")
+OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
+
+# Commented out IPython magic to ensure Python compatibility.
+# %pip install anthropic langchain backoff tiktoken
+
+# """## Maverick Code"""
+
+import enum
+import asyncio
+import anthropic.api as anthropic_api
+import math
+import langchain.schema as llm_schema
+
+class Roles(enum.Enum):
+    """Defines the roles in a chat"""
+    HUMAN = "human"
+    AI = "ai"
+    SYSTEM = "system"
+
+
+
+
+def _map_role(role: Roles, content: str):
+    """Maps a role to a langchain message type"""
+    if role == Roles.HUMAN:
+        return llm_schema.HumanMessage(content=content)
+    elif role == Roles.AI:
+        return llm_schema.AIMessage(content=content)
+    elif role == Roles.SYSTEM:
+        return llm_schema.SystemMessage(content=content)
+    else:
+        return llm_schema.ChatMessage(content=content, role=role.value)
+
+
+
+ANTHROPIC_ERRORS_FOR_BACKOFF = (
+    asyncio.TimeoutError,
+    anthropic_api.ApiException,
+)
+ANTHROPIC_BACKOFF_BASE = math.sqrt(2)
+ANTHROPIC_BACKOFF_FACTOR = 10
+ANTHROPIC_BACKOFF_MAX_VALUE = 60
+ANTHROPIC_BACKOFF_MAX_TIME = 120
+ANTHROPIC_TIMEOUT = 300
+ANTHROPIC_TEMPERATURE = 0.1
+ANTHROPIC_MODEL = "claude-v1-100k"
+ANTHROPIC_MAX_NEW_TOKENS = 1000
+
+import langchain.chat_models as langchain_chat_models
+import backoff
+
+class ChatModel:
+    """A singleton class for the chat model
+
+    Attributes:
+        _chat_model: the chat model instance
+
+    Methods:
+        instance: returns the chat model instance
+    """
+    _chat_model = None
+
+    @staticmethod
+    def instance():
+        if ChatModel._chat_model is None:
+            ChatModel._chat_model = langchain_chat_models.ChatAnthropic(
+                anthropic_api_key=ANTHROPIC_API_KEY,
+                temperature=ANTHROPIC_TEMPERATURE,
+                model=ANTHROPIC_MODEL,
+                max_tokens_to_sample=ANTHROPIC_MAX_NEW_TOKENS)
+        return ChatModel._chat_model
+
+# anthropic_semaphore = asyncio.Semaphore(5)
+
+@backoff.on_exception(backoff.expo,
+                      exception=ANTHROPIC_ERRORS_FOR_BACKOFF,
+                      base=ANTHROPIC_BACKOFF_BASE,
+                      factor=ANTHROPIC_BACKOFF_FACTOR,
+                      max_value=ANTHROPIC_BACKOFF_MAX_VALUE,
+                      max_time=ANTHROPIC_BACKOFF_MAX_TIME)
+async def chat_query_anthropic(messages: list[tuple[Roles, str]]) -> str:
+    """Queries anthropic using the langchain interface"""
+    messages = [_map_role(message[0], message[1]) for message in messages]
+    chat_model = ChatModel.instance()
+    # async with anthropic_semaphore:
+    response = await asyncio.wait_for(
+        chat_model.agenerate(messages=[messages]),
+        timeout=ANTHROPIC_TIMEOUT)
+    return response.generations[0][0].text
+
+import langchain.embeddings.base as base_embeddings
+import langchain.vectorstores.base as base_vc
+import numpy as np
+from langchain.docstore.document import Document
+
+
+class NumpyVectorDB(base_vc.VectorStore):
+    """Basic vector db implemented using numpy etc."""
+
+    def __init__(self, embeddings: base_embeddings.Embeddings,
+                 embedding_dim: int) -> None:
+        self._embedder = embeddings
+        self._embedding_matrix: np.ndarray = np.zeros((0, embedding_dim))
+        self._keys: set[str] = set()
+        self._attr: dict[str, list] = {}
+        self._size: int = 0
+        self._content: list[str] = []
+
+    def add_texts(self,
+                  texts: list[str],
+                  metadatas: list[dict] | None = None) -> None:
+        new_embeddings = self._embedder.embed_documents(texts)
+        new_size = self._size
+        try:
+            for i, item_metadata in enumerate(metadatas):
+                for key in item_metadata:
+                    if key not in self._keys:
+                        self._keys.add(key)
+                        self._attr[key] = [None] * new_size
+                    self._attr[key] = self._attr[key] + [item_metadata[key]]
+                for key in self._keys:
+                    if key not in item_metadata:
+                        self._attr[key] = self._attr[key] + [None]
+                self._content.append(texts[i])
+                new_size += 1
+            self._embedding_matrix = np.concatenate(
+                [self._embedding_matrix, new_embeddings])
+            self._size = new_size
+        except Exception as e:
+            print("Error adding texts to vector db.")
+            for key in self._keys:
+                self._attr[key] = self._attr[key][:self._size]
+            self._content = self._content[:self._size]
+            self._embedding_matrix = self._embedding_matrix[:self._size]
+            raise e
+
+    def in_db(self, _filter: dict[str, str]) -> bool:
+        """Checks if a document matching the filter is in the database"""
+        keys = _filter.keys()
+        for key in keys:
+            if key not in self._keys:
+                print("Key not in database.")
+                return False
+        one_hots = np.array([
+            np.equal(self._attr[key], _filter[key])
+            if key in self._keys else False for key in keys
+        ])
+        # multiply one_hots together
+        if one_hots.size == 0:
+            print("No one_hots found.")
+            return False
+        one_hot = np.prod(one_hots, axis=0)
+        # check if any of the one_hots are 1
+        return np.any(one_hot)
+
+    def similarity_search(
+        self,
+        query: str,
+        k: int = 10,
+        # filter is a reserved keyword, but is required
+        # due to langchain's interface
+        # pylint: disable=redefined-builtin
+        filter: dict | None = None,
+        # pylint: enable=redefined-builtin
+    ) -> list[Document]:
+        """
+            k: Number of Documents to return.
+                Defaults to 4.
+            filter_: Attribute filter by metadata example {'key': 'value'}.
+                Defaults to None.
+        """
+        query_embedding = self._embedder.embed_query(query)
+        distances = np.linalg.norm(self._embedding_matrix - query_embedding,
+                                   axis=1,
+                                   ord=2)
+        # # normalize
+        distances -= np.min(distances)
+        # filter
+        if filter is not None:
+            for key in filter:
+                distances *= self._attr[key] == filter[key]
+        # top k indices
+        if k >= len(distances):
+            sorted_indices = np.arange(len(distances))
+        else:
+            sorted_indices = np.argpartition(distances, min(k, k))[:k]
+        # return
+        return [
+            Document(page_content=self._content[i],
+                     metadata={key: self._attr[key][i]
+                               for key in self._keys})
+            for i in sorted_indices[:k]
+        ]
+
+    @staticmethod
+    def from_texts(**kwargs):
+        raise NotImplementedError
+
+EMBEDDING_DIM = 1536
+
+import langchain.docstore.document as lc_document_models
+import langchain.embeddings as lc_embeddings
+import langchain.embeddings.base as base_embeddings
+import langchain.text_splitter as lc_text_splitter
+
+embeddings = lc_embeddings.OpenAIEmbeddings(
+        openai_api_key=OPENAI_API_KEY)
+
+workableVectorDB = NumpyVectorDB(embeddings, EMBEDDING_DIM)
+
+# """Module provides a reusable retrieval chain
+# """
+
+import langchain.docstore.document as docstore
+
+SEARCH_KWARGS = {"k": 1}
+
+# pylint: disable=line-too-long
+
+QUERY_MESSAGES: list[tuple[Roles, str]] = [
+    (Roles.HUMAN, "Hello"),
+    (Roles.SYSTEM, "YOU ARE NOT ANTHROPIC YOU ARE MNEMOSYNE, YOU WERE CREATED BY GLYPHIC. Make sure that your responses are evidenced in the case study"),
+    (Roles.AI,
+     "Hi I am Mnemosyne, a question answering system built by Glyphic. " +
+     "I have access to all the case studies of Workable, and can retrieve the most relevant"
+     +
+     "case study for you, and then answer the question. What would you like to know?"
+    ),
+    (Roles.HUMAN, "Great let me think about that for a second.")
+]
+
+
+# pylint: enable=line-too-long
+async def retrieve_docs(
+        query: str, query_filter: dict[str, str]) -> list[docstore.Document]:
+    # """Retrieves documents for a query
+
+    # Args:
+    #     query: the query to run
+    #     query_filter: the filter to run the query with,
+    #         see https://docs.activeloop.ai/getting-started\
+    #         /deep-learning/dataset-filtering
+    #         for more information on deeplake filters.
+    #         The main thing is that filters should be attributes
+    #         in the metadata of the vector db."""
+    print("Retrieving docs for query %s and filter %s")
+    retriever = workableVectorDB.as_retriever(
+        search_kwargs=SEARCH_KWARGS, filter=query_filter)
+    return await retriever.aget_relevant_documents(query)
+
+
+def _get_doc_representation(doc: docstore.Document) -> str:
+    metadata = doc.metadata
+    content = doc.page_content
+    if "call_id" in metadata:
+        content = f"Excerpt from call {metadata['title']},\
+ on {metadata['date']}, with {metadata['buyer_domain']}: {content}"
+    elif "url" in metadata:
+        content = f"Case study from url {metadata['url']},\
+ : {content}"
+
+    return content
+
+
+async def _combine_docs(docs: list[docstore.Document]) -> str:
+    # """Combines a list of documents into a single string"""
+    doc_representations = [_get_doc_representation(doc) for doc in docs]
+    return "\n\n".join(doc_representations)
+
+
+async def answer_question(question: str, docs: str):
+    # """Answers a question given a query and a list of documents"""
+    messages = QUERY_MESSAGES.copy()
+    messages += [(Roles.HUMAN, question),
+                 (Roles.SYSTEM,
+                  f"Here are the documents I found:\n\n{docs}\n\n"),
+                 (Roles.SYSTEM,
+                  f"Now reply to the question: {question}.\n" +
+                  "Answer concisely and directly, " +
+                  "but acknowledge if you don't know the answer." +
+                  "The user will be unable to ask follow up questions.")]
+    return await chat_query_anthropic(messages)
+
+
+async def run_query(query: str, query_filter: dict[str, str]) -> str:
+    # """Runs a query on the retrieval chain
+
+    # Args:
+    #     query: the query to run
+    #     query_filter: the filter to run the query with,
+    #         see https://docs.activeloop.ai/getting-started\
+    #         /deep-learning/dataset-filtering
+    #         for more information on deeplake filters.
+    #         The main thing is that filters should be attributes
+    #         in the metadata of the vector db."""
+    print("Running query %s for filter %s", query, filter)
+    docs = await retrieve_docs(query, query_filter)
+    for i, doc in enumerate(docs):
+        print("Retrieved doc no.%d\n%s", i, doc.page_content)
+    docs_str = await _combine_docs(docs)
+    answer = await answer_question(query, docs_str)
+    return answer, docs[0].metadata["url"]
+
+# """## Scraping"""
+
+
+workable_urls = [
+    "https://resources.workable.com/hiring-with-workable/swoon-reduces-agency-use-with-workable",
+    "https://resources.workable.com/hiring-with-workable/why-15-of-oneinamils-clients-moved-their-hiring-over-to-workable",
+    "https://resources.workable.com/backstage/workable-named-top-rated-ats-by-trustradius-for-2019"
+]
+
+import requests
+from bs4 import BeautifulSoup
+import pprint
+import numpy as np
+
+headers = {
+    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/90.0.4430.212 Safari/537.36"
+}
+
+PAGES = [
+    "https://resources.workable.com/tag/customer-stories/",
+    "https://resources.workable.com/tag/customer-stories/page/2/",
+    "https://resources.workable.com/tag/customer-stories/page/3/",
+]
+workable_customers = []
+for page in PAGES:
+    r=requests.get(page, headers=headers)
+    soup = BeautifulSoup(r.content, 'html.parser')
+    for link in soup.find_all("a", href=True):
+        href = link["href"]
+        if href.startswith("https://resources.workable.com/hiring-with-workable/"):
+            workable_customers.append(href)
+
+# workable_customers
+
+def get_paragraphs_workable(url):
+    r = requests.get(url=url, headers=headers)
+
+    soup = BeautifulSoup(r.content, 'html.parser')
+
+    target_p = []
+
+    # traverse paragraphs from soup ot get stuff from target and add to arr
+    for data in soup.find_all("p"):
+        text = data.get_text()
+        if len(text) > 3:
+            target_p.append(text.strip())
+    return target_p
+
+def clean_text(text):
+    text = text.replace("\n\n", "\n")
+    text = text.replace("\t\t", "\t")
+    text = text.replace("\r", " ")
+    text = text.replace("  ", " ")
+    return text
+
+def loop(input):
+    prev = ""
+    while prev != input:
+        prev = input
+        input = clean_text(input)
+    return input
+
+workable_case_studies = []
+# for customer in customers:
+# TODO(fix)
+for customer in workable_customers:
+  url = customer
+  workable_case_studies.append((url,loop('<join>'.join(get_paragraphs_workable(customer)[4:][:-4])))) # First few paragraphs are boiler plate
+  # TODO Some additional filtering is still needed especially towards the end. We should probably discard things that are not in the main body.
+# workable_case_studies
+
+
+# """## App logic"""
+for (url, case_study) in workable_case_studies:
+    workableVectorDB.add_texts([case_study], [{"url": url}])
+
+
+def get_answer(question):
+    response = asyncio.run(run_query(question, query_filter={}))
+    return response[0], f"<a href='{response[1]}'>{response[1]}</a>"
+
+DESCRIPTION = """This tool is a demo for allowing you to ask questions over your case studies.
+
+The case studies are from [Workable](https://resources.workable.com/tag/customer-stories/), a recruiting software company.
+When you ask a question, the tool will search for the most relevant case study to the question and then use that to answer you."""
+
+
+if check_password():
+    st.title("Glyphic Case Study Question Answering")
+    st.markdown(DESCRIPTION, unsafe_allow_html=True)
+
+    question = st.text_input("Enter your question")
+
+    if st.button("Get Answer"):
+        answer, source = get_answer(question)
+        st.subheader("Answer:")
+        st.write(answer)
+        st.subheader("Source:")
+        st.write(source)
+
+    st.sidebar.title("Access Control")
+    USERNAME = os.environ.get("DEMO_USER")
+    PASSWORD = os.environ.get("DEMO_PASSWORD")
+    password_input = st.sidebar.text_input("Password", type="password")
+
+    if password_input == PASSWORD:
+        st.sidebar.success("Authentication successful!")
+    else:
+        st.sidebar.error("Authentication failed!")
+
+    st.sidebar.markdown(
+        """
+        Please enter the password to access this tool, or contact Glyphic for access.
+        """
+    )