Update app.py

app.py

@@ -61,27 +61,45 @@ def format_zero(val):
         return 0.0
     return val
 
-# Function to plot the triangle
+# Function to plot the triangle with all points in different colors and a legend
 def plot_triangle(x1, y1, x2, y2, x3, y3, I_x, I_y, Ux, Uy, G_x, G_y):
     fig, ax = plt.subplots(figsize=(8, 6))
     triangle = Polygon([(x1, y1), (x2, y2), (x3, y3)], closed=True, edgecolor='b', facecolor='lightblue')
     ax.add_patch(triangle)
     
+    # Define colors for different points
+    vertex_color = 'blue'
+    centroid_color = 'green'
+    incenter_color = 'red'
+    circumcenter_color = 'purple'
+    
+    # Plot the triangle vertices
     vertices = [(x1, y1), (x2, y2), (x3, y3)]
     for i, (vx, vy) in enumerate(vertices):
-        ax.scatter(vx, vy, color="blue", zorder=5)
-        ax.text(vx, vy, f"P{i+1} ({vx:.1f}, {vy:.1f})", fontsize=10, ha="left", va="bottom", color="blue")
+        ax.scatter(vx, vy, color=vertex_color, zorder=5)
+        ax.text(vx, vy, f"P{i+1} ({vx:.1f}, {vy:.1f})", fontsize=10, ha="left", va="bottom", color=vertex_color)
     
+    # Plot key points with their corresponding colors
     key_points = [
-        (I_x, I_y, "Incenter"),
-        (Ux, Uy, "Circumcenter"),
-        (G_x, G_y, "Centroid")
+        (I_x, I_y, incenter_color, "Incenter"),
+        (Ux, Uy, circumcenter_color, "Circumcenter"),
+        (G_x, G_y, centroid_color, "Centroid")
     ]
     
-    for x, y, label in key_points:
-        ax.scatter(x, y, color="red", zorder=5)
-        ax.text(x, y, f"{label}\n({x:.1f}, {y:.1f})", fontsize=10, ha="right", va="top", color="red")
+    for x, y, color, label in key_points:
+        ax.scatter(x, y, color=color, zorder=5)
+        ax.text(x, y, f"{label}\n({x:.1f}, {y:.1f})", fontsize=10, ha="right", va="top", color=color)
+    
+    # Add legend
+    handles = [
+        plt.Line2D([0], [0], marker='o', color='w', markerfacecolor=vertex_color, markersize=10, label="Vertices"),
+        plt.Line2D([0], [0], marker='o', color='w', markerfacecolor=centroid_color, markersize=10, label="Centroid"),
+        plt.Line2D([0], [0], marker='o', color='w', markerfacecolor=incenter_color, markersize=10, label="Incenter"),
+        plt.Line2D([0], [0], marker='o', color='w', markerfacecolor=circumcenter_color, markersize=10, label="Circumcenter")
+    ]
+    ax.legend(handles=handles, loc='upper left', fontsize=12)
     
+    # Adjust the plot limits and aspect ratio
     padding = 3
     ax.set_xlim([min(x1, x2, x3) - padding, max(x1, x2, x3) + padding])
     ax.set_ylim([min(y1, y2, y3) - padding, max(y1, y2, y3) + padding])
@@ -169,8 +187,8 @@ def main():
             st.markdown(f"Perimeter: **{perimeter:.3f}** units")
             st.markdown(f"Centroid: **({G_x:.3f}, {G_y:.3f})**")
 
-        # Display triangle graph with incenter, circumcenter, and centroid
+        # Display triangle graph with colored points and legend
         plot_triangle(x1, y1, x2, y2, x3, y3, I_x, I_y, Ux, Uy, G_x, G_y)
 
 if __name__ == "__main__":
-    main()
+    main()