Update 配置.py

配置.py

@@ -73,7 +73,7 @@ def _sniff_delimiter(path):
 
 def 读取CSV预览(file_obj):
     if file_obj is None:
-        return "未选择文件", [], ""
+        return "未选择文件", "", ""
     path = file_obj.name
     delim = _sniff_delimiter(path)
     headers = []
@@ -87,12 +87,13 @@ def 读取CSV预览(file_obj):
                 if i < 5:
                     head_rows.append(row)
                 total += 1
-        info = f"分隔符: '{delim}' | 列: {headers} | 总行(不含表头): ~{total}"
-        return info, headers, "\n".join([str(x) for x in head_rows])
+        info = f"分隔符: '{delim}' | 列: {len(headers)} | 总行(不含表头): ~{total}"
+        preview = "\n".join([",".join(row) for row in head_rows])
+        return info, ",".join(headers), preview
     except Exception as e:
-        return f"读取失败: {e}", [], ""
+        return f"读取失败: {e}", "", ""
 
-def 标准化CSV(file_obj, 映射, 输出文件名):
+def 标准化CSV(file_obj, 输出文件名, 时间列, 开列, 高列, 低列, 收列, 量列):
     if file_obj is None:
         return "请先上传CSV", ""
     path = file_obj.name
@@ -102,13 +103,11 @@ def 标准化CSV(file_obj, 映射, 输出文件名):
     os.makedirs(save_dir, exist_ok=True)
     out_path = os.path.join(save_dir, 输出文件名)
 
-    # 校验映射
-    必要 = ["时间","开","高","低","收"]
-    for k in 必要:
-        if not 映射 or not 映射.get(k):
-            return f"映射缺少必要列: {k}", ""
-    vol_col = 映射.get("量")  # 可选
-
+    # 校验必要列
+    必要 = [时间列, 开列, 高列, 低列, 收列]
+    if not all(必要):
+        return f"请确保时间、开、高、低、收列都已选择", ""
+    
     # 读取、转换并写出标准格式
     delim = _sniff_delimiter(path)
     写入 = 0
@@ -120,12 +119,12 @@ def 标准化CSV(file_obj, 映射, 输出文件名):
             wr.writerow(["时间","开","高","低","收","量"])  # 标准表头
             for row in r:
                 try:
-                    tms = _parse_time_to_ms(row[映射["时间"]])
-                    o = float(row[映射["开"]])
-                    h = float(row[映射["高"]])
-                    l = float(row[映射["低"]])
-                    c = float(row[映射["收"]])
-                    v = float(row[vol_col]) if vol_col and row.get(vol_col) not in (None, "") else 0.0
+                    tms = _parse_time_to_ms(row[时间列])
+                    o = float(row[开列])
+                    h = float(row[高列])
+                    l = float(row[低列])
+                    c = float(row[收列])
+                    v = float(row[量列]) if 量列 and 量列 in row and row[量列] not in (None, "") else 0.0
                     if tms is None:
                         continue
                     wr.writerow([tms, o, h, l, c, v])
@@ -213,58 +212,70 @@ def 构建设置面板(容器, 默认参数=None, 记录函数=None):
                 with gr.Row():
                     with gr.Column(scale=1):
                         csv文件 = gr.File(label="上传 1m K线 CSV", file_types=[".csv"])
-                        预览信息 = gr.Textbox(label="文件预览", lines=3, interactive=False)
+                        
+                        # 预览区域
+                        预览信息 = gr.Textbox(label="文件信息", lines=2, interactive=False)
+                        表头显示 = gr.Textbox(label="检测到的表头", lines=2, interactive=False)
                         预览前几行 = gr.Textbox(label="前几行示例", lines=8, interactive=False)
-                        表头下拉容器 = gr.Column()
-                        with 表头下拉容器:
-                            # 动态下拉在上传后创建（避免列名未知）
-                            pass
+                        
+                        # 映射下拉菜单（静态创建）
+                        gr.Markdown("### 列映射")
+                        时间列 = gr.Dropdown(label="时间列", choices=[], value=None)
+                        开列 = gr.Dropdown(label="开盘价列", choices=[], value=None)
+                        高列 = gr.Dropdown(label="最高价列", choices=[], value=None)
+                        低列 = gr.Dropdown(label="最低价列", choices=[], value=None)
+                        收列 = gr.Dropdown(label="收盘价列", choices=[], value=None)
+                        量列 = gr.Dropdown(label="成交量列（可选）", choices=[], value=None)
+                        
                         输出文件名 = gr.Textbox(label="输出文件名", value="uploaded_1m.csv")
                         保存按钮 = gr.Button("标准化并保存")
                         保存结果 = gr.Textbox(label="结果", interactive=False)
                         输出路径 = gr.Textbox(label="输出路径", interactive=False)
