Update app.py

app.py

@@ -206,14 +206,14 @@ def main():
     preset_option = st.selectbox(
         "Select a preset for quality and inference time",
         options=[
-            "Maximum quality, highest inference time 🏆",
+            "Maximum quality, longest inference time 🏆",
             "Medium quality, medium inference time ⚖️",
-            "Low quality, lowest inference time ⚡",
+            "Low quality, shortest inference time ⚡",
             "Creative morph 🎨",
             "Custom ⚙️"
         ],
         index=0,
-        label_visibility="collapsed"
+        label_visibility="collapsed"  # Hide the label in the UI but keep it for accessibility
     )
 
     # Determine preset defaults based on selection
@@ -234,6 +234,7 @@ def main():
         preset_film = True
         preset_lcm = True
     else:
+        # "Custom"
         preset_model = None
         preset_film = None
         preset_lcm = None
@@ -264,7 +265,6 @@ def main():
             )
             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")
-            keyframe_duration = st.number_input("Keyframe Duration (seconds, only if not using FILM)", min_value=0.01, max_value=5.0, value=0.1, step=0.01)
         
         # Right Column: Inter-frame Interpolator Parameters (FILM)
         with col_right:
@@ -272,9 +272,12 @@ def main():
             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_fps = st.number_input("FILM FPS (1–120)", min_value=1, max_value=120, value=30, help="Output video frames per second")
             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 4
+            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)
 
     st.markdown("<hr>", unsafe_allow_html=True)
@@ -283,7 +286,7 @@ 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
+    # 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.")
 
     if st.button("Run Morphing Pipeline", key="run_pipeline"):
@@ -307,8 +310,6 @@ def main():
             os.makedirs(output_dir, exist_ok=True)
             os.makedirs(film_output_dir, exist_ok=True)
             
-            duration_ms = int(keyframe_duration * 1000)
-            
             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"
@@ -326,7 +327,7 @@ def main():
                 "--output_path", output_dir,
                 "--film_output_folder", film_output_dir,
                 "--num_frames", str(num_frames),
-                "--duration", str(duration_ms)
+                "--fps", str(output_fps)
             ]
             
             if enable_lcm_lora:
@@ -338,9 +339,10 @@ def main():
             if use_film:
                 cmd.append("--use_film")
             
-            cmd.extend(["--film_fps", str(film_fps)])
+            # 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",
@@ -351,22 +353,25 @@ def main():
             st.info("Initializing pipeline. This may take a few minutes...")
             progress_bar = st.progress(0)
             status_text = st.empty()
-            
+
+            # Update progress status
             for i in range(1, 11):
                 status_text.text(f"Step {i}/10: {'Preparing images' if i <= 2 else 'Generating keyframes' if i <= 6 else 'Interpolating frames' if i <= 9 else 'Finalizing video'}")
                 progress_bar.progress(i * 10)
                 
-                if i == 3:
+                if i == 3: # Start processing!
                     try:
                         subprocess.run(cmd, check=True)
                     except subprocess.CalledProcessError as e:
                         st.error(f"Error running morphing pipeline: {e}")
                         return
                     break
-            
+
+            # Set to 100% when done
             progress_bar.progress(100)
             status_text.text("Processing complete!")
             
+            # 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:
@@ -381,7 +386,7 @@ def main():
             
             if video_found:
                 st.success("Morphing complete! 🎉")
-                st.video(final_video_path)
+                # st.video(final_video_path) # Comment to remove video preview, buggy
                 try:
                     with open(final_video_path, "rb") as f:
                         video_bytes = f.read()