Fix

app.py

@@ -194,6 +194,9 @@ def main():
         )
 
         if uploaded_file is None:
+            st.session_state.submit = False
+            st.session_state.done = False
+
             with st.expander("Don't know what to upload? Try these examples", expanded=True):
                 with open('./data/examples/7DK2_AB_C.pdb', 'r') as f:
                     st.download_button(
@@ -209,11 +212,6 @@ def main():
                     )
                 st.text('Please upload the downloaded PDB file to run the demo.')
 
-    if 'submit' not in st.session_state:
-        st.session_state.submit = False
-    if 'done' not in st.session_state:
-        st.session_state.done = False
-
     # Step 1.2: Retrieve uploaded PDB
     if uploaded_file is not None:
         pdb_path = os.path.join(tempdir_path, 'structure.pdb')
@@ -241,7 +239,8 @@ def main():
             if docking:
                 st.warning('No antibodies detected. Will try to run docking (very slow).')
 
-            form = st.form('design_form')
+            # form = st.form('design_form')
+            form = st.container()
             with form:
                 if H_chain is None and L_chain is None:
                     # Antigen only
@@ -293,13 +292,26 @@ def main():
                 if not GPU_AVAILABLE:
                     st.warning('No GPU available. Sampling might be very slow.')
 
-                submit = st.form_submit_button('Run')
+                btn_placeholder = st.empty()
+                submit = btn_placeholder.button('Run', key="run_btn_real")
                 st.session_state.submit = st.session_state.submit or submit
                 if submit:
                     st.session_state.done = False
+                    btn_placeholder.empty()
 
     # Step 3: Prepare configuration and run design
     if uploaded_file is not None and st.session_state.submit:
+
+        with left_col:
+            output_display = st.empty()
+        output_display.code('[INFO] Your job has been submitted. Please wait...\n')
+
+        with right_col:
+            result_molecule_display = st.empty()
+            result_select_widget = st.empty()
+            result_table_display = st.empty()
+            result_download_btn = st.empty()
+
         config, config_path = get_config(
             save_dir = tempdir_path,
             mode = design_mode,
@@ -307,12 +319,6 @@ def main():
             num_samples = num_designs,
         )
 
-        with right_col:
-            result_molecule_display = st.empty()
-            result_select_widget = st.empty()
-            result_table_display = st.empty()
-            result_download_btn = st.empty()
-            output_display = st.empty()
         if not st.session_state.done:
             run_design(
                 pdb_path = renum_path,