Update app.py

app.py

@@ -135,10 +135,21 @@ def load_model():
 
 model = load_model()
 
+# Initialize session state for restart functionality
+if 'restart_clicked' not in st.session_state:
+    st.session_state.restart_clicked = False
+
 # Create tabs for better organization
 tab1, tab2 = st.tabs(["Loan Application", "About the System"])
 
 with tab1:
+    # Reset all form values if restart was clicked
+    if st.session_state.restart_clicked:
+        st.session_state.restart_clicked = False  # Reset flag
+        
+        # Add any other reset logic here if needed
+        # Note: Streamlit will reset most inputs automatically on rerun
+    
     # Personal Information Section
     st.markdown('<div class="section-card"><h3>Personal Information</h3>', unsafe_allow_html=True)
     
@@ -278,10 +289,15 @@ with tab1:
     
     st.markdown('</div>', unsafe_allow_html=True)
     
-    # Prediction button with enhanced styling
-    st.markdown('<div style="padding: 1.5rem 0;">', unsafe_allow_html=True)
-    predict_button = st.button("Check Loan Approval Status")
-    st.markdown('</div>', unsafe_allow_html=True)
+    # Prediction and restart buttons
+    col1, col2 = st.columns([3, 1])
+    
+    with col1:
+        predict_button = st.button("Check Loan Approval Status", use_container_width=True)
+    
+    with col2:
+        restart_button = st.button("🔄 Restart", use_container_width=True, 
+                                 help="Reset all form fields and start over")
     
     def preprocess_input():
         # Convert categorical inputs to numerical format based on encoding reference
@@ -311,6 +327,11 @@ with tab1:
             credit_amount_interaction, income_term_ratio
         ]])
     
+    # Handle restart button click
+    if restart_button:
+        st.session_state.restart_clicked = True
+        st.experimental_rerun()
+    
     # Display prediction
     if predict_button:
         with st.spinner("Processing your application..."):