First test

.python-version

@@ -0,0 +1 @@
+3.9.15

app.py

@@ -1,7 +1,35 @@
 import gradio as gr
+from pandas import DataFrame
+from pandasai import PandasAI
+from pandasai.llm.starcoder import Starcoder
+from datasets import load_dataset
 
-def greet(name):
-    return "Hello " + name + "!!"
+def send_prompt(token: str, df: DataFrame, prompt: str) -> str:
+    llm = Starcoder(api_token=token)
+    pandas_ai = PandasAI(llm, conversational=False)
+    return pandas_ai(df, prompt=prompt) 
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+def get_result(token, dataset, config, split, prompt) -> str:
+    try:
+        dataset = load_dataset(dataset, config, split=split)
+        df = dataset.to_pandas()
+        return send_prompt(token, df, prompt=prompt)
+    except Exception as e:
+        return str(e)
+
+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/index")
+
+    hf_token = gr.Textbox(label="hugging face token", placeholder="hf_xxx", type="password")
+    dataset = gr.Textbox(label="dataset", placeholder="mstz/iris")
+    config = gr.Textbox(label="config", placeholder="iris")
+    split = gr.Textbox(label="split", placeholder="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.")
+    output = gr.Textbox(label="Output Box")
+    get_result_button = gr.Button("Submit")
+    get_result_button.click(get_result, inputs=[hf_token, dataset, config, split, prompt], outputs=output)
+
+if __name__ == "__main__":
+    demo.launch()

packages.txt

@@ -0,0 +1,2 @@
+pandasai
+datasets

poetry.toml

@@ -0,0 +1,3 @@
+[virtualenvs]
+in-project = true
+prefer-active-python = true

pyproject.toml

@@ -0,0 +1,15 @@
+[tool.poetry]
+name = "pandasai-datasets-demo"
+version = "0.1.0"
+description = "Demo for pandasai using datasets library"
+authors = ["Andrea Soria <[email protected]>"]
+
+[tool.poetry.dependencies]
+gradio = "3.28.1"
+python = "^3.9.15"
+pandasai = "^0.2.14"
+datasets = "^2.12.0"
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"