def update_buttons(df_string):

app.py

@@ -18,8 +18,13 @@ def process_file(file):
     # 将 DataFrame 转换为字符串
     df_string = df.to_string()
 
-    # 返回 DataFrame 字符串，以用作聊天机器人的系统提示
-    return df_string
+    return update_buttons(df_string), df_string
+
+def update_buttons(df_string):
+    # 根据上传的文件内容生成问题
+    questions = generate_questions(df_string)
+    return questions[:3]  # 确保返回三个问题
+
 
 def generate_questions(df_string):
     # 使用 OpenAI 生成基于上传数据的问题
@@ -70,6 +75,10 @@ def respond(user_message, df_string_output, chat_history):
     return "", chat_history
 
 
+def on_button_click(button, df_string_output, chat_history):
+    # 当按钮被点击时发送对应的问题
+    return respond(button.text, df_string_output, chat_history)
+
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
@@ -80,8 +89,14 @@ with gr.Blocks() as demo:
 
         with gr.Column():
             df_string_output = gr.Textbox(label="raw data")
-            question_buttons = gr.Container()
 
+            with gr.Group():
+                gr.Markdown("## 常用问题")
+                btn_1 = gr.Button()
+                btn_2 = gr.Button()
+                btn_3 = gr.Button()
+
+            
     send_button.click(
         respond, 
         inputs=[msg, df_string_output, chatbot], 
@@ -89,15 +104,14 @@ with gr.Blocks() as demo:
     )
 
     # file_upload.change(process_file, inputs=file_upload, outputs=df_string_output)
+    file_upload.change(process_file, inputs=file_upload, outputs=[(btn_1, "text"), (btn_2, "text"), (btn_3, "text"), df_string_output])
+
+
+    # 连接按钮点击事件
+    btn_1.click(on_button_click, inputs=[btn_1, df_string_output, chatbot], outputs=[msg, chatbot])
+    btn_2.click(on_button_click, inputs=[btn_2, df_string_output, chatbot], outputs=[msg, chatbot])
+    btn_3.click(on_button_click, inputs=[btn_3, df_string_output, chatbot], outputs=[msg, chatbot])
 
-    file_upload.change(process_file, inputs=file_upload, outputs=[df_string_output, question_buttons])
 
-    # 根据生成的问题创建按钮
-    for question in generate_questions(""):  # 初始时为空
-        question_buttons.append(gr.Button(question, elem_id=question).click(
-            send_question, 
-            inputs=[question, df_string_output, chatbot], 
-            outputs=[msg, chatbot]
-        ))
 
 demo.launch()