import gradio as gr
import chromadb
import pandas as pd

def greet(name):
    client = chromadb.Client()
    collection = client.create_collection("bolivian-recipes")
    df = pd.read_parquet("hf://datasets/asoria/bolivian-recipes@~parquet/default/other/0000.parquet")
    text_column = "preparation"
    ids = [str(i) for i in range(df.shape[0])]
    documents = df[text_column].to_list()
    metadatas = df.drop(text_column, axis=1).to_dict("records")

    collection.add(ids=ids, documents=documents, metadatas=metadatas)

    results = collection.query(query_texts=["sausage"], n_results=4)
    print(results)
    return "Hello " + name + "!!"

iface = gr.Interface(fn=greet, inputs="text", outputs="text")
iface.launch()