from manim import *

class PythagoreanTheoremAnimation(Scene):
    def construct(self):
        # 绘制一个直角三角形
        triangle = Polygon([0, 0, 0], [3, 0, 0], [3, 4, 0], color=BLUE)

        # 标注边长
        a_text = Text("a = 3").move_to([1.5, -0.5, 0])
        b_text = Text("b = 4").move_to([3.5, 2, 0])
        c_text = Text("c =?").move_to([1.5, 2, 0])

        # 绘制直角符号
        right_angle = RightAngle(triangle[0], triangle[1], triangle[2])

        # 显示三角形和标注
        self.play(Create(triangle), Create(right_angle), Write(a_text), Write(b_text), Write(c_text))

        # 计算并显示斜边长度
        c = math.sqrt(3**2 + 4**2)
        c_value_text = Text(f"c = {c:.2f}").move_to([1.5, 4, 0])
        self.play(Write(c_value_text))

        self.wait()