Update app.py

app.py

@@ -5,33 +5,54 @@ import random
 def load_jsonl(file):
     """读取上传的jsonl文件并解析为字典列表"""
     data = []
-    file = open(file, "r")
-    for line in file:
-        data.append(json.loads(line))
+    with open(file, "r", encoding="utf-8") as f:
+        for line in f:
+            data.append(json.loads(line))
     return data
 
 def random_data_viewer(file):
-    """随机抽取一条数据并格式化输出"""
+    """读取文件并随机抽取一条数据"""
     if file is None:
-        return "请上传一个JSONL文件！"
+        return "请上传一个JSONL文件！", None
     
     data = load_jsonl(file)
     if len(data) == 0:
-        return "文件为空或格式不正确！"
+        return "文件为空或格式不正确！", None
     
     random_entry = random.choice(data)
-    # 将数据格式化为Markdown形式
+    # 格式化输出为Markdown
+    output = "\n".join([f"**{key}**: {value}" for key, value in random_entry.items()])
+    return output, data  # 返回数据以存储供后续使用
+
+def sample_more(data):
+    """从已有数据中再采样一条"""
+    if not data:
+        return "没有可用数据，请先上传JSONL文件！"
+    
+    random_entry = random.choice(data)
+    # 格式化输出为Markdown
     output = "\n".join([f"**{key}**: {value}" for key, value in random_entry.items()])
     return output
 
-# 创建Gradio界面
-iface = gr.Interface(
-    fn=random_data_viewer,
-    inputs=gr.File(file_types=[".jsonl"], label="上传JSONL文件"),
-    outputs=gr.Textbox(label="随机数据", lines=10, max_lines=20),  # 使用Textbox组件显示Markdown
-    title="JSONL 数据查看器",
-    description="上传一个JSONL文件，随机展示其中一条数据。",
-)
+# Gradio 界面
+with gr.Blocks() as app:
+    gr.Markdown("# JSONL 数据查看器")
+    gr.Markdown("上传一个JSONL文件，随机展示其中一条数据。点击按钮可以重新采样。")
+    
+    with gr.Row():
+        file_upload = gr.File(file_types=[".jsonl"], label="上传JSONL文件")
+        sample_button = gr.Button("再采样")
+    
+    output_box = gr.Textbox(label="随机数据", lines=10, max_lines=20)
+    
+    # 用于存储加载后的数据
+    state_data = gr.State()
+    
+    # 绑定事件：上传文件后随机取一条数据
+    file_upload.change(random_data_viewer, inputs=file_upload, outputs=[output_box, state_data])
+    
+    # 绑定事件：点击按钮后从已有数据中再采样
+    sample_button.click(sample_more, inputs=state_data, outputs=output_box)
 
-# 启动Gradio应用
-iface.launch()
+# 启动 Gradio 应用
+app.launch()