修复单个target的bug，增加模拟和云端的通信功能

app.py

@@ -15,6 +15,7 @@ from datasets import Features, Image, Audio, Sequence
 from typing import List, Tuple, Callable
 
 from utils import ndarray_to_base64, clean_up_df, create_statistic, create_plot
+from comm_utils import save_to_file, send_msg_to_server
 
 class AppError(RuntimeError):
     pass
@@ -233,10 +234,13 @@ with gr.Blocks() as demo:
             statistics_textbox = gr.DataFrame()
         with gr.Column(scale=3):
             plot = gr.Plot()
-    user_input_text = gr.Textbox(placeholder="输入一些内容")
+    with gr.Row():
+        user_input_box = gr.Textbox(placeholder="输入一些内容", label="输入", lines=5, interactive=True)
+        user_output_box = gr.Textbox(label="回答", lines=5, interactive=False)
+    user_io_buttom = gr.Button("发送", interactive=True)
     # componets.append({"select_sample_box": select_sample_box,
     #                   "statistics_textbox": statistics_textbox,
-    #                   "user_input_text": user_input_text,
+    #                   "user_input_box": user_input_box,
     #                   "plot": plot})
 
     # with gr.Row():
@@ -261,11 +265,13 @@ with gr.Blocks() as demo:
                 if type(page) == str:
                     page = [page]
                 df_list, id_list = [], []
+                # TODO: 将以下内容封装为函数
                 for i, page in enumerate(page):
                     df, max_page, info = get_page(dataset, config, split, page)
                     global tot_samples, tot_targets
-                    tot_samples, tot_targets = max_page, len(df['target'][0]) if isinstance(df['target'][0], np.ndarray) else 1
-                    
+                    tot_samples, tot_targets = max_page, len(df['target'][0]) if isinstance(df['target'][0], np.ndarray) and df['target'][0].dtype == 'O' else 1
+                    if 'all' in sub_targets:
+                        sub_targets = [i for i in range(tot_targets)]
                     df = clean_up_df(df, sub_targets)
                     row = df.iloc[0]
                     id_list.append(row['item_id'])
@@ -334,25 +340,24 @@ with gr.Blocks() as demo:
         except AppError as err:
             return show_error(str(err))
 
-    def save_to_file(user_input):
-        with open("user_input.txt", "w") as file:
-            file.write(user_input)
-
     all_outputs = [cp_config, cp_split,
                 #    cp_page, cp_goto_page, cp_goto_next_page,
                    cp_result, cp_info, cp_error,
                    select_sample_box, select_subtarget_box,
-                   select_buttom, statistics_textbox, user_input_text, plot]
+                   select_buttom, statistics_textbox, plot]
     cp_go.click(show_dataset, inputs=[cp_dataset], outputs=all_outputs)
     cp_config.change(show_dataset_at_config, inputs=[cp_dataset, cp_config], outputs=all_outputs)
     cp_split.change(show_dataset_at_config_and_split, inputs=[cp_dataset, cp_config, cp_split], outputs=all_outputs)
     # cp_goto_page.click(show_dataset_at_config_and_split_and_page, inputs=[cp_dataset, cp_config, cp_split, cp_page], outputs=all_outputs)
     # cp_goto_next_page.click(show_dataset_at_config_and_split_and_next_page, inputs=[cp_dataset, cp_config, cp_split, cp_page], outputs=all_outputs)
-    user_input_text.submit(save_to_file, inputs=user_input_text)
+    user_io_buttom.click(send_msg_to_server, inputs=[user_input_box], outputs=[user_output_box])
     select_buttom.click(show_dataset_at_config_and_split_and_page, inputs=[cp_dataset, cp_config, cp_split, select_sample_box, select_subtarget_box], outputs=all_outputs)
 
 
 if __name__ == "__main__":
 
     app = gr.mount_gradio_app(app, demo, path="/")
-    uvicorn.run(app, host="127.0.0.1", port=7860)
+    host = "127.0.0.1" if os.getenv("DEV") else "0.0.0.0"
+    import subprocess
+    subprocess.Popen(["python", "test_server.py"])
+    uvicorn.run(app, host=host, port=7860)

comm_utils.py

@@ -0,0 +1,25 @@
+import requests
+
+
+API_URL = "http://127.0.0.1:5000/api/process"
+
+def save_to_file(user_input):
+    with open("user_input.txt", "w") as file:
+        file.write(user_input)
+
+
+def send_msg_to_server(input_text):
+    try:
+        # 构造请求数据
+        payload = {"text": input_text}
+        headers = {"Content-Type": "application/json"}
+        
+        # 发送请求
+        response = requests.post(API_URL, json=payload, headers=headers)
+        response.raise_for_status()  # 检查是否请求成功
+        
+        # 返回响应结果
+        result = response.json()  # 假设服务器返回的是 JSON 格式
+        return result.get("processed_text", "No result returned.")
+    except requests.RequestException as e:
+        return f"请求失败：{e}"

test_server.py

@@ -0,0 +1,17 @@
+import time
+from flask import Flask, request, jsonify
+
+app = Flask(__name__)
+
+@app.route('/api/process', methods=['POST'])
+def process_text():
+    data = request.get_json()
+    input_text = data.get("text", "")
+
+    time.sleep(1)
+    processed_text = f"{input_text[::-1]}"
+    
+    return jsonify({"processed_text": processed_text})
+
+if __name__ == "__main__":
+    app.run(host="127.0.0.1", port=5000)

utils.py

@@ -114,10 +114,7 @@ def clean_up_df(df: pd.DataFrame, rows_to_include: list[int]) -> pd.DataFrame:
     """
     清理数据集，将嵌套的np.ndarray列展平为多列。
     """
-    if 'all' in rows_to_include:
-        rows_to_include = list(range(len(df['target'][0]))) if isinstance(df['target'][0], np.ndarray) else 1
-    else:
-        rows_to_include = sorted(rows_to_include)
+    rows_to_include = sorted(rows_to_include)
 
     df['timestamp'] = df.apply(lambda row: pd.date_range(
         start=row['start'],