update code

app.py

@@ -394,11 +394,17 @@ def images_to_video(image_folder, output_video, fps=30):
     # Define input image pattern (expects images named like "%04d.png")
     image_pattern = os.path.join(image_folder, "%04d.png")
 
-    # FFmpeg command to encode video
+    # FFmpeg command to encode video (with -y to overwrite)
     command = [
-        ffmpeg_exe, '-framerate', str(fps), '-i', image_pattern,
-        '-c:v', 'libx264', '-preset', 'slow', '-crf', '18',  # High-quality H.264 encoding
-        '-pix_fmt', 'yuv420p', '-b:v', '5000k',  # Ensure compatibility & increase bitrate
+        ffmpeg_exe,
+        '-y',  # ✅ Overwrite output file without asking
+        '-framerate', str(fps),
+        '-i', image_pattern,
+        '-c:v', 'libx264',
+        '-preset', 'slow',
+        '-crf', '18',
+        '-pix_fmt', 'yuv420p',
+        '-b:v', '5000k',
         output_video
     ]
 
@@ -630,9 +636,9 @@ def avatar_generation(items, save_path_base, video_path_input, source_type, is_s
             images_to_video(save_frames_path_depth, os.path.join(save_path_base, image_name + video_name+ '_depth.mp4'))
 
             logging.info(f"✅ Video generation completed successfully!")
-            return os.path.join(save_path_base, image_name + video_name+ '_out.mp4'),  os.path.join(save_path_base, image_name + video_name+'_depth.mp4'), ""
+            return os.path.join(save_path_base, image_name + video_name+ '_out.mp4'),  os.path.join(save_path_base, image_name + video_name+'_depth.mp4')
     except Exception as e:
-        return None, None, f"❌ error：{str(e)}"
+        return None, None
 
 
 def get_image_base64(path):
@@ -802,7 +808,7 @@ def launch_gradio_app():
         )
         gr.HTML(
             """
-            <div style="color: #333; text-align: left; font-size: 16px; line-height: 1.6; margin-top: 20px; padding: 10px; border: 1px solid #ddd; border-radius: 8px; background-color: #f9f9f9;">
+            <div style="color: inherit; text-align: left; font-size: 16px; line-height: 1.6; margin-top: 20px; padding: 16px; border-radius: 10px; border: 1px solid rgba(0,0,0,0.1); background-color: rgba(240, 240, 240, 0.6); backdrop-filter: blur(2px);">
                 <strong>🧑‍🎨 How to use this demo:</strong>
                 <ol style="margin-top: 10px; padding-left: 20px;">
                     <li><strong>Select or upload a source image</strong> – this will be the avatar's face.</li>
@@ -937,7 +943,7 @@ def launch_gradio_app():
                             label="CFG Scale"
                         )
                         strength_slider = gr.Slider(
-                            minimum=0.4, maximum=0.85, value=0.65, step=0.05,
+                            minimum=0.4, maximum=0.85, value=0.75, step=0.05,
                             label="SDEdit Strength"
                         )
                         style_button = gr.Button("🎨 Apply Style", interactive=False)
@@ -975,16 +981,16 @@ def launch_gradio_app():
         def apply_style_and_mark(processed_image, style_choice, cfg, strength, working_dir, image_name_true):
             try:
                 styled = style_transfer(processed_image, styles[style_choice], cfg, strength, working_dir, image_name_true)
-                return styled, True, ""
+                return styled, True
             except Exception as e:
-                return None, True, f"❌ error：{str(e)}"
+                return None, True
 
         def process_image_and_enable_style(input_image, source_type, is_styled, wd):
             try:
                 processed_result, updated_source_type, image_name_true = process_image(input_image, source_type, is_styled, wd)
-                return processed_result, updated_source_type, gr.update(interactive=True), gr.update(interactive=True), image_name_true, ""
+                return processed_result, updated_source_type, gr.update(interactive=True), gr.update(interactive=True), image_name_true
             except Exception as e:
-                return None, updated_source_type, gr.update(interactive=False), gr.update(interactive=False), image_name_true, f"❌ error：{str(e)}"
+                return None, updated_source_type, gr.update(interactive=False), gr.update(interactive=False), image_name_true
         
         processed_image_button.click(
             fn=prepare_working_dir,
@@ -994,18 +1000,18 @@ def launch_gradio_app():
         ).success(
             fn=process_image_and_enable_style,
             inputs=[input_image, source_type, is_styled, working_dir],
-            outputs=[processed_image, source_type, style_button, submit, image_name_true, error_box],
+            outputs=[processed_image, source_type, style_button, submit, image_name_true],
             queue=True
         )
         style_button.click(
             fn=apply_style_and_mark,
             inputs=[processed_image, style_choice, cfg_slider, strength_slider, working_dir, image_name_true],
-            outputs=[style_image, is_styled, error_box]
+            outputs=[style_image, is_styled]
         )
         submit.click(
             fn=avatar_generation,
             inputs=[processed_image, working_dir, video_input, source_type, is_styled, style_image, image_name_true],
-            outputs=[output_video, output_video_1, error_box],  # ⏳ 稍后展示视频
+            outputs=[output_video, output_video_1],  # ⏳ 稍后展示视频
             queue=True
         )