Update app.py

app.py

@@ -9,7 +9,7 @@ import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
 from sklearn.linear_model import LinearRegression
-
+import sympy as sp
 
 
 np.random.seed(42)
@@ -47,6 +47,21 @@ def draw_polynomial(degree=2):
     lin_reg = LinearRegression()
     lin_reg.fit(X_poly, y)
     lin_reg.intercept_, lin_reg.coef_
+
+    
+    # Sample polynomial coefficients
+    coefficients = lin_reg.coef_[0]
+    
+    # Create a symbolic variable for x
+    x = sp.symbols('x')
+    
+    # Create the symbolic polynomial expression
+    polynomial_expr = lin_reg.intercept_[0]+ sum(coeff * x**power for power, coeff in enumerate(coefficients))
+    
+    # Convert the symbolic expression to a LaTeX string
+    latex_expression = sp.latex(polynomial_expr, mode='inline')
+
+
     
     ## Visualized fitted quadratic line on training/testing data
     X_new=np.linspace(-3, 3, 100).reshape(100, 1)
@@ -96,13 +111,14 @@ def draw_polynomial(degree=2):
     #plt.show()
     fig.tight_layout()
     plt.savefig('plot_line.png', dpi=300)
-    return results, 'plot_line.png'
+    return latex_expression, results, 'plot_line.png'
 
 
 #### Define input component
 input_degree = gr.inputs.Slider(1, 35, step=1, default=2, label='Degree of Polynomial Regression')
 
 #### Define output component
+set_formula = gr.outputs.Textbox(label ='Polynomial Model')
 set_output = gr.outputs.Dataframe(type='pandas', label ='Evaluation Results')
 output_plot1 = gr.outputs.Image(label="Regression plot", type='pil')
 
@@ -111,7 +127,7 @@ output_plot1 = gr.outputs.Image(label="Regression plot", type='pil')
 ### configure gradio, detailed can be found at https://www.gradio.app/docs/#i_slider
 interface = gr.Interface(fn=draw_polynomial, 
                          inputs=[input_degree], 
-                         outputs=[set_output, output_plot1],
+                         outputs=[set_formula, set_output, output_plot1],
                          title="CSCI4750/5750: Polynomial Regression models \n (Model Complexity)", 
                          theme = 'huggingface',
                          layout = 'vertical'