Update app.py

app.py

@@ -393,19 +393,11 @@ def create_demo():
                     interactive=False
                 )
         
-        # 分数演进图表
-        score_chart = gr.LinePlot(
-            label="分数演进",
-            x="iteration",
-            y="score",
-            y_lim=[1.0, 0],
-            x_title="迭代次数",
-            y_title="分数",
-            height=250
-        )
+        # 分数演进 - 用 HTML 进度条
+        score_display = gr.HTML(label="分数演进")
         
-        # 额外显示分数列表
-        score_list = gr.JSON(label="分数历史")
+        # 分数历史
+        score_list = gr.JSON(label="分数历史", visible=False)
         
         gr.Markdown("### 迭代详情")
         
@@ -422,20 +414,17 @@ def create_demo():
         
         def run_task(task: str, max_iter: int, target: float):
             if not task or not task.strip():
-                yield "错误: 请输入任务描述", pd.DataFrame(columns=["iteration", "score"]), [], "", "", "", ""
+                yield "错误: 请输入任务描述", "", "", "", "", ""
                 return
             
             chart_data = []
             current_score = 0.0
             
             empty_md = "*等待开始...*"
-            # 初始状态：显示"准备中"
-            yield "状态: 准备执行任务...", pd.DataFrame(columns=["iteration", "score"]), [], empty_md, empty_md, empty_md, empty_md
-            
-            # 存储所有迭代的完整历史
-            all_history = []
+            empty_html = "<p>等待开始...</p>"
+            # 初始状态
+            yield "状态: 准备执行任务...", empty_html, empty_md, empty_md, empty_md, empty_md
             
-            # 先执行完整的一次 PEES 迭代，然后一起更新
             for i in range(1, int(max_iter) + 1):
                 # 执行完整迭代
                 results, current_score = run_pees_iteration(task, i, current_score, target)
@@ -446,7 +435,7 @@ def create_demo():
                 execute2_result = results[2]
                 summary_result = results[3]
                 
-                # 格式化每个阶段的输出（更美观）
+                # 格式化每个阶段的输出
                 plan_md = f"""### 迭代 {i} - Plan 计划
 **时间**: {plan_result['timestamp']}
 
@@ -499,12 +488,36 @@ def create_demo():
 </details>
 """
                 
-                # 更新图表数据
+                # 更新数据
                 chart_data.append({"iteration": i, "score": round(current_score, 2)})
                 
-                # 每次迭代完成后才更新 UI
+                # 生成 HTML 进度条 - 完全可控
+                bars_html = "<div style='display: flex; gap: 10px; align-items: flex-end; height: 200px; padding: 20px; background: #f9f9f9; border-radius: 8px;'>"
+                for item in chart_data:
+                    score_val = item["score"]
+                    iter_num = item["iteration"]
+                    # 颜色：低=红，中=黄，高=绿
+                    if score_val < 0.4:
+                        color = "#ef4444"
+                    elif score_val < 0.7:
+                        color = "#eab308"
+                    else:
+                        color = "#22c55e"
+                    bar_height = score_val * 100
+                    bars_html += f"""
+                    <div style='display: flex; flex-direction: column; align-items: center; flex: 1;'>
+                        <div style='font-weight: bold; margin-bottom: 5px;'>{score_val:.2f}</div>
+                        <div style='width: 100%; background: #e5e7eb; border-radius: 4px; height: 150px; position: relative;'>
+                            <div style='position: absolute; bottom: 0; width: 100%; background: {color}; border-radius: 4px; height: {bar_height}%; transition: height 0.3s;'></div>
+                        </div>
+                        <div style='margin-top: 5px;'>迭代{iter_num}</div>
+                    </div>
+                    """
+                bars_html += "</div>"
+                
+                # 每次迭代完成后更新 UI
                 status = f"状态: 第 {i}/{int(max_iter)} 次迭代完成 (分数: {current_score:.2f})"
-                yield status, pd.DataFrame(chart_data), chart_data, plan_md, exec1_md, exec2_md, summary_md
+                yield status, bars_html, plan_md, exec1_md, exec2_md, summary_md
                 
                 # 检查是否达到目标
                 if current_score >= target:
@@ -513,12 +526,12 @@ def create_demo():
                 time.sleep(0.3)
             
             final_status = f"状态: 任务完成\n最终分数: {current_score:.2f}\n总迭代次数: {len(chart_data)}"
-            yield final_status, pd.DataFrame(chart_data), chart_data, plan_md, exec1_md, exec2_md, summary_md
+            yield final_status, bars_html, plan_md, exec1_md, exec2_md, summary_md
         
         run_btn.click(
             fn=run_task,
             inputs=[task_input, max_iterations, target_score],
-            outputs=[status_output, score_chart, score_list, plan_output, execute1_output, execute2_output, summary_output]
+            outputs=[status_output, score_display, plan_output, execute1_output, execute2_output, summary_output]
         )
         
         gr.Markdown("""