Update src/streamlit_app.py

src/streamlit_app.py

@@ -7,18 +7,18 @@ from sklearn.metrics import mean_squared_error
 
 st.set_page_config(page_title="Linear Regression Playground", layout="centered")
 
-# FIX: ensure equation always displays fully
+# FIX: make equation always fully visible + box formatting
 st.markdown("""
 <style>
 .eq-box {
-    width: 100%;
-    background-color: #f7f7f9;
-    padding: 18px;
-    border-radius: 10px;
     border: 2px solid #333;
+    border-radius: 8px;
+    background: #ffffff;
+    padding: 14px;
+    width: 100%;
     font-size: 22px;
     text-align: center;
-    margin-top: 12px;
+    margin-top: 14px;
 }
 </style>
 """, unsafe_allow_html=True)
@@ -26,7 +26,9 @@ st.markdown("""
 st.title("📉 Linear Regression Playground (2D & 3D)")
 st.write("Experiment with regression, noise, slope, intercept — and visualize the model!")
 
+# ------------------------------------
 # Sidebar Controls
+# ------------------------------------
 st.sidebar.header("⚙️ Controls")
 
 mode = st.sidebar.radio("Choose Mode", ["2D Regression", "3D Regression"])
@@ -47,7 +49,10 @@ if mode != st.session_state.current_mode:
     st.session_state.trained = False
     st.session_state.current_mode = mode
 
+
+# ------------------------------------
 # Generate dataset
+# ------------------------------------
 if train_btn:
 
     with st.spinner("⏳ Training model..."):
@@ -82,12 +87,15 @@ if train_btn:
 
         st.session_state.trained = True
 
+
+# ------------------------------------
 # Visualization
+# ------------------------------------
 if st.session_state.trained:
 
     st.success("🎉 Model trained successfully!")
 
-    # -------- 2D Regression --------
+    # ----------------- 2D Regression -----------------
     if mode == "2D Regression":
         X, y, y_pred, mse, model = st.session_state.data
 
@@ -104,17 +112,12 @@ if st.session_state.trained:
         with col2:
             st.metric("MSE", f"{mse:.4f}")
 
-            # FULL BOX EQUATION
-            st.markdown(
-                f"""
-                <div class="eq-box">
-                    $$y = {model.coef_[0]:.3f}x + {model.intercept_:.3f}$$
-                </div>
-                """,
-                unsafe_allow_html=True
-            )
-
-    # -------- 3D Regression --------
+            # FIXED FULL EQUATION BOX
+            equation = rf"y = {model.coef_[0]:.3f}x + {model.intercept_:.3f}"
+            st.markdown(f"<div class='eq-box'>${equation}$</div>", unsafe_allow_html=True)
+
+
+    # ----------------- 3D Regression -----------------
     else:
         X1, X2, Z, Z_pred, mse, model = st.session_state.data
 
@@ -127,8 +130,10 @@ if st.session_state.trained:
                 ax = fig.add_subplot(111, projection="3d")
 
                 idx = np.random.choice(len(Z.ravel()), min(350, len(Z.ravel())), replace=False)
-                ax.scatter(X1.ravel()[idx], X2.ravel()[idx], Z.ravel()[idx],
-                           color="orange", alpha=0.25, s=8)
+                ax.scatter(
+                    X1.ravel()[idx], X2.ravel()[idx], Z.ravel()[idx],
+                    color="orange", alpha=0.25, s=8
+                )
 
                 ax.plot_surface(X1, X2, Z_pred, alpha=0.75, color="blue")
                 ax.set_title("3D Linear Regression")
@@ -159,15 +164,10 @@ if st.session_state.trained:
             b = model.coef_[1]
             c = model.intercept_
 
-            # FULL BOX EQUATION (3D)
-            st.markdown(
-                f"""
-                <div class="eq-box">
-                    $$z = {a:.3f}x_1 + {b:.3f}x_2 + {c:.3f}$$
-                </div>
-                """,
-                unsafe_allow_html=True
-            )
+            equation3d = rf"z = {a:.3f}x_1 + {b:.3f}x_2 + {c:.3f}"
+
+            # FIXED FULL EQUATION BOX
+            st.markdown(f"<div class='eq-box'>${equation3d}$</div>", unsafe_allow_html=True)
 
 else:
     st.info("Click **Generate & Train Model** to begin.")