chatbot

app.py

@@ -1,7 +1,14 @@
 import gradio as gr
 import pandas as pd
+import os
+from openai import OPENAI
 
-def process_file(file, question):
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+client = OpenAI(api_key=OPENAI_API_KEY)
+
+
+
+def process_file(file):
     # 读取文件
     if file.name.endswith('.csv'):
         df = pd.read_csv(file)
@@ -11,13 +18,71 @@ def process_file(file, question):
     # 将 DataFrame 转换为字符串
     df_string = df.to_string()
 
-    # 返回 DataFrame 字符串和用户问题
-    return df_string + "\nQ: " + question
+    # 返回 DataFrame 字符串，以用作聊天机器人的系统提示
+    return df_string
+
+def respond(user_message, chat_history):
+    print("=== 變數：user_message ===")
+    print(user_message)
+    print("=== 變數：chat_history ===")
+    print(chat_history)
+
+    sys_content = f"你是一個資料分析師，請用 {df_string_output} 為資料進行對話"
+    messages = [
+        {"role": "system", "content": sys_content},
+        {"role": "user", "content": user_message}
+    ]
+
+    print("=====messages=====")
+    print(messages)
+    print("=====messages=====")
+
+
+    request_payload = {
+        "model": "gpt-4-1106-preview",
+        "messages": messages,
+        "max_tokens": 2000  # 設定一個較大的值，可根據需要調整
+    }
+
+    response = client.chat.completions.create(**request_payload)
+    print(response)
+    
+    response_text = response.choices[0].message.content.strip()
+
+    # 更新聊天历史
+    new_chat_history = (user_message, response_text)
+    if chat_history is None:
+        chat_history = [new_chat_history]
+    else:
+        chat_history.append(new_chat_history)
+
+    # 返回聊天历史和空字符串清空输入框
+    return "", chat_history
+
+
+with gr.Blocks() as demo:
+    with gr.Row():
+        file_upload = gr.File(label="Upload your file")
+        df_string_output = gr.Textbox(label="DataFrame as String", visible=False)
+
+    with gr.Row():
+        chatbot = gr.Chatbot()
+    
+    with gr.Row():
+        with gr.Column():
+            with gr.Group():
+                # password_input = gr.Textbox(label="Password", type="password")
+                msg = gr.Textbox(label="請输入對話內容")
+                send_button = gr.Button("發送")
+
+    # 当文件上传时，更新 DataFrame 字符串
+    file_upload.change(process_file, inputs=file_upload, outputs=df_string_output)
 
-iface = gr.Interface(
-    fn=process_file,
-    inputs=[gr.File(label="Upload your file"), gr.Textbox(label="Your Question")],
-    outputs=gr.Text(label="Output")
-)
+    # 处理聊天机器人的对话
+    send_button.click(
+        respond, 
+        inputs=[msg, chatbot], 
+        outputs=[msg, chatbot]
+    )
 
-iface.launch()
+demo.launch()