Update app.py

app.py

@@ -10,26 +10,50 @@ def load_jsonl(file):
         data.append(json.loads(line))
     return data
 
-def random_data_viewer(file):
+def random_data_viewer(file, state):
     """随机抽取一条数据并格式化输出"""
     if file is None:
-        return "请上传一个jsonl文件！"
+        return "请上传一个jsonl文件！", state
     
-    data = load_jsonl(file)
-    if len(data) == 0:
-        return "文件为空或格式不正确！"
+    if state is None:
+        state = load_jsonl(file)
     
-    random_entry = random.choice(data)
+    if len(state) == 0:
+        return "文件为空或格式不正确！", state
+    
+    random_entry = random.choice(state)
     output = "\n".join([f"{key}: {value}" for key, value in random_entry.items()])
-    return output
+    return output, state
 
 # 创建Gradio界面
 iface = gr.Interface(
     fn=random_data_viewer,
-    inputs=gr.File(file_types=[".jsonl"], label="上传JSONL文件"),
-    outputs=gr.Markdown(label="随机数据", render=True),
+    inputs=[
+        gr.File(file_types=[".jsonl"], label="上传JSONL文件"),
+        gr.State()  # 用于存储加载的数据
+    ],
+    outputs=[
+        gr.Markdown(label="随机数据", render=True),
+        gr.State()  # 用于保持数据状态
+    ],
+    title="JSONL 数据查看器",
+    description="上传一个JSONL文件，随机展示其中一条数据。点击按钮重新采样。"
+)
+
+# 添加一个按钮用于重新采样
+iface = gr.Interface(
+    fn=random_data_viewer,
+    inputs=[
+        gr.File(file_types=[".jsonl"], label="上传JSONL文件"),
+        gr.State()  # 用于存储加载的数据
+    ],
+    outputs=[
+        gr.Markdown(label="随机数据", render=True),
+        gr.State()  # 用于保持数据状态
+    ],
     title="JSONL 数据查看器",
-    description="上传一个JSONL文件，随机展示其中一条数据。"
+    description="上传一个JSONL文件，随机展示其中一条数据。点击按钮重新采样。",
+    live=True  # 允许实时更新
 )
 
 # 启动Gradio应用