Update app.py

app.py

@@ -3,6 +3,7 @@ import subprocess
 import os
 from huggingface_hub import hf_hub_download
 import glob
+from datetime import datetime
 
 # Ensure 'checkpoint' directory exists
 os.makedirs("checkpoint", exist_ok=True)
@@ -27,6 +28,15 @@ def run_xportrait(
     num_mix, 
     ddim_steps
 ):
+    # Check if the model weights are in place
+    if not os.path.exists(resume_dir):
+        return "Model weights not found in checkpoint directory. Please download them first."
+
+    # Create a unique output directory name based on current date and time
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    output_dir = os.path.join(output_dir_base, f"output_{timestamp}")
+    os.makedirs(output_dir, exist_ok=True)
+    
     # Construct the command
     command = [
         "python3", "core/test_xportrait.py",