luca2040 / foglietti-cat

Use the CheshireCat AI to explain and read medicine informations
1 stars 0 forks source link

foglietti-cat

Repository del plugin per il framework CheshireCat per riassumere e dare informazioni a riguardo dei foglietti illustrativi dei medicinali.
La repo è divisa in due parti:

Lo stato attuale del progetto è descritto nella sezione Conclusioni, contenente un riassunto degli obiettivi raggiunti a fine stage e degli eventuali problemi lasciati aperti.
Se si volessero più dettagli si lascia disponibile nella sezione Diario stage 3/06/2024 - 21/06/2024, una descrizione giornaliera delle attività svolte giorno per giorno con i commenti originali.

output


CheshireCatAPI

L'entry point è il file API_upload.py

Funzionamento:

Prima di iniziare a fare l'upload dei file viene aperta una websocket con un nome utente specifico per il caricamento dei file in modo da poter ricevere i callback al completamento del caricamento dei PDF.
Per la websocket si è usata la libreria del gatto:

cheshire_cat_api

Esempio di connessione tramite websocket (API_upload.py):

import cheshire_cat_api as ccat

cat_client = ccat.CatClient(
    config=config,
    on_open=on_open,
    on_close=on_close,
    on_message=on_message,
    on_error=on_error,
)

cat_client.connect_ws()

Successivamente viene inviato un file tramite API del gatto usando la libreria requests. uploadfile.py

import requests

def upload_file(
    filepath: str,
    filename: str,
    url: str,
    stream=None,
    user="user",
    type="text/plain",
) -> any:

    files = {"file": (filename, stream or open(filepath, "rb"), type)}
    data = {"chunk_size": "", "chunk_overlap": ""}

    headers = {"accept": "application/json", "user_id": user}

    response = requests.post(url, headers=headers, files=files, data=data)

    return response

Per ricevere la callback del caricamento completo del file si usa la funzione on_message (websocket_functions.py) per ricevere il testo inviato dal gatto (Purtroppo non è strutturato).

def on_message(message: str) -> None:
    json_message = json.loads(message)

    if json_message["type"] == "notification":
        content_message: str = json_message["content"]

        log(content_message, "INFO")

        if content_message.startswith(config.rabbit_finished_message):
            config.message_called()

Appena il file viene caricato si continua per ogni file presenti nella cartella di input.

Configurazione:

Per poter configurare l'upload è presente il file (config.json)[/CheshireCatAPI/config.json].

{
    "websocket": {
        "base_url": "localhost",
        "port": 1865,
        "user_id": "upload_websocket",
        "auth_key": "",
        "secure_connection": 0
    },
    "rabbithole": {
        "url": "http://localhost:1865/rabbithole/",
        "finished_message": "Finished reading "
    },
    "folder": "files/",
    "log_file": "log.txt"
}

Websocket:

CheshireCat - docker compose

Per la libreria img2table usata per fare il parsing di tabelle nei PDF servono queste dipendenze, quindi si è usato questo comando nel container del gatto:

    command:
      - /bin/bash
      - -c
      - |
        bash -c "apt update && apt install -y default-jre && apt install -y tesseract-ocr && apt install -y libtesseract-dev"
        bash -c "cd /usr/share/tesseract-ocr/4.00/tessdata && curl -H 'Accept: application/vnd.github.v3.raw' -O -L https://api.github.com/repos/tesseract-ocr/tessdata/contents/ita.traineddata"
        bash -c "python3 -m cat.main"

Per poter usare Qdrant serve:

CheshireCat - plugin

Durante la fase di test era necessario poter specificare il nome della medicina selezionata e poter ricevere la lista di tutti i file parsati. Per fare questo sono stati creati due tool:

plugin.py

@tool(return_direct=True)
def how_many_medicines_known(tool_input, cat):
    """Reply only to the question "How many medicines you know?" or to others which are stricly similar.
    Ignore generic questions about medicines.
    Input is always None
    Use this tool only if the user specifies to use a tool using #use tool#"""

@tool(return_direct=True)
def filename(tool_input: str, cat):
    """Reply to a file name, ONLY if specified using the word "file"
    Input is the file name
    Use this tool only if the user specifies to use a tool using #use tool#"""

Per poter parsare correttamente i PDF si è dovuta creare una nuova classe, la quale parsa i PDF in due passaggi:


Per filtrare solo i point con metadata corretto (Quelli che provengono da uno specifico foglietto illustrativo) si è usato l'hook before_cat_recalls_declarative_memories

plugin.py

@hook
def before_cat_recalls_declarative_memories(declarative_recall_config, cat):
    # medicine is the filename or the tag
    declarative_recall_config["metadata"] = {"source": medicine}

    return declarative_recall_config

Se si vuole che il gatto risponda usando certe parole invece che altre, esse si possono specificare nel file wordlist.txt, che è la lista di parole che si vuole che il gatto usi.
Il file viene inserito direttamente nel prefix, quindi è possibile specificare qualsiasi parola.

plugin.py

@hook
def agent_prompt_prefix(prefix, cat):
    with open("/app/cat/wordlist/wordlist.txt", "r") as file:
        wordlist = file.read()

    prefix = f"""
    ........
    Nelle tue risposte usa maggiormente queste parole come sinonimi ad altre.
    {wordlist}
    ........
    """

    return prefix