added stuff to make it easy

fourier_series.py

@@ -1,10 +1,12 @@
 from sympy import *
 import numpy as np
 import matplotlib.pyplot as plt
+import io
 
 
 x, n = symbols("x n")
 
+
 def fourier_series_evaluator(func, start, end):
     T = end - start
     cos_term = cos((2 * n * pi * x) / T)
@@ -18,6 +20,22 @@ def fourier_series_evaluator(func, start, end):
     return (a0, an, bn)
 
 
+def fourier_series_latex(func, start, end):
+    a0, an, bn = fourier_series_evaluator(func, start, end)
+    a0, an, bn = (
+        nsimplify(a0, tolerance=0.01, full=True),
+        nsimplify(an, tolerance=0.01, full=True),
+        nsimplify(bn, tolerance=0.01, full=True),
+    )
+
+    latex_string = latex(a0) + " + "
+    latex_string += "\\sum_{n=1} ^{\\infty} "
+    latex_string += latex(an)
+    latex_string += latex(bn)
+
+    return latex_string
+
+
 def fourier_graph_compute(func, start, end, terms=6, repeat=1, dir=1):
     a0, an, bn = fourier_series_evaluator(func, start, end)
     T = end - start
@@ -58,6 +76,21 @@ def fourier_visualizer(func, start, end, terms=6, repeat=1, dir=1):
     plt.show()
 
 
+def fourier_visualizer_svg(func, start, end, terms=6, repeat=1, dir=1):
+    x_axis, y_axis = fourier_graph_compute(func, start, end, terms, repeat, dir)
+
+    for i in y_axis:
+        plt.plot(x_axis, i)
+
+    buf = io.BytesIO()
+    plt.savefig(buf, format="svg")
+    buf.seek(0)
+
+    plt.close()
+
+    return buf
+
+
 if __name__ == "__main__":
     func = eval(input("Enter Function to be Plot\n\tf(x) = "))
     start = float(input("Enter start of period: "))

graph_2d.py

@@ -1,6 +1,7 @@
 import matplotlib.pyplot as plt
 import numpy as np
 from sympy import *
+import io
 
 
 x = symbols("x")
@@ -20,6 +21,19 @@ def graph(func, start=-3.14, end=3.14):
     plt.show()
 
 
+def graph_svg(func, start=-3.14, end=3.14):
+    x_axis, y_axis = graph_compute_points(func, start, end)
+    plt.plot(x_axis, y_axis)
+
+    buf = io.BytesIO()
+    plt.savefig(buf, format="svg")
+    buf.seek(0)
+
+    plt.close()
+
+    return buf
+
+
 if __name__ == "__main__":
     func = input("f(x): ")
     graph(eval(func))