show table

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import chromadb
 import pandas as pd
+import json
 
 client = chromadb.Client()
 collection = client.create_collection("bolivian-recipes")
@@ -13,24 +14,23 @@ collection.add(ids=ids, documents=documents, metadatas=metadatas)
 
 
 with gr.Blocks() as demo:
-    gr.Markdown(" ## PandasAI demo using datasets library")
-    gr.Markdown(" pandasai library https://github.com/gventuri/pandas-ai")
-    gr.Markdown(" datasets library https://huggingface.co/docs/datasets")
-
-    dataset = gr.Textbox(label="dataset", placeholder="mstz/iris", value="mstz/iris")
-    config = gr.Textbox(label="config", placeholder="iris", value="iris")
-    split = gr.Textbox(label="split", placeholder="train", value="train")
-    prompt = gr.Textbox(label="prompt (str)", placeholder="How many records do I have?. Give me the median of sepal_width. Show me the first 5 rows.")
-    cached_responses_table = gr.DataFrame()
+    gr.Markdown(" ## Chroma demo using datasets server parquet files")
+    query = gr.Textbox(label="query", placeholder="anticucho")
     get_result_button = gr.Button("Submit")
+    cached_responses_table = gr.DataFrame()
+
+    def get_result(query) -> str:
+        result = collection.query(query_texts=[query], n_results=2)
+        ids = result["ids"][0]
+        distances = result["distances"][0]
+        metadatas = [json.dumps(data) for data in result["metadatas"][0]]
+        documents = result["documents"][0]
 
-    def get_result(dataset, config, split, prompt) -> str:
-        result = collection.query(query_texts=[prompt], n_results=4)
         return {
-                    cached_responses_table: gr.update(value=result)
+            cached_responses_table: gr.update(value=pd.DataFrame(data={"ids": ids, "distances":distances, "metadatas": metadatas, "documents":documents})),
         }
 
-    get_result_button.click(get_result, inputs=[dataset, config, split, prompt], outputs=cached_responses_table)
+    get_result_button.click(get_result, inputs=query, outputs=[cached_responses_table])
 
 if __name__ == "__main__":
-    demo.launch(debug=True)
+    demo.launch()