File size: 1,832 Bytes
ddb99ef |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 |
import numpy as np
import matplotlib.pyplot as plt
import gradio as gr
def plot_polynomial(degree, coeff_input):
try:
coeffs = list(map(float, coeff_input.strip().split()))
if len(coeffs) != degree + 1:
return "⚠️ Please enter exactly {} coefficients.".format(degree + 1), None
coeffs = np.array(coeffs)
roots = np.roots(coeffs)
real_roots = [r.real for r in roots if np.isclose(r.imag, 0, atol=1e-5)]
# Plotting
x_vals = np.linspace(-10, 10, 400)
y_vals = np.polyval(coeffs, x_vals)
fig, ax = plt.subplots(figsize=(6, 4))
ax.plot(x_vals, y_vals, label="Polynomial Curve", color="blue")
ax.axhline(0, color='black', linewidth=0.5)
ax.axvline(0, color='black', linewidth=0.5)
for root in real_roots:
ax.scatter(root, 0, color="red", s=100)
ax.set_title("📈 Graph of the Polynomial")
ax.set_xlabel("x")
ax.set_ylabel("f(x)")
ax.grid(True, linestyle='--', linewidth=0.5)
ax.legend()
return "✅ Real roots: {}".format(real_roots if real_roots else "None found."), fig
except Exception as e:
return f"❌ Error: {e}", None
# Gradio UI
with gr.Blocks() as demo:
gr.Markdown("## 🎉 Welcome to MathWizz 🎉\nPlot and Analyze Your Polynomial")
degree = gr.Slider(label="Select Degree", minimum=1, maximum=10, value=2, step=1)
coeff_input = gr.Textbox(label="Enter Coefficients (space-separated)", placeholder="e.g. 1 -3 2")
output_text = gr.Textbox(label="Result")
output_plot = gr.Plot(label="Polynomial Curve")
submit = gr.Button("Plot Polynomial")
submit.click(fn=plot_polynomial,
inputs=[degree, coeff_input],
outputs=[output_text, output_plot])
demo.launch()
|