Update app.py

app.py

@@ -14,6 +14,7 @@ import sys
 import gc 
 import numpy as np
 from gradio_modal import Modal
+import tempfile
 
 def draw_points(img, points, labels):
     out = img.copy()
@@ -521,48 +522,12 @@ def animate_scene(input_video, max_duration_s, edited_frame, rc_str,
 
     print(f'{session_id} inference started with {rc_str} and duration of {max_duration_s}')
 
-    try:
-        
-        output_video_path = _animate(
-            input_video, max_duration_s, edited_frame_png, rc_bool,
-            pts_by_frame, lbs_by_frame, session_id, progress
-        )
-    except Exception as e:
-        err = str(e).lower()
-        print(f"{session_id} failed due to {err}")
-
-        if "zerogpu quotas" not in err and "pro gpu quota" not in err:
-            raise
-
-        if max_duration_s > 2:
-            raise
-
-        max_duration_s = -1
-
-        try:
-            pts_by_frame, lbs_by_frame = {}, {}
-            
-            output_video_path = _animate(
-                input_video, max_duration_s, edited_frame_png, rc_bool,
-                pts_by_frame, lbs_by_frame, session_id, progress
-            )
-        except Exception as e:
-            err = str(e).lower()
-            print(f"{session_id} failed due to {err}")
-
-            if "unlogged" in err:
-                max_duration_s = -2
-
-                gr.Info("Sign up for free or login to get at least 2 generations for free")
-
-                output_video_path = _animate(
-                    input_video, max_duration_s, edited_frame_png, rc_bool,
-                    pts_by_frame, lbs_by_frame, session_id, progress
-                )
-            else:
-                raise   
+    output_video_path = _animate(
+        input_video, max_duration_s, edited_frame_png, rc_bool,
+        pts_by_frame, lbs_by_frame, session_id, progress
+    )
 
-    final_video_path = os.path.join(output_dir, 'final_result.mp4')
+    final_video_path = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False).name
 
     preprocess_dir = os.path.join(output_dir, "preprocess_dir")
     pose_video = os.path.join(preprocess_dir, 'src_pose.mp4')