+                        
                     with gr.Column(scale=1):
-                        gr.Markdown("标准格式说明")
-                        gr.Markdown("目标列: 时间(ms), 开, 高, 低, 收, 量\n- 时间列自动识别秒/毫秒/ISO时间\n- 缺量则填0\n- 输出路径位于 /tmp/app_data/csv/")
+                        gr.Markdown("### 使用说明")
+                        gr.Markdown("1. 上传 CSV 文件\n2. 系统自动检测表头\n3. 选择对应的列映射\n4. 点击保存生成标准格式")
+                        gr.Markdown("标准格式: 时间(ms), 开, 高, 低, 收, 量")
 
-                # 上传后预览并生成映射下拉
-                def on_csv_upload(file_obj):
-                    info, headers, head = 读取CSV预览(file_obj)
-                    if not headers:
-                        return info, head, gr.update(visible=False), None, None, None, None, None
+                # CSV 上传后的处理函数
+                def 处理CSV上传(file_obj):
+                    if file_obj is None:
+                        return ("", "", "", [], None, None, None, None, None, None)
+                    
+                    info, headers_str, preview = 读取CSV预览(file_obj)
+                    if not headers_str:
+                        return (info, "", preview, [], None, None, None, None, None, None)
+                    
+                    headers = headers_str.split(",")
                     映射 = _guess_mapping(headers)
-                    # 生成下拉组件
-                    with 表头下拉容器:
-                        映射_时间 = gr.Dropdown(choices=headers, label="映射：时间列", value=映射.get("时间"))
-                        映射_开 = gr.Dropdown(choices=headers, label="映射：开(Open)", value=映射.get("开"))
-                        映射_高 = gr.Dropdown(choices=headers, label="映射：高(High)", value=映射.get("高"))
-                        映射_低 = gr.Dropdown(choices=headers, label="映射：低(Low)", value=映射.get("低"))
-                        映射_收 = gr.Dropdown(choices=headers, label="映射：收(Close)", value=映射.get("收"))
-                        映射_量 = gr.Dropdown(choices=headers, label="映射：量(Volume，可选)", value=映射.get("量"))
-                    return info, head, gr.update(visible=True), 映射_时间, 映射_开, 映射_高, 映射_低, 映射_收, 映射_量
-
-                # 用于保存时收集映射
-                动态映射状态 = gr.State(value=None)
-
-                def 收集映射(时间c, 开c, 高c, 低c, 收c, 量c):
-                    return {"时间": 时间c, "开": 开c, "高": 高c, "低": 低c, "收": 收c, "量": 量c}
+                    
+                    return (
+                        info, 
+                        headers_str, 
+                        preview, 
+                        headers,
+                        映射.get("时间"), 
+                        映射.get("开"), 
+                        映射.get("高"), 
+                        映射.get("低"), 
+                        映射.get("收"), 
+                        映射.get("量")
+                    )
 
-                # 绑定上传
-                out = csv文件.upload(
-                    fn=on_csv_upload,
+                # 更新下拉菜单选项和默认值
+                csv文件.change(
+                    fn=处理CSV上传,
                     inputs=csv文件,
-                    outputs=[预览信息, 预览前几行, 表头下拉容器, gr.State(), gr.State(), gr.State(), gr.State(), gr.State(), gr.State()],
+                    outputs=[
+                        预览信息, 表头显示, 预览前几行,
+                        时间列, 时间列, 开列, 高列, 低列, 收列, 量列
+                    ]
                 )
-
-                # 这里 out.outputs[3:] 分别是我们返回的 6 个下拉组件（以State占位接收）
-                def 保存触发(file_obj, 输出名, m时间, m开, m高, m低, m收, m量):
-                    映射 = {"时间": m时间, "开": m开, "高": m高, "低": m低, "收": m收, "量": m量}
-                    msg, path = 标准化CSV(file_obj, 映射, 输出名)
-                    return msg, path
-
+                
+                # 保存按钮点击事件
                 保存按钮.click(
-                    保存触发,
-                    inputs=[csv文件, 输出文件名] + out[1][3:],  # 取动态返回的6个下拉
+                    fn=标准化CSV,
+                    inputs=[csv文件, 输出文件名, 时间列, 开列, 高列, 低列, 收列, 量列],
                     outputs=[保存结果, 输出路径]
                 )