Update src/streamlit_app.py

src/streamlit_app.py

@@ -102,7 +102,7 @@ def transform_data(X, y, transformations):
 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", 
+                       subplot_titles=("Residuals vs Fitted", 
                                       "Q-Q Plot", 
                                       "Residual Histogram", 
                                       "Residuals vs Order"))
@@ -114,16 +114,22 @@ def plot_residual_analysis(y_true, y_pred, residuals):
     )
     fig.add_hline(y=0, line_dash="dot", row=1, col=1)
     
-    # Q-Q Plot
-    qq = sm.qqplot(residuals, line='s', fit=True)
-    plt.close()
+    # Q-Q Plot - Proper implementation
+    qq = sm.ProbPlot(residuals)
+    theoretical = qq.theoretical_quantiles
+    sample = qq.sample_quantiles
+    
     fig.add_trace(
-        go.Scatter(x=qq.theory_quantiles, y=qq.sample_quantiles, mode='markers'),
+        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=qq.theory_quantiles, y=qq.theory_quantiles*qq.slope + qq.intercept, 
-                  mode='lines', line=dict(color='red')),
+        go.Scatter(x=line_x, y=line_y, mode='lines', 
+                  line=dict(color='red'), name='Reference Line'),
         row=1, col=2
     )
     
@@ -135,12 +141,18 @@ 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'),
+        go.Scatter(x=np.arange(len(residuals)), y=residuals, 
+                  mode='lines+markers', name='Residuals'),
         row=2, col=2
     )
     fig.add_hline(y=0, line_dash="dot", row=2, col=2)
     
-    fig.update_layout(height=800, showlegend=False, template='plotly_white')
+    fig.update_layout(
+        height=800, 
+        showlegend=False, 
+        template='plotly_white',
+        margin=dict(l=50, r=50, b=50, t=50)
+    )
     st.plotly_chart(fig, use_container_width=True)
 
 def main():