elise/src/utils/gradio.py

"""
Gradio app utility functions
"""
from typing import List

import pandas as pd


def df_to_json(df_list: List[pd.DataFrame], queries: List[str]):
    """Converts df to json for gradio output"""
    out_dict = {}
    for item, query in zip(df_list, queries):
        temp_list = []
        for row in item.iterrows():
            temp_list.append({"Name": row[0], "Score": row[1].values[0]})
        out_dict[query] = temp_list

    return out_dict