Update app.py

app.py

@@ -36,6 +36,7 @@ FVSM_PATH = "./ckpt/FVSM/FloVD_FVSM_Controlnet.pt"
 OMSM_PATH = "./ckpt/OMSM/"
 DEPTH_CKPT_PATH = "./ckpt/others/depth_anything_v2_metric_hypersim_vitb.pth"
 OUTPUT_PATH = "./results/"
+GEN_VID_DIR = os.path.join(OUTPUT_PATH, "generated_videos")
 POSE_TYPE = "re10k"
 CONTROLNET_GUIDANCE_END = 0.4
 SPEED = 1.0
@@ -43,16 +44,29 @@ NUM_FRAMES = 81
 FPS = 16
 INFER_STEPS = 50
 
-os.makedirs(os.path.join(OUTPUT_PATH, "generated_videos"), exist_ok=True)
+os.makedirs(GEN_VID_DIR, exist_ok=True)
+
+def list_generated_videos():
+    try:
+        files = os.listdir(GEN_VID_DIR)
+        print(f"\n📂 Contents of {GEN_VID_DIR}:")
+        for f in files:
+            print(f"   - {f}")
+    except Exception as e:
+        print(f"[⚠️] Could not list contents: {str(e)}")
 
 def run_flovd(prompt, image, cam_pose_name):
     try:
-        # Save input image
+        print("\n----------------------------")
+        print("🚀 Starting video generation")
+        print("----------------------------")
+
+        list_generated_videos()
+
         image_path = "./temp_input.png"
         image.save(image_path)
         print(f"📸 Image saved at {image_path}")
 
-        # Run video generation
         generate_video(
             prompt=prompt,
             fvsm_path=FVSM_PATH,
@@ -71,19 +85,22 @@ def run_flovd(prompt, image, cam_pose_name):
             num_inference_steps=INFER_STEPS,
         )
 
-        # Construct video path
+        print("\n✅ Generation finished.")
+        list_generated_videos()
+
         prompt_short = prompt[:30].strip().replace(" ", "_").replace(".", "").replace(",", "")
         video_filename = f"{prompt_short}_{cam_pose_name}.mp4"
-        video_path = os.path.join(OUTPUT_PATH, "generated_videos", video_filename)
+        video_path = os.path.join(GEN_VID_DIR, video_filename)
 
-        print(f"📍 Looking for generated video at: {video_path}")
+        print(f"\n📍 Looking for generated video at: {video_path}")
         if os.path.exists(video_path):
-            print("✅ Video generated:", video_path)
+            print("✅ Video file found.")
             return video_path
         else:
             print("❌ File missing:", video_path)
-            return gr.Textbox.update(value=f"[ERROR] File not found at: {video_path}")
-            
+            list_generated_videos()
+            return f"[ERROR] File not found at: {video_path}"
+
     except Exception as e:
         print(f"🔥 Exception occurred: {str(e)}")
         return f"An error occurred during generation: {str(e)}"