Modified the Object behavior / Method plot_vector_add into a previous instantiation

app.py

@@ -18,10 +18,10 @@ class Vector_Addition:
         v3_text = f'The new vector 3 produced = vector 1: {self.v1} + vector 2: {self.v2} = {v3}'
         return v3_text
 
-    def plot_vector_add(self):
+     def plot_vector_add(self):
         plt.figure(figsize=(10,5))
         plt.plot([0, self.v1[0]], [0, self.v1[1]], 'b', label='v1')
-        plt.plot([0, self.v2[0]+self.v1[0]], [0, self.v2[1]+self.v1[1]], 'r', label='v2')
+        plt.plot([0, self.v2[0]], [0, self.v2[1]], 'r', label='v2')
         plt.plot([0, self.v1[0]+ self.v2[0]], [0, self.v1[1]+ self.v2[1]], 'k', label='v1+v2')
         plt.xlabel('X')
         plt.ylabel('Y')
@@ -36,6 +36,13 @@ class Vector_Addition:
         plt.close()
         return image
 
+        buf = BytesIO()
+        plt.savefig(buf, format='png')
+        buf.seek(0)
+        image = Image.open(buf)
+        plt.close()
+        return image
+
 def vector_addition(v1, v2):
     vector_addition = Vector_Addition(v1, v2)
     v3 = vector_addition.vector_add()