Update app.py

app.py

@@ -21,10 +21,11 @@ snapshot_download(
 # -----------------------------------
 # Step 2: Setup paths and config
 # -----------------------------------
-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/"
+BASE_DIR = "/home/user/app"
+FVSM_PATH = os.path.join(BASE_DIR, "ckpt/FVSM/FloVD_FVSM_Controlnet.pt")
+OMSM_PATH = os.path.join(BASE_DIR, "ckpt/OMSM/")
+DEPTH_CKPT_PATH = os.path.join(BASE_DIR, "ckpt/others/depth_anything_v2_metric_hypersim_vitb.pth")
+OUTPUT_PATH = os.path.join(BASE_DIR, "results")
 GEN_VID_DIR = os.path.join(OUTPUT_PATH, "generated_videos")
 
 POSE_TYPE = "re10k"
@@ -56,7 +57,7 @@ def run_flovd(prompt, image, cam_pose_name):
         print("🚀 Starting video generation")
         print("----------------------------")
 
-        image_path = "./temp_input.png"
+        image_path = os.path.join(BASE_DIR, "temp_input.png")
         image.save(image_path)
         print(f"📸 Image saved at {image_path}")
 
@@ -83,16 +84,16 @@ def run_flovd(prompt, image, cam_pose_name):
         video_path = os.path.join(GEN_VID_DIR, video_filename)
 
         print(f"\n📍 Looking for generated video at: {video_path}")
-        if os.path.exists(video_path):
+        if os.path.exists(video_path) and os.path.getsize(video_path) > 0:
             print("✅ Video file found.")
-            return video_path, list_generated_videos(), None
+            return video_path, "✅ Video generated successfully.", list_generated_videos(), None
         else:
-            print("❌ File missing:", video_path)
-            return f"[ERROR] File not found at: {video_path}", list_generated_videos(), None
+            print("❌ File missing or empty.")
+            return None, f"❌ File not found at: {video_path}", list_generated_videos(), None
 
     except Exception as e:
         print(f"🔥 Exception occurred: {str(e)}")
-        return f"[EXCEPTION] {str(e)}", list_generated_videos(), None
+        return None, f"🔥 Exception: {str(e)}", list_generated_videos(), None
 
 def get_video_file(video_name):
     video_path = os.path.join(GEN_VID_DIR, video_name)
@@ -116,6 +117,7 @@ with gr.Blocks(title="FloVD + CogVideoX") as demo:
             run_btn = gr.Button("Generate Video")
         with gr.Column():
             output_video = gr.Video(label="Generated Video")
+            status_text = gr.Markdown(label="Status / Logs")  # shows errors/success
             video_selector = gr.Dropdown(choices=list_generated_videos(), label="Select Video to Download")
             download_btn = gr.Button("Download Selected")
             file_output = gr.File(label="Download Link")
@@ -123,7 +125,7 @@ with gr.Blocks(title="FloVD + CogVideoX") as demo:
     run_btn.click(
         fn=run_flovd,
         inputs=[prompt, image, pose_file],
-        outputs=[output_video, video_selector, file_output],
+        outputs=[output_video, status_text, video_selector, file_output],
     )
 
     download_btn.click(