Create 3 page sys, lastt

app.py

@@ -46,12 +46,9 @@ def create_temp_folder():
     return run_folder
 
 def main():
-    # Initialize session state for tracking processing status
-    if 'processing' not in st.session_state:
-        st.session_state.processing = False
-    
-    if 'process_complete' not in st.session_state:
-        st.session_state.process_complete = False
+    # Initialize session state variables
+    if 'page' not in st.session_state:
+        st.session_state.page = 'input'  # States: 'input', 'processing', 'result'
     
     if 'temp_dir' not in st.session_state:
         st.session_state.temp_dir = None
@@ -59,18 +56,20 @@ def main():
     if 'final_video_path' not in st.session_state:
         st.session_state.final_video_path = None
     
-    # Initialize the download container in session state
-    if 'show_download' not in st.session_state:
-        st.session_state.show_download = False
+    if 'process_started' not in st.session_state:
+        st.session_state.process_started = False
+        
+    # Function to switch to processing page and start morphing
+    def start_processing():
+        st.session_state.page = 'processing'
+        st.session_state.process_started = False  # Will be set to True when processing starts
         
-    # Restart function to reset session state
-    def restart():
-        st.session_state.processing = False
-        st.session_state.process_complete = False 
+    # Function to return to input page
+    def return_to_input():
+        st.session_state.page = 'input'
         st.session_state.temp_dir = None
         st.session_state.final_video_path = None
-        st.session_state.show_download = False
-        st.rerun()  # Use st.rerun() instead of experimental_rerun
+        st.session_state.process_started = False
 
     # ---------------- CUSTOM CSS FOR A PROFESSIONAL, DARK THEME ----------------
     st.markdown(
@@ -144,31 +143,20 @@ def main():
             transform: scale(1.02);
             box-shadow: 0 0 20px rgba(142,68,173,0.8), 0 0 30px rgba(114,45,145,0.6);
         }
