| import gradio as gr |
| import pandas as pd |
| import random |
| import time |
| from pandasai import PandasAI |
| from pandasai.llm.openai import OpenAI |
| import os |
|
|
| df = pd.DataFrame() |
| OPENAPI_KEY_VAL = "" |
| llm = "" |
| pandas_ai = "" |
|
|
| def process_file(file, OPENAPI_KEY): |
| global df, OPENAPI_KEY_VAL, llm, pandas_ai |
| df = pd.read_csv(file.name) |
| OPENAPI_KEY_VAL = OPENAPI_KEY |
| |
| llm = OpenAI(api_token = OPENAPI_KEY_VAL) |
| pandas_ai = PandasAI(llm) |
| |
| return df.head(2) |
|
|
| with gr.Blocks() as demo: |
| |
| with gr.Row(): |
| file = gr.inputs.File(type="file", label="Upload CSV") |
| key = gr.Textbox(placeholder="Paste OPENAI API KEY") |
| out = gr.DataFrame(type="pandas") |
| |
| btn = gr.Button("Execute") |
| btn.click(fn=process_file, inputs=[file, key], outputs=out) |
| |
| chatbot = gr.Chatbot() |
| msg = gr.Textbox() |
| clear = gr.Button("Clear") |
|
|
| def respond(message, chat_history): |
| bot_message = "Hi" |
| try: |
| bot_message = pandas_ai.run(df, prompt=message) |
| except Exception as e: |
| print(e) |
| bot_message = "I'm sorry, but as an AI language model, I am unable to provide a meaningful response to this question." |
| |
| chat_history.append((message, bot_message)) |
| time.sleep(1) |
| return "", chat_history |
|
|
| msg.submit(respond, [msg, chatbot], [msg, chatbot]) |
| clear.click(lambda: None, None, chatbot, queue=False) |
|
|
| demo.launch() |
