Update pages/40_simpleline.py

pages/40_simpleline.py

@@ -10,28 +10,24 @@ 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(2, 1, figsize=(6, 4), gridspec_kw={'height_ratios': [1, 1]})
-fig.subplots_adjust(hspace=0.5)
+fig, ax = plt.subplots(figsize=(6, 4))
+fig.subplots_adjust(top=0.8, bottom=0.1)
 
-# Plot for the x-axis
-ax[0].hlines(0, 0, 10, color='blue', linestyle='--')  # X-axis
-ax[0].plot(x, 0, 'ro')  # Plot the x point
-ax[0].set_xlim(0, 10)
-ax[0].set_ylim(-1, 1)
-ax[0].set_xlabel('X-axis')
-ax[0].set_yticks([])  # Hide y-axis ticks
-ax[0].set_title('X-axis Position')
-ax[0].grid(True)
+# Plot for the y-axis (top line)
+ax.hlines(1, -10, 10, color='blue', linestyle='--')  # Y-axis
+ax.plot(y, 1, 'ro')  # Plot the y point
 
-# Plot for the y-axis
-ax[1].hlines(0, -10, 10, color='blue', linestyle='--')  # Y-axis
-ax[1].plot(y, 0, 'ro')  # Plot the y point
-ax[1].set_xlim(-10, 10)
-ax[1].set_ylim(-1, 1)
-ax[1].set_xlabel('Y-axis')
-ax[1].set_yticks([])  # Hide y-axis ticks
-ax[1].set_title(f'y = {w} * x + {b}')
-ax[1].grid(True)
+# Plot for the x-axis (bottom line)
+ax.hlines(-1, 0, 10, color='blue', linestyle='--')  # X-axis
+ax.plot(x, -1, 'ro')  # Plot the x point
+
+# Set plot limits and labels
+ax.set_xlim(-10, 10)
+ax.set_ylim(-2, 2)
+ax.set_xlabel('X-axis and Y-axis')
+ax.set_yticks([])  # Hide y-axis ticks
+ax.set_title(f'y = {w} * x + {b}')
+ax.grid(True)
 
 # Display the plot in Streamlit
 st.pyplot(fig)