Update app.py

app.py

@@ -9,21 +9,25 @@ import io
 SERVER_URL = "http://103.235.229.133:7000/predict"
 # -------------------------------------------------------
 
-# 默认的模型名称和输入数据，方便用户测试
-DEFAULT_MODEL_NAME = "/data/tsq/MODELS/Qwen2.5-1.5B-Instruct"
+# ----------------- 固定的模型名称 -----------------
+# 根据要求，模型名称是固定的，不再作为UI输入
+FIXED_MODEL_NAME = "/data/tsq/MODELS/Qwen2.5-1.5B-Instruct"
+# -------------------------------------------------------
+
+# 默认的输入数据，方便用户测试
 DEFAULT_DATAFRAME_VALUE = [
     ["France", "capital city of", "Paris", "she traveled to France for the first time"],
-    ["Germany", "capital city of", "Berlin", "the capital of Germany is Berlin"]
+    ["Germany", "capital city of", "Berlin", "Germany is a country in Central Europe"],
+    ["Japan", "capital city of", "Tokyo", "Tokyo is the largest city in Japan"]
 ]
 
-def call_api(model_name, only_final_result, data_df):
+def call_api(only_final_result, data_df):
     """
     这个函数会被Gradio调用，它负责收集UI上的输入，
     将其转换成API需要的格式，然后发送请求。
+    (已修改：不再接收 model_name 参数)
     """
-    # 1. 数据校验
-    if not model_name.strip():
-        raise gr.Error("模型名称不能为空！")
+    # 1. 数据校验 (不再需要校验模型名称)
     if data_df is None or data_df.empty:
         raise gr.Error("输入数据不能为空！请在表格中至少添加一行数据。")
 
@@ -35,9 +39,9 @@ def call_api(model_name, only_final_result, data_df):
     data_df_with_id.to_csv(output, index=False)
     csv_data = output.getvalue()
     
-    # 3. 构建发送到API的JSON数据体
+    # 3. 构建发送到API的JSON数据体 (使用固定的模型名称)
     payload = {
-        "model_name": model_name,
+        "model_name": FIXED_MODEL_NAME,
         "data_csv": csv_data,
         "only_final_result": only_final_result
     }
@@ -77,58 +81,56 @@ with gr.Blocks(theme=gr.themes.Glass(), css=".gradio-container {max-width: 1280p
         """
     )
     
-    with gr.Row():
-        # 左侧列：输入配置
-        with gr.Column(scale=2):
-            gr.Markdown("### 1. 配置模型和参数")
-            with gr.Group():
-                model_name_input = gr.Textbox(
-                    label="模型名称 (model_name)",
-                    value=DEFAULT_MODEL_NAME,
-                    info="请输入部署在服务器上的模型路径。"
-                )
-                only_final_result_input = gr.Checkbox(
-                    label="仅返回最终结果 (only_final_result)",
-                    value=False,
-                    info="勾选此项，API将只返回最终聚合结果，而不是所有层的详细信息。"
-                )
-            
-            gr.Markdown("### 2. 输入数据")
-            
-            # -------------------- 主要修改部分在这里 --------------------
-            # 将原来的 info 参数内容，移到一个独立的 Markdown 组件中
-            gr.Markdown("ℹ️ **操作提示**: 点击表格右下方的 **✏️ Edit** 按钮，然后点击 **+ Add row** 来添加新行，或点击垃圾桶图标删除行。")
+    # ---------- 新布局：输入部分 ----------
+    with gr.Column():
+        gr.Markdown("### 1. 配置参数")
+        with gr.Group():
+            # 使用 Markdown 或不可编辑的 Textbox 展示固定的模型名称
+            gr.Markdown(f"**当前模型 (固定):** `{FIXED_MODEL_NAME}`")
             
-            data_input = gr.DataFrame(
-                label="数据表格",
-                headers=["subject", "relation", "object", "prompt_source"],
-                value=DEFAULT_DATAFRAME_VALUE,
-                row_count=(1, "dynamic"),
-                col_count=(4, "fixed"),
-                interactive=True
-                # 已移除 'info' 参数
+            only_final_result_input = gr.Checkbox(
+                label="仅返回最终结果 (only_final_result)",
+                value=False,
+                info="勾选此项，API将只返回最终聚合结果，而不是所有层的详细信息。"
             )
-            # -------------------- 修改结束 --------------------
-            
-            submit_btn = gr.Button("提交推理", variant="primary", scale=1)
+        
+        gr.Markdown("### 2. 输入数据")
+        
+        gr.Markdown("ℹ️ **操作提示**: 点击表格右下方的 **✏️ Edit** 按钮，然后点击 **+ Add row** 来添加新行，或点击垃圾桶图标删除行。")
+        
+        # -------------------- 主要修改部分在这里 --------------------
+        # 增加了 height 参数来让表格更高，并且放在了更宽的布局中
+        data_input = gr.DataFrame(
+            label="数据表格",
+            headers=["subject", "relation", "object", "prompt_source"],
+            value=DEFAULT_DATAFRAME_VALUE,
+            row_count=(3, "dynamic"), # 默认显示3行
+            col_count=(4, "fixed"),
+            interactive=True,
+            height=300  # 明确设置组件高度，单位为像素
+        )
+        # -------------------- 修改结束 --------------------
+        
+        submit_btn = gr.Button("提交推理", variant="primary")
 
-        # 右侧列：输出结果
-        with gr.Column(scale=3):
-            gr.Markdown("### 3. 查看推理结果")
-            with gr.Tabs():
-                with gr.TabItem("结果表格"):
-                    output_dataframe = gr.DataFrame(
-                        label="推理结果详情", 
-                        interactive=False,
-                        headers=["subject", "prompt_source", "prompt_target", "layer_source", "is_correct_patched", "generations"]
-                    )
-                with gr.TabItem("原始JSON响应"):
-                    output_json = gr.JSON(label="API 返回的完整JSON")
+    # ---------- 新布局：输出部分 ----------
+    with gr.Column():
+        gr.Markdown("### 3. 查看推理结果")
+        with gr.Tabs():
+            with gr.TabItem("结果表格"):
+                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=[model_name_input, only_final_result_input, data_input],
+        inputs=[only_final_result_input, data_input], # 移除了 model_name_input
         outputs=[output_json, output_dataframe]
     )