Update pages/40_simpleline.py

pages/40_simpleline.py

@@ -1,6 +1,5 @@
 import streamlit as st
 import matplotlib.pyplot as plt
-import numpy as np
 
 # Initialize sliders
 x = st.slider('X position', min_value=0.0, max_value=10.0, value=5.0, step=0.1)
@@ -11,10 +10,10 @@ b = st.slider('B (intercept)', min_value=-10.0, max_value=10.0, value=0.0, step=
 y = w * x + b
 
 # Create the plot
-fig, ax = plt.subplots()
-ax.axhline(y=2, color='gray', linestyle='--')  # First horizontal line
-ax.axhline(y=-2, color='gray', linestyle='--')  # Second horizontal line
-ax.plot(x, y, 'ro')  # Plot the point
+fig, ax = plt.subplots(figsize=(6, 4))
+ax.axhline(y=2, color='blue', linestyle='--')  # First horizontal line
+ax.axhline(y=-2, color='blue', linestyle='--')  # Second horizontal line
+ax.plot([x], [y], 'ro')  # Plot the point
 
 # Set plot limits and labels
 ax.set_xlim(0, 10)
@@ -25,4 +24,4 @@ ax.set_title(f'y = {w} * x + {b}')
 ax.grid(True)
 
 # Display the plot in Streamlit
-st.pyplot(fig)
+st.pyplot(fig)