Update app.py to freeze on running once

app.py

@@ -157,22 +157,6 @@ def main():
             padding-left: 1rem;
             margin-left: 1rem;
         }
-        /* Processing overlay */
-        .processing-overlay {
-            position: fixed;
-            top: 0;
-            left: 0;
-            right: 0;
-            bottom: 0;
-            background-color: rgba(0, 0, 0, 0.7);
-            z-index: 1000;
-            display: flex;
-            flex-direction: column;
-            justify-content: center;
-            align-items: center;
-            color: white;
-            font-size: 1.5rem;
-        }
         </style>
         """,
         unsafe_allow_html=True
@@ -231,24 +215,13 @@ def main():
             
             # Add a restart button to clear the session state and start over
             if st.button("Start New Morphing Project", key="restart"):
-                restart()  # Use the restart function
-                
+                restart()
         except Exception as e:
             st.error(f"Error preparing video for download: {e}")
         
         # Early return to not show the input forms again when we have a result
         return
 
-    # Show processing indicator if currently processing
-    if st.session_state.processing:
-        progress_container = st.container()
-        with progress_container:
-            st.info("Processing your morphing request. Please wait...")
-            progress_bar = st.progress(0)
-            status_text = st.empty()
-            status_text.text("Initializing pipeline. Please do not close this page.")
-        st.warning("⚠️ Please wait for the current process to complete. Do not press the run button again.")
-        
     # ---------------- SECTION 1: IMAGE & PROMPT INPUTS ----------------
     st.subheader("1. Upload Source Images & Prompts")
     st.markdown("**Note:** Your uploaded images must be of similar topology and same size to achieve the best results.")
@@ -380,7 +353,6 @@ def main():
 
     # Create a container for the run button and its status messages
     run_container = st.container()
-    
     with run_container:
         # Button text changes based on processing state
         button_text = "Processing... Please Wait" if st.session_state.processing else "Run Morphing Pipeline"
@@ -393,11 +365,6 @@ def main():
                 # Set the processing state to True to prevent multiple runs
                 st.session_state.processing = True
                 
-                # Create progress indicators
-                progress_bar = st.progress(0)
-                status_text = st.empty()
-                status_text.text("Initializing pipeline. This may take a few minutes...")
-                
                 # Instead of using /tmp, create a folder in the repo for temporary processing.
                 temp_dir = create_temp_folder()
                 st.session_state.temp_dir = temp_dir
@@ -448,57 +415,33 @@ def main():
                     # Add film recursion parameter
                     cmd.extend(["--film_num_recursions", str(film_recursions)])
                     
-                    # If SLAB execution is enabled, prepend the srun command prefix.
-                    # if using_slab:
-                    #     slab_prefix = [
-                    #         "srun", "-p", "rtx3090_slab", "-w", "slabgpu05", "--gres=gpu:1",
-                    #         "--job-name=test", "--kill-on-bad-exit=1"
-                    #     ]
-                    #     cmd = slab_prefix + cmd
+                    subprocess.run(cmd, check=True)
                     
-                    # Update progress status for preparing phase
-                    progress_bar.progress(10)
-                    status_text.text("Preparing images and configuration...")
+                    possible_outputs = [f for f in os.listdir(film_output_dir) if f.endswith(".mp4")]
+                    video_found = False
+                    if possible_outputs:
+                        final_video_path = os.path.join(film_output_dir, possible_outputs[0])
+                        video_found = True
                     
-                    # Run the morphing process
-                    try:
-                        subprocess.run(cmd, check=True)
-                        
-                        # Update progress
-                        progress_bar.progress(90)
-                        status_text.text("Processing complete! Preparing results...")
-                        
-                        # Check for output video
-                        video_found = False
-                        possible_outputs = [f for f in os.listdir(film_output_dir) if f.endswith(".mp4")]
+                    if not video_found:
+                        possible_outputs = [f for f in os.listdir(output_dir) if f.endswith(".mp4")]
                         if possible_outputs:
-                            final_video_path = os.path.join(film_output_dir, possible_outputs[0])
+                            final_video_path = os.path.join(output_dir, possible_outputs[0])
                             video_found = True
-                        
-                        if not video_found:
-                            possible_outputs = [f for f in os.listdir(output_dir) if f.endswith(".mp4")]
-                            if possible_outputs:
-                                final_video_path = os.path.join(output_dir, possible_outputs[0])
-                                video_found = True
-                        
-                        if video_found:
-                            st.session_state.final_video_path = final_video_path
-                            st.session_state.process_complete = True
-                            
-                            # Set progress to 100%
-                            progress_bar.progress(100)
-                            
-                            st.rerun()
-                        else:
-                            status_text.warning("No output video was generated. Check logs for details.")
-                            st.session_state.processing = False
-                            st.session_state.process_complete = False
-                            
-                    except subprocess.CalledProcessError as e:
-                        st.error(f"Error running morphing pipeline: {e}")
+                    
+                    if video_found:
+                        st.session_state.final_video_path = final_video_path
+                        st.session_state.process_complete = True
+                        st.rerun()
+                    else:
+                        st.error("No output video was generated. Check logs for details.")
                         st.session_state.processing = False
                         st.session_state.process_complete = False
-                    
+                        
+                except subprocess.CalledProcessError as e:
+                    st.error(f"Error running morphing pipeline: {e}")
+                    st.session_state.processing = False
+                    st.session_state.process_complete = False
                 except Exception as e:
                     st.error(f"An error occurred during processing: {e}")
                     st.session_state.processing = False