Add download timeout and better progress messages

app.py

@@ -56,13 +56,22 @@ def get_checkpoint_path(checkpoint_file, default_local_path, repo_id=None, filen
             cache_dir = "/workspace/checkpoints" if os.path.exists("/workspace") else "/tmp/checkpoints"
             os.makedirs(cache_dir, exist_ok=True)
             
+            # Add timeout and better error handling
+            import time
+            start_time = time.time()
+            print(f"Starting download from {repo_id}...")
+            
             checkpoint_path = hf_hub_download(
                 repo_id=repo_id,
                 filename=filename,
                 cache_dir=cache_dir,
                 local_files_only=False,
-                resume_download=True  # Resume if interrupted
+                resume_download=True,  # Resume if interrupted
+                timeout=600  # 10 minute timeout per request
             )
+            
+            elapsed = time.time() - start_time
+            print(f"Download completed in {elapsed:.1f} seconds")
             print(f"✅ Checkpoint downloaded to: {checkpoint_path}")
             return checkpoint_path
         except Exception as e:
@@ -132,9 +141,10 @@ def generate_image(prompt, aspect_ratio, cfg, seed, checkpoint_file, config_path
         ]
         
         status_msg += "Running generation...\n"
+        status_msg += "Note: First run includes checkpoint download (~10-20 min) and model loading (~2-5 min).\n"
         
-        # Run with timeout (30 minutes max)
-        result = subprocess.run(cmd, capture_output=True, text=True, cwd=os.getcwd(), timeout=1800)
+        # Run with timeout (45 minutes max - allows for download + generation)
+        result = subprocess.run(cmd, capture_output=True, text=True, cwd=os.getcwd(), timeout=2700)
         
         if result.returncode != 0:
             error_msg = f"Error during generation:\n{result.stderr}\n\nStdout:\n{result.stdout}"