-        /* Disabled button styling */
-        div.stButton > button:disabled {
-            background-image: linear-gradient(45deg, #6b6b6b, #4a4a4a);
-            box-shadow: none;
-            cursor: not-allowed;
-            opacity: 0.7;
-        }
-        /* File uploader label styling */
-        .stFileUploader label {
-            font-size: 1rem;
-            color: #f1f1f1;
+        /* Processing animation */
+        .processing-container {
+            text-align: center;
+            padding: 3rem 0;
         }
-        /* Advanced Options divider styling */
-        .right-column-divider {
-            border-left: 2px solid #f1f1f1;
-            padding-left: 1rem;
-            margin-left: 1rem;
+        .processing-text {
+            font-size: 1.8rem;
+            margin-bottom: 2rem;
+            animation: pulse 2s infinite;
         }
-        /* Results section styling */
-        .results-section {
-            margin-top: 2rem;
-            padding: 1.5rem;
-            background-color: rgba(0, 0, 0, 0.2);
-            border-radius: 8px;
-            border: 1px solid rgba(255, 255, 255, 0.1);
+        @keyframes pulse {
+            0% { opacity: 0.6; }
+            50% { opacity: 1; }
+            100% { opacity: 0.6; }
         }
         </style>
         """,
@@ -182,7 +170,7 @@ def main():
         st.error("Please make sure all required scripts are in the same directory as this Streamlit app.")
         return
 
-    # ---------------- HEADER & LOGO ----------------
+    # Load logo for all pages
     logo_path = "metamorphLogo_nobg.png"
     if os.path.exists(logo_path):
         try:
@@ -200,302 +188,340 @@ def main():
             st.warning(f"Logo could not be loaded: {e}")
 
     st.markdown("<h1 class='header-title'>Metamorph Web App</h1>", unsafe_allow_html=True)
-    st.markdown(
-        """
-        <p style='text-align: center; font-size: 1.1rem;'>
-            DiffMorpher is used for keyframe generation by default, with FILM for interpolation.
-            Optionally, you can enable LCM-LoRA for accelerated inference (with slight decrease in quality).
-            Upload two images, optionally provide textual prompts, and fine-tune the settings to create a smooth, high-quality morphing video.
-        </p>
-        <hr>
-        """,
-        unsafe_allow_html=True
-    )
 
-    # Create a persistent container for the results section
-    results_container = st.container()
+    # =============== INPUT PAGE ===============
+    if st.session_state.page == 'input':
+        st.markdown(
+            """
+            <p style='text-align: center; font-size: 1.1rem;'>
+                DiffMorpher is used for keyframe generation by default, with FILM for interpolation.
+                Optionally, you can enable LCM-LoRA for accelerated inference (with slight decrease in quality).
+                Upload two images, optionally provide textual prompts, and fine-tune the settings to create a smooth, high-quality morphing video.
+            </p>
+            <hr>
+            """,
+            unsafe_allow_html=True
+        )
 
-    # Show results if processing is complete
-    if st.session_state.process_complete and st.session_state.final_video_path is not None:
-        with results_container:
-            st.markdown("<div class='results-section'>", unsafe_allow_html=True)
-            st.success("Morphing complete! 🎉")
-            try:
-                with open(st.session_state.final_video_path, "rb") as f:
-                    video_bytes = f.read()
-                    st.session_state.video_bytes = video_bytes  # Store video bytes in session state
-                    st.video(video_bytes)
-                    st.download_button(
-                        "Download Morphing Video",
-                        data=video_bytes,
-                        file_name="metamorph_result.mp4",
-                        mime="video/mp4"
-                    )
-                st.session_state.show_download = True
-                
-                # Add a restart button to clear the session state and start over
-                if st.button("Start New Morphing Project", key="restart"):
-                    restart()
-                    
-            except Exception as e:
-                st.error(f"Error preparing video for download: {e}")
-            st.markdown("</div>", unsafe_allow_html=True)
+        # ---------------- 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.")
+        col_imgA, col_imgB = st.columns(2)
+        with col_imgA:
+            st.markdown("#### Image A")
+            uploaded_image_A = st.file_uploader("Upload your first image", type=["png", "jpg", "jpeg"], key="imgA")
+            if uploaded_image_A is not None:
+                st.image(uploaded_image_A, caption="Preview - Image A", use_container_width=True)
+            prompt_A = st.text_input("Short Description for Image A (optional)", value="", key="promptA", 
+                                  help="For added interpolation between the two descriptions")
+        with col_imgB:
+            st.markdown("#### Image B")
+            uploaded_image_B = st.file_uploader("Upload your second image", type=["png", "jpg", "jpeg"], key="imgB")
+            if uploaded_image_B is not None:
+                st.image(uploaded_image_B, caption="Preview - Image B", use_container_width=True)
+            prompt_B = st.text_input("Short Description for Image B (optional)", value="", key="promptB",
+                                  help="For added interpolation between the two descriptions")
 
-    # ---------------- 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.")
-    col_imgA, col_imgB = st.columns(2)
-    with col_imgA:
-        st.markdown("#### Image A")
-        uploaded_image_A = st.file_uploader("Upload your first image", type=["png", "jpg", "jpeg"], key="imgA", 
-                                         disabled=st.session_state.processing)
-        if uploaded_image_A is not None:
-            st.image(uploaded_image_A, caption="Preview - Image A", use_container_width=True)
-        prompt_A = st.text_input("Short Description for Image A (optional)", value="", key="promptA", 
-                              help="For added interpolation between the two descriptions",
-                              disabled=st.session_state.processing)
-    with col_imgB:
-        st.markdown("#### Image B")
-        uploaded_image_B = st.file_uploader("Upload your second image", type=["png", "jpg", "jpeg"], key="imgB",
-                                         disabled=st.session_state.processing)
-        if uploaded_image_B is not None:
-            st.image(uploaded_image_B, caption="Preview - Image B", use_container_width=True)
-        prompt_B = st.text_input("Short Description for Image B (optional)", value="", key="promptB",
-                              help="For added interpolation between the two descriptions",
-                              disabled=st.session_state.processing)
+        st.markdown("<hr>", unsafe_allow_html=True)
 
-    st.markdown("<hr>", unsafe_allow_html=True)
+        # ---------------- SECTION 2: CONFIGURE MORPHING PIPELINE ----------------
+        st.subheader("2. Configure Morphing Pipeline")
+        st.markdown(
+            """
+            <p style="font-size: 1rem;">
+                Select a preset below to automatically adjust quality and inference time. 
+                If you choose <strong>Custom ⚙️</strong>, the advanced settings will automatically expand so you can fine-tune the configuration.
+            </p>
+            """,
+            unsafe_allow_html=True
+        )
 
-    # ---------------- SECTION 2: CONFIGURE MORPHING PIPELINE ----------------
-    st.subheader("2. Configure Morphing Pipeline")
-    st.markdown(
-        """
-        <p style="font-size: 1rem;">
-            Select a preset below to automatically adjust quality and inference time. 
-            If you choose <strong>Custom ⚙️</strong>, the advanced settings will automatically expand so you can fine-tune the configuration.
-        </p>
-        """,
-        unsafe_allow_html=True
-    )
+        # Preset Options (Dropdown)
+        st.markdown("**Preset Options**")
+        preset_option = st.selectbox(
+            "Select a preset for quality and inference time",
+            options=[
+                "Maximum quality, longest inference time 🏆",
+                "Medium quality, medium inference time ⚖️",
+                "Low quality, shortest inference time ⚡",
+                "Creative morph 🎨",
+                "Custom ⚙️"
+            ],
+            index=0,
+            label_visibility="collapsed"
+        )
 
-    # Preset Options (Dropdown)
-    st.markdown("**Preset Options**")
-    preset_option = st.selectbox(
-        "Select a preset for quality and inference time",
-        options=[
-            "Maximum quality, longest inference time 🏆",
-            "Medium quality, medium inference time ⚖️",
-            "Low quality, shortest inference time ⚡",
-            "Creative morph 🎨",
-            "Custom ⚙️"
-        ],
-        index=0,
-        label_visibility="collapsed",
-        disabled=st.session_state.processing
-    )
+        # Determine preset defaults based on selection
+        if preset_option.startswith("Maximum quality"):
+            preset_model = "Base Stable Diffusion V2-1"
+            preset_film = True
+            preset_lcm = False
+        elif preset_option.startswith("Medium quality"):
+            preset_model = "Base Stable Diffusion V2-1"
+            preset_film = False
+            preset_lcm = False
+        elif preset_option.startswith("Low quality"):
+            preset_model = "Base Stable Diffusion V1-5"
+            preset_film = False
+            preset_lcm = True
+        elif preset_option.startswith("Creative morph"):
+            preset_model = "Dreamshaper-7 (fine-tuned SD V1-5)"
+            preset_film = True
+            preset_lcm = True
+        else:
+            # "Custom"
+            preset_model = None
+            preset_film = None
+            preset_lcm = None
 
-    # Determine preset defaults based on selection
-    if preset_option.startswith("Maximum quality"):
-        preset_model = "Base Stable Diffusion V2-1"
-        preset_film = True
-        preset_lcm = False
-    elif preset_option.startswith("Medium quality"):
-        preset_model = "Base Stable Diffusion V2-1"
-        preset_film = False
-        preset_lcm = False
-    elif preset_option.startswith("Low quality"):
-        preset_model = "Base Stable Diffusion V1-5"
-        preset_film = False
-        preset_lcm = True
-    elif preset_option.startswith("Creative morph"):
-        preset_model = "Dreamshaper-7 (fine-tuned SD V1-5)"
-        preset_film = True
-        preset_lcm = True
-    else:
-        # "Custom"
-        preset_model = None
-        preset_film = None
-        preset_lcm = None
+        advanced_expanded = True if preset_option.endswith("⚙️") else False
 
-    advanced_expanded = True if preset_option.endswith("⚙️") else False
+        # Advanced Options for fine-tuning
+        with st.expander("Advanced Options", expanded=advanced_expanded):
+            options_list = [
+                "Base Stable Diffusion V1-5",
+                "Dreamshaper-7 (fine-tuned SD V1-5)",
+                "Base Stable Diffusion V2-1"
+            ]
+            default_model = preset_model if preset_model is not None else "Base Stable Diffusion V1-5"
+            default_index = options_list.index(default_model)
+            model_option = st.selectbox("Select Model Card", options=options_list, index=default_index)
+            
+            col_left, col_right = st.columns(2)
+            # Left Column: Keyframe Generator Parameters
+            with col_left:
+                st.markdown("##### Keyframe Generator Parameters")
+                num_frames = st.number_input("Number of keyframes (2–50)", min_value=2, max_value=50, value=16)
+                lcm_default = preset_lcm if preset_lcm is not None else False
+                enable_lcm_lora = st.checkbox(
+                    "Enable LCM-LoRA",
+                    value=lcm_default,
+                    help="Accelerates inference with slight quality decrease"
+                )
+                use_adain = st.checkbox("Use AdaIN", value=True, help="Adaptive Instance Normalization for improved generation")
+                use_reschedule = st.checkbox("Use reschedule sampling", value=True, help="Better sampling strategy")
+            
+            # Right Column: Inter-frame Interpolator Parameters (FILM)
+            with col_right:
+                st.markdown("<div class='right-column-divider'>", unsafe_allow_html=True)
+                st.markdown("##### Inter-frame Interpolator Parameters")
+                default_use_film = preset_film if preset_film is not None else True
+                use_film = st.checkbox("Use FILM interpolation", value=default_use_film, help="Frame Interpolation for Large Motion - creates smooth transitions")
+                film_recursions = st.number_input("FILM recursion passes (1–6)", min_value=1, max_value=6, value=3, 
+                                                help="Higher values create more intermediate frames (smoother but slower)")
+                # Set default FPS based on whether FILM is enabled
+                default_fps = 30 if use_film else 8
+                output_fps = st.number_input("Output FPS (1–120)", min_value=1, max_value=120, value=default_fps, 
+                                        help="Output video frames per second")
+                st.markdown("</div>", unsafe_allow_html=True)
 
-    # Advanced Options for fine-tuning
-    with st.expander("Advanced Options", expanded=advanced_expanded):
-        options_list = [
-            "Base Stable Diffusion V1-5",
-            "Dreamshaper-7 (fine-tuned SD V1-5)",
-            "Base Stable Diffusion V2-1"
-        ]
-        default_model = preset_model if preset_model is not None else "Base Stable Diffusion V1-5"
-        default_index = options_list.index(default_model)
-        model_option = st.selectbox("Select Model Card", options=options_list, index=default_index, disabled=st.session_state.processing)
-        
-        col_left, col_right = st.columns(2)
-        # Left Column: Keyframe Generator Parameters
-        with col_left:
-            st.markdown("##### Keyframe Generator Parameters")
-            num_frames = st.number_input("Number of keyframes (2–50)", min_value=2, max_value=50, value=16, disabled=st.session_state.processing)
-            lcm_default = preset_lcm if preset_lcm is not None else False
-            enable_lcm_lora = st.checkbox(
-                "Enable LCM-LoRA",
-                value=lcm_default,
-                help="Accelerates inference with slight quality decrease",
-                disabled=st.session_state.processing
-            )
-            use_adain = st.checkbox("Use AdaIN", value=True, help="Adaptive Instance Normalization for improved generation", disabled=st.session_state.processing)
-            use_reschedule = st.checkbox("Use reschedule sampling", value=True, help="Better sampling strategy", disabled=st.session_state.processing)
-        
-        # Right Column: Inter-frame Interpolator Parameters (FILM)
-        with col_right:
-            st.markdown("<div class='right-column-divider'>", unsafe_allow_html=True)
-            st.markdown("##### Inter-frame Interpolator Parameters")
-            default_use_film = preset_film if preset_film is not None else True
-            use_film = st.checkbox("Use FILM interpolation", value=default_use_film, help="Frame Interpolation for Large Motion - creates smooth transitions", disabled=st.session_state.processing)
-            film_recursions = st.number_input("FILM recursion passes (1–6)", min_value=1, max_value=6, value=3, 
-                                             help="Higher values create more intermediate frames (smoother but slower)",
-                                             disabled=st.session_state.processing)
-            # Set default FPS based on whether FILM is enabled
-            default_fps = 30 if use_film else 8
-            output_fps = st.number_input("Output FPS (1–120)", min_value=1, max_value=120, value=default_fps, 
-                                    help="Output video frames per second",
-                                    disabled=st.session_state.processing)
-            st.markdown("</div>", unsafe_allow_html=True)
+        st.markdown("<hr>", unsafe_allow_html=True)
 
-    st.markdown("<hr>", unsafe_allow_html=True)
+        # ---------------- SECTION 3: EXECUTE MORPH PIPELINE ----------------
+        st.subheader("3. Generate Morphing Video")
+        st.markdown("Once satisfied with your inputs, click below to start the process.")
 
-    # ---------------- SECTION 3: EXECUTE MORPH PIPELINE ----------------
-    st.subheader("3. Generate Morphing Video")
-    st.markdown("Once satisfied with your inputs, click below to start the process.")
+        # Create a container for the run button
+        run_container = st.container()
+        
+        with run_container:
+            # Save values to session state so we can access them in the processing page
+            if st.button("Run Morphing Pipeline", key="run_pipeline"):
+                if not (uploaded_image_A and uploaded_image_B):
+                    st.error("Please upload both images before running the morphing pipeline.")
+                else:
+                    # Save all settings to session state
+                    st.session_state.uploaded_image_A = uploaded_image_A
+                    st.session_state.uploaded_image_B = uploaded_image_B
+                    st.session_state.prompt_A = prompt_A
+                    st.session_state.prompt_B = prompt_B
+                    st.session_state.model_option = model_option
+                    st.session_state.num_frames = num_frames
+                    st.session_state.enable_lcm_lora = enable_lcm_lora
+                    st.session_state.use_adain = use_adain
+                    st.session_state.use_reschedule = use_reschedule
+                    st.session_state.use_film = use_film
+                    st.session_state.film_recursions = film_recursions
+                    st.session_state.output_fps = output_fps
+                    
+                    # Switch to processing page
+                    start_processing()
+                    st.rerun()
 
-    # Create a container for the run button and its status messages
-    run_container = st.container()
-    
-    # Create a separate container for displaying the download button
-    download_container = st.container()
-    
-    # Check if we should display the download button even when the form is disabled
-    if st.session_state.show_download and st.session_state.final_video_path is not None:
-        with download_container:
-            st.markdown("<div class='results-section'>", unsafe_allow_html=True)
-            st.success("Morphing complete! 🎉")
+    # =============== PROCESSING PAGE ===============
+    elif st.session_state.page == 'processing':
+        st.markdown(
+            """
+            <div class="processing-container">
+                <h2 class="processing-text">Processing Your Morphing Request</h2>
+                <p>Please wait while we generate your morphing video...</p>
+            </div>
+            """, 
+            unsafe_allow_html=True
+        )
+        
+        # Use a progress bar for visual feedback
+        progress_bar = st.progress(0)
+        
+        # Only start processing if not already started
+        if not st.session_state.process_started:
+            st.session_state.process_started = True
+            
+            # Create a temporary folder for processing
+            temp_dir = create_temp_folder()
+            st.session_state.temp_dir = temp_dir
+            
             try:
-                with open(st.session_state.final_video_path, "rb") as f:
-                    video_bytes = f.read()
-                    st.video(video_bytes)
-                    st.download_button(
-                        "Download Morphing Video",
-                        data=video_bytes,
-                        file_name="metamorph_result.mp4",
-                        mime="video/mp4"
-                    )
+                # Update progress
+                progress_bar.progress(10)
                 
-                # Add a restart button to clear the session state and start over
-                if st.button("Start New Morphing Project", key="restart_bottom"):
-                    restart()
-                    
-            except Exception as e:
-                st.error(f"Error preparing video for download: {e}")
-            st.markdown("</div>", unsafe_allow_html=True)
-    
-    with run_container:
-        # Button text changes based on processing state
-        button_text = "Processing... Please Wait" if st.session_state.processing else "Run Morphing Pipeline"
-        
-        # Button is disabled during processing
-        if st.button(button_text, key="run_pipeline", disabled=st.session_state.processing):
-            if not (uploaded_image_A and uploaded_image_B):
-                st.error("Please upload both images before running the morphing pipeline.")
-            else:
-                # Set the processing state to True to prevent multiple runs
-                st.session_state.processing = True
-                st.info("Processing your morphing request. Please wait...")
+                # Extract variables from session state
+                uploaded_image_A = st.session_state.uploaded_image_A
+                uploaded_image_B = st.session_state.uploaded_image_B
+                prompt_A = st.session_state.prompt_A
+                prompt_B = st.session_state.prompt_B
+                model_option = st.session_state.model_option
+                num_frames = st.session_state.num_frames
+                enable_lcm_lora = st.session_state.enable_lcm_lora
+                use_adain = st.session_state.use_adain
+                use_reschedule = st.session_state.use_reschedule
+                use_film = st.session_state.use_film
+                film_recursions = st.session_state.film_recursions
+                output_fps = st.session_state.output_fps
+                
+                # Save uploaded images
+                imgA_path = os.path.join(temp_dir, "imageA.png")
+                imgB_path = os.path.join(temp_dir, "imageB.png")
+                save_uploaded_file(uploaded_image_A, imgA_path)
+                save_uploaded_file(uploaded_image_B, imgB_path)
+                
+                # Update progress
+                progress_bar.progress(20)
                 
-                # Create a temporary folder for processing
-                temp_dir = create_temp_folder()
-                st.session_state.temp_dir = temp_dir
+                # Create output directories
+                timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+                output_dir = os.path.join(temp_dir, f"morph_results_{timestamp}")
+                film_output_dir = os.path.join(temp_dir, f"film_output_{timestamp}")
+                os.makedirs(output_dir, exist_ok=True)
+                os.makedirs(film_output_dir, exist_ok=True)
                 
+                actual_model_path = (
+                    "lykon/dreamshaper-7" if model_option == "Dreamshaper-7 (fine-tuned SD V1-5)" 
+                    else "stabilityai/stable-diffusion-2-1-base" if model_option == "Base Stable Diffusion V2-1"
+                    else "sd-legacy/stable-diffusion-v1-5"
+                )
+                
+                # Update progress
+                progress_bar.progress(30)
+                
+                # Build the command for run_morphing.py
+                cmd = [
+                    sys.executable, "run_morphing.py",
+                    "--model_path", actual_model_path,
+                    "--image_path_0", imgA_path,
+                    "--image_path_1", imgB_path,
+                    "--prompt_0", prompt_A,
+                    "--prompt_1", prompt_B,
+                    "--output_path", output_dir,
+                    "--film_output_folder", film_output_dir,
+                    "--num_frames", str(num_frames),
+                    "--fps", str(output_fps)
+                ]
+                
+                if enable_lcm_lora:
+                    cmd.append("--use_lcm")
+                if use_adain:
+                    cmd.append("--use_adain")
+                if use_reschedule:
+                    cmd.append("--use_reschedule")
+                if use_film:
+                    cmd.append("--use_film")
+                
+                # Add film recursion parameter
+                cmd.extend(["--film_num_recursions", str(film_recursions)])
+                
+                # Run the morphing process
                 try:
-                    # Save uploaded images
-                    imgA_path = os.path.join(temp_dir, "imageA.png")
-                    imgB_path = os.path.join(temp_dir, "imageB.png")
-                    save_uploaded_file(uploaded_image_A, imgA_path)
-                    save_uploaded_file(uploaded_image_B, imgB_path)
-                    
-                    # Create output directories
-                    timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
-                    output_dir = os.path.join(temp_dir, f"morph_results_{timestamp}")
-                    film_output_dir = os.path.join(temp_dir, f"film_output_{timestamp}")
-                    os.makedirs(output_dir, exist_ok=True)
-                    os.makedirs(film_output_dir, exist_ok=True)
-                    
-                    actual_model_path = (
-                        "lykon/dreamshaper-7" if model_option == "Dreamshaper-7 (fine-tuned SD V1-5)" 
-                        else "stabilityai/stable-diffusion-2-1-base" if model_option == "Base Stable Diffusion V2-1"
-                        else "sd-legacy/stable-diffusion-v1-5"
-                    )
+                    # Update progress - processing takes the longest
+                    progress_bar.progress(40)
                     
-                    # Build the command for run_morphing.py
-                    cmd = [
-                        sys.executable, "run_morphing.py",
-                        "--model_path", actual_model_path,
-                        "--image_path_0", imgA_path,
-                        "--image_path_1", imgB_path,
-                        "--prompt_0", prompt_A,
-                        "--prompt_1", prompt_B,
-                        "--output_path", output_dir,
-                        "--film_output_folder", film_output_dir,
-                        "--num_frames", str(num_frames),
-                        "--fps", str(output_fps)
-                    ]
+                    subprocess.run(cmd, check=True)
                     
-                    if enable_lcm_lora:
-                        cmd.append("--use_lcm")
-                    if use_adain:
-                        cmd.append("--use_adain")
-                    if use_reschedule:
-                        cmd.append("--use_reschedule")
-                    if use_film:
-                        cmd.append("--use_film")
+                    # Update progress
+                    progress_bar.progress(90)
                     
-                    # Add film recursion parameter
-                    cmd.extend(["--film_num_recursions", str(film_recursions)])
+                    # 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(film_output_dir, possible_outputs[0])
+                        video_found = True
                     
-                    # 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 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
-                            st.session_state.show_download = True
-                            st.rerun()  # This is crucial for showing the results page
-                        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
-                    st.session_state.process_complete = False
+                    if video_found:
+                        st.session_state.final_video_path = final_video_path
+                        st.session_state.page = 'result'
+                        progress_bar.progress(100)
+                        st.rerun()
+                    else:
+                        st.error("No output video was generated. Check logs for details.")
+                        if st.button("Return to Settings"):
+                            return_to_input()
+                            st.rerun()
+                        
+                except subprocess.CalledProcessError as e:
+                    st.error(f"Error running morphing pipeline: {e}")
+                    if st.button("Return to Settings"):
+                        return_to_input()
+                        st.rerun()
+                
+            except Exception as e:
+                st.error(f"An error occurred during processing: {e}")
+                if st.button("Return to Settings"):
+                    return_to_input()
+                    st.rerun()
+
+    # =============== RESULT PAGE ===============
+    elif st.session_state.page == 'result':
+        st.markdown(
+            """
+            <div style="text-align: center;">
+                <h2>Morphing Complete! 🎉</h2>
+                <p>Your morphing video has been successfully generated. You can download it below.</p>
+            </div>
+            """,
+            unsafe_allow_html=True
+        )
+        
+        # Show the result video and download button
+        try:
+            if st.session_state.final_video_path:
+                # Display video preview
+                video_file = open(st.session_state.final_video_path, 'rb')
+                video_bytes = video_file.read()
+                video_file.close()
+                
+                st.video(video_bytes)
+                
+                # Download button
+                st.download_button(
+                    "Download Morphing Video",
+                    data=video_bytes,
+                    file_name="metamorph_result.mp4",
+                    mime="video/mp4"
+                )
+        except Exception as e:
+            st.error(f"Error preparing video for download: {e}")
+        
+        # Button to start a new project
+        if st.button("Start New Morphing Project"):
+            return_to_input()
+            st.rerun()
 
 if __name__ == "__main__":
     main()