Update app.py

app.py

@@ -25,13 +25,10 @@ def call_api(only_final_result, data_df):
     """
     这个函数会被Gradio调用，它负责收集UI上的输入，
     将其转换成API需要的格式，然后发送请求。
-    (已修改：不再接收 model_name 参数)
     """
-    # 1. 数据校验 (不再需要校验模型名称)
     if data_df is None or data_df.empty:
         raise gr.Error("输入数据不能为空！请在表格中至少添加一行数据。")
 
-    # 2. 将输入的DataFrame转换成API需要的CSV字符串
     data_df_with_id = data_df.copy()
     data_df_with_id.insert(0, 'id', [f'{i:03d}' for i in range(len(data_df_with_id))])
     
@@ -39,7 +36,6 @@ def call_api(only_final_result, data_df):
     data_df_with_id.to_csv(output, index=False)
     csv_data = output.getvalue()
     
-    # 3. 构建发送到API的JSON数据体 (使用固定的模型名称)
     payload = {
         "model_name": FIXED_MODEL_NAME,
         "data_csv": csv_data,
@@ -47,13 +43,10 @@ def call_api(only_final_result, data_df):
     }
 
     try:
-        # 4. 发送POST请求
         response = requests.post(SERVER_URL, json=payload, timeout=300, stream=True)
         response.raise_for_status()
-
         result_json = response.json()
         
-        # 5. 处理和返回结果
         dataframe_result_data = result_json.get("result", [])
         
         if isinstance(dataframe_result_data, list) and len(dataframe_result_data) > 0:
@@ -81,13 +74,10 @@ with gr.Blocks(theme=gr.themes.Glass(), css=".gradio-container {max-width: 1280p
         """
     )
     
-    # ---------- 新布局：输入部分 ----------
     with gr.Column():
         gr.Markdown("### 1. 配置参数")
         with gr.Group():
-            # 使用 Markdown 或不可编辑的 Textbox 展示固定的模型名称
             gr.Markdown(f"**当前模型 (固定):** `{FIXED_MODEL_NAME}`")
-            
             only_final_result_input = gr.Checkbox(
                 label="仅返回最终结果 (only_final_result)",
                 value=False,
@@ -95,25 +85,26 @@ with gr.Blocks(theme=gr.themes.Glass(), css=".gradio-container {max-width: 1280p
             )
         
         gr.Markdown("### 2. 输入数据")
-        
         gr.Markdown("ℹ️ **操作提示**: 点击表格右下方的 **✏️ Edit** 按钮，然后点击 **+ Add row** 来添加新行，或点击垃圾桶图标删除行。")
         
         # -------------------- 主要修改部分在这里 --------------------
-        # 增加了 height 参数来让表格更高，并且放在了更宽的布局中
+        # 移除了 'height' 参数，因为它在旧版Gradio中不受支持。
+        # 我们通过增加 row_count 的第一个参数（初始行数）来为表格提供更多初始垂直空间。
         data_input = gr.DataFrame(
             label="数据表格",
             headers=["subject", "relation", "object", "prompt_source"],
             value=DEFAULT_DATAFRAME_VALUE,
-            row_count=(3, "dynamic"), # 默认显示3行
+            # 将初始行数从3增加到5，使表格在加载时就更高。
+            # (要显示的行数, "dynamic"表示用户可以增删行)
+            row_count=(5, "dynamic"),
             col_count=(4, "fixed"),
-            interactive=True,
-            height=300  # 明确设置组件高度，单位为像素
+            interactive=True
+            # 已移除不受支持的 'height' 参数
         )
         # -------------------- 修改结束 --------------------
         
         submit_btn = gr.Button("提交推理", variant="primary")
 
-    # ---------- 新布局：输出部分 ----------
     with gr.Column():
         gr.Markdown("### 3. 查看推理结果")
         with gr.Tabs():
@@ -121,16 +112,14 @@ with gr.Blocks(theme=gr.themes.Glass(), css=".gradio-container {max-width: 1280p
                 output_dataframe = gr.DataFrame(
                     label="推理结果详情", 
                     interactive=False,
-                    # 增加了列数以适应可能的更多输出
                     headers=["id", "subject", "relation", "object", "prompt_source", "prompt_target", "layer_source", "is_correct_patched", "generations"]
                 )
             with gr.TabItem("原始JSON响应"):
                 output_json = gr.JSON(label="API 返回的完整JSON")
             
-    # 定义按钮点击事件 (已更新 inputs 列表)
     submit_btn.click(
         fn=call_api,
-        inputs=[only_final_result_input, data_input], # 移除了 model_name_input
+        inputs=[only_final_result_input, data_input],
         outputs=[output_json, output_dataframe]
     )