Try 2nd ver

app.py

@@ -215,7 +215,8 @@ def main():
             
             # Add a restart button to clear the session state and start over
             if st.button("Start New Morphing Project", key="restart"):
-                restart()
+                restart()  # Use the restart function
+                
         except Exception as e:
             st.error(f"Error preparing video for download: {e}")
         
@@ -348,11 +349,9 @@ def main():
     st.subheader("3. Generate Morphing Video")
     st.markdown("Once satisfied with your inputs, click below to start the process.")
 
-    # New checkbox for SLAB execution toggle
-    # using_slab = st.checkbox("Using SLAB GPU Cluster?", value=False, help="If enabled, the pipeline command will be prefixed with SLAB cluster execution parameters.", disabled=st.session_state.processing)
-
     # 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"
@@ -365,7 +364,7 @@ def main():
                 # Set the processing state to True to prevent multiple runs
                 st.session_state.processing = True
                 
-                # Instead of using /tmp, create a folder in the repo for temporary processing.
+                # Create a temporary folder for processing
                 temp_dir = create_temp_folder()
                 st.session_state.temp_dir = temp_dir
                 
@@ -415,33 +414,37 @@ def main():
                     # Add film recursion parameter
                     cmd.extend(["--film_num_recursions", str(film_recursions)])
                     
-                    subprocess.run(cmd, check=True)
-                    
-                    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
-                    
-                    if not video_found:
-                        possible_outputs = [f for f in os.listdir(output_dir) if f.endswith(".mp4")]
+                    # Run the morphing process
+                    try:
+                        subprocess.run(cmd, check=True)
+                        
+                        # Check for output video
+                        video_found = False
+                        possible_outputs = [f for f in os.listdir(film_output_dir) if f.endswith(".mp4")]
                         if possible_outputs:
-                            final_video_path = os.path.join(output_dir, possible_outputs[0])
+                            final_video_path = os.path.join(film_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
-                        st.rerun()
-                    else:
-                        st.error("No output video was generated. Check logs for details.")
+                        
+                        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
+                            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 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