Update app.py

app.py

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