需求说明

实现对Excel工具的自然语言问数，即可以通过界面上传Excel文件，然后在文本框里通过语言对话的形式问出要统计的内容。比如：

用户数有多少？

语文成绩低于90的用户有多少？

.....

实现思路

Pandas AI是基于Pandas库构建的，它扩展了Pandas的功能，使其能够支持生成式AI的任务。Pandas本身是一个用于数据处理和分析的Python库，其强大的数据处理能力也是Pandas AI的基础。这里借助于Pandas AI框架集成Pandas和大模型的能力达到这个目的，详见如下代码。

代码过程

import gradio as gr
import pandas as pd
from pandasai import SmartDataframe
from pandasai.llm.local_llm import LocalLLM


ollama_llm = LocalLLM(api_base="http://127.0.0.1:11434/v1", model="qwen2.5:72b")


def analyze_excel(file_path, user_query):
    """
    使用PandasAI分析Excel文件并回答用户查询
    """
    try:
        # 读取Excel文件
        df = pd.read_excel(file_path)

        # 创建SmartDataframe

        ##smart_df = SmartDataframe(df, config={"llm": llm})
        smart_df = SmartDataframe(df, config={"llm": ollama_llm})

        # 执行查询
        response = smart_df.chat(user_query)

        # 返回响应和原始数据的前5行
        return response, df.head()
    except Exception as e:
        return f"发生错误: {str(e)}", None


# 创建Gradio界面
with gr.Blocks(title="Excel 数据分析助手") as demo:
    gr.Markdown("Excel 数据分析助手")
    gr.Markdown("上传Excel文件，然后用自然语言提问，AI会帮你分析数据！")

    with gr.Row():
        file_input = gr.File(label="上传Excel文件", file_types=[".xlsx", ".xls"])
        query_input = gr.Textbox(label="输入你的问题", placeholder="例如：哪个月的销售额最高？")

    submit_btn = gr.Button("分析")

    with gr.Row():
        output_text = gr.Textbox(label="分析结果", interactive=False)
        output_data = gr.Dataframe(label="数据预览")

    # 示例问题
    examples = [
        #["示例数据.xlsx", "哪个月的销售额最高？"],
        #["示例数据.xlsx", "按产品类别分组并计算平均价格"],
        #["示例数据.xlsx", "绘制销售额随时间变化的趋势图"]
    ]

    # 设置交互
    submit_btn.click(
        fn=analyze_excel,
        inputs=[file_input, query_input],
        outputs=[output_text, output_data]
    )

    gr.Examples(
        examples=examples,
        inputs=[file_input, query_input],
        outputs=[output_text, output_data],
        fn=analyze_excel,
        cache_examples=True
    )

# 启动应用
if __name__ == "__main__":
    demo.launch(server_name="0.0.0.0", server_port=7860)