Update app.py

app.py

@@ -1207,44 +1207,44 @@ elif app_mode == "Predictions":
         except Exception as e:
             st.error(f"Batch prediction failed: {e}")
 
-    # Prediction Analysis
-    st.subheader("📊 Prediction Analysis")
-    if st.checkbox("Analyze Predictions"):
-        try:
-            y_pred = model.predict(st.session_state.X_test_selected)
-            y_test = st.session_state.y_test
+# Prediction Analysis
+st.subheader("📊 Prediction Analysis")
+if st.checkbox("Analyze Predictions"):
+    try:
+        y_pred = model.predict(st.session_state.X_test_selected)
+        y_test = st.session_state.y_test
 
-            if hasattr(model, 'predict'):
-                fig, ax = plt.subplots()
-                ax.scatter(y_test, y_pred)
-                ax.plot([y_test.min(), y_test.max()], [y_test.min(), y_test.max()], 'k--', lw=2)
-                ax.set_xlabel('Actual')
-                ax.set_ylabel('Predicted')
-                ax.set_title('Actual vs Predicted')
-                st.pyplot(fig)
-            else:
-                conf_matrix = confusion_matrix(y_test, y_pred)
-                fig, ax = plt.subplots()
-                sns.heatmap(conf_matrix, annot=True, fmt='d', cmap='Blues', ax=ax)
-                ax.set_xlabel('Predicted Labels')
-                ax.set_ylabel('True Labels')
-                ax.set_title('Confusion Matrix')
-                st.pyplot(fig)
-        except Exception as e:
-            st.error(f"Prediction analysis failed: {e}")
-
-    # Prediction Export
-    st.subheader("💾 Export Predictions")
-    if st.button("Export Predictions as PDF"):
-        try:
-            from fpdf import FPDF
-            pdf = FPDF()
-            pdf.add_page()
-            pdf.set_font("Arial", size=12)
-            pdf.cell(200, 10, txt="Predictions Report", ln=True, align='C')
-            pdf.cell(200, 10, txt=f"Model Type: {type(model).__name__}", ln=True)
-            pdf.cell(200, 10, txt=f"Problem Type: {'Regression' if hasattr(model, 'predict') else 'Classification'}", ln=True)
-            pdf.output("predictions_report.pdf")
-            st.success("Predictions exported successfully!")
-except Exception as e:
-    st.error(f"An unexpected error occurred: {e}")
+        if hasattr(model, 'predict'):
+            fig, ax = plt.subplots()
+            ax.scatter(y_test, y_pred)
+            ax.plot([y_test.min(), y_test.max()], [y_test.min(), y_test.max()], 'k--', lw=2)
+            ax.set_xlabel('Actual')
+            ax.set_ylabel('Predicted')
+            ax.set_title('Actual vs Predicted')
+            st.pyplot(fig)
+        else:
+            conf_matrix = confusion_matrix(y_test, y_pred)
+            fig, ax = plt.subplots()
+            sns.heatmap(conf_matrix, annot=True, fmt='d', cmap='Blues', ax=ax)
+            ax.set_xlabel('Predicted Labels')
+            ax.set_ylabel('True Labels')
+            ax.set_title('Confusion Matrix')
+            st.pyplot(fig)
+    except Exception as e:
+        st.error(f"Prediction analysis failed: {e}")
+
+# Prediction Export
+st.subheader("💾 Export Predictions")
+if st.button("Export Predictions as PDF"):
+    try:
+        from fpdf import FPDF
+        pdf = FPDF()
+        pdf.add_page()
+        pdf.set_font("Arial", size=12)
+        pdf.cell(200, 10, txt="Predictions Report", ln=True, align='C')
+        pdf.cell(200, 10, txt=f"Model Type: {type(model).__name__}", ln=True)
+        pdf.cell(200, 10, txt=f"Problem Type: {'Regression' if hasattr(model, 'predict') else 'Classification'}", ln=True)
+        pdf.output("predictions_report.pdf")
+        st.success("Predictions exported successfully!")
+    except Exception as e:
+        st.error(f"An unexpected error occurred: {e}")