Update src/streamlit_app.py

src/streamlit_app.py

@@ -103,9 +103,9 @@ def plot_residual_analysis(y_true, y_pred, residuals):
     """Create comprehensive residual analysis plots"""
     fig = make_subplots(rows=2, cols=2,
                        subplot_titles=("Residuals vs Fitted", 
-                                      "Q-Q Plot", 
-                                      "Residual Histogram", 
-                                      "Residuals vs Order"))
+                                     "Q-Q Plot", 
+                                     "Residual Histogram", 
+                                     "Residuals vs Order"))
     
     # Residuals vs Fitted
     fig.add_trace(
@@ -119,17 +119,20 @@ def plot_residual_analysis(y_true, y_pred, residuals):
     theoretical = qq.theoretical_quantiles
     sample = qq.sample_quantiles
     
+    # Calculate regression line for Q-Q plot
+    slope, intercept = np.polyfit(theoretical, sample, 1)
+    line_x = np.array([theoretical.min(), theoretical.max()])
+    line_y = slope * line_x + intercept
+    
     fig.add_trace(
         go.Scatter(x=theoretical, y=sample, mode='markers', name='Q-Q Points'),
         row=1, col=2
     )
     
     # Add reference line
-    line_x = np.array([theoretical.min(), theoretical.max()])
-    line_y = qq.slope * line_x + qq.intercept
     fig.add_trace(
         go.Scatter(x=line_x, y=line_y, mode='lines', 
-                  line=dict(color='red'), name='Reference Line'),
+                 line=dict(color='red'), name='Reference Line'),
         row=1, col=2
     )
     
@@ -142,7 +145,7 @@ def plot_residual_analysis(y_true, y_pred, residuals):
     # Residuals vs Order
     fig.add_trace(
         go.Scatter(x=np.arange(len(residuals)), y=residuals, 
-                  mode='lines+markers', name='Residuals'),
+                 mode='lines+markers', name='Residuals'),
         row=2, col=2
     )
     fig.add_hline(y=0, line_dash="dot", row=2, col=2)