Update app.py

app.py

@@ -31,7 +31,7 @@ def _inject_params(prompt: Dict[str, Any], r: T2VReq) -> Dict[str, Any]:
         elif class_type == "EmptyLatentImage":
             if r.width is not None: node_data["inputs"]["width"] = r.width
             if r.height is not None: node_data["inputs"]["height"] = r.height
-            if r.length is not None: node_data["inputs"]["batch_size"] = r.length # batch_size ही Frames होते हैं
+            if r.length is not None: node_data["inputs"]["batch_size"] = r.length
         elif class_type == "CreateVideo":
             if r.fps is not None: node_data["inputs"]["fps"] = r.fps
         elif class_type == "SaveVideo":
@@ -58,14 +58,18 @@ def _get_history(prompt_id: str, token: str) -> Dict[str, Any]:
     return r.json().get(prompt_id, {})
 
 def _extract_video_from_history(history: Dict[str, Any]) -> Dict[str, str]:
+    if "outputs" not in history:
+        raise RuntimeError(f"Server crashed before creating outputs! History dump: {json.dumps(history)}")
+        
     outputs = history.get("outputs", {})
     for _, node_out in outputs.items():
         for key in ("images", "videos", "files", "gifs"):
             if key in node_out:
                 for it in node_out[key]:
-                    if it.get("filename", "").lower().endswith((".mp4", ".webm", ".gif", ".mov")):
-                        return {"filename": it["filename"], "subfolder": it.get("subfolder", ""), "type": it.get("type", "")}
-    raise RuntimeError("No video file found in history outputs")
+                    if isinstance(it, dict) and it.get("filename", "").lower().endswith((".mp4", ".webm", ".gif", ".mov")):
+                        return {"filename": it["filename"], "subfolder": it.get("subfolder", ""), "type": it.get("type", "output")}
+                        
+    raise RuntimeError(f"Video not found! Node outputs were: {json.dumps(outputs)}")
 
 sample_prompts = [
     "A majestic cinematic shot of Lord Shiva meditating in the snowy Himalayas, cosmic universe and stars swirling around him, highly detailed, photorealistic",
@@ -127,26 +131,37 @@ with gr.Blocks(title="T2V UI", theme=gr.themes.Soft(primary_hue="blue")) as demo
         p = 0
         last_emit = -1
         
-        while True:
-            out = ws.recv()
-            if isinstance(out, bytes):
-                continue
-            msg = json.loads(out)
-            if msg.get("type") == "executing":
-                data = msg.get("data", {})
-                if data.get("prompt_id") != prompt_id:
+        try:
+            while True:
+                out = ws.recv()
+                if isinstance(out, bytes):
                     continue
-                node = data.get("node")
-                if node is None:
-                    break
-                if node not in seen:
-                    seen.add(node)
-                    p = min(99, int(len(seen) / total_nodes * 100))
-                    if p != last_emit:
-                        last_emit = p
-                        yield p, None
-        ws.close()
-        
+                msg = json.loads(out)
+                
+                # यहाँ हम असली Error पकड़ेंगे!
+                if msg.get("type") == "execution_error":
+                    ws.close()
+                    err_data = msg.get("data", {})
+                    node_type = err_data.get("node_type", "Unknown Node")
+                    exc_msg = err_data.get("exception_message", "Unknown Error")
+                    raise gr.Error(f"❌ Server Crashed at '{node_type}': {exc_msg}")
+                    
+                if msg.get("type") == "executing":
+                    data = msg.get("data", {})
+                    if data.get("prompt_id") != prompt_id:
+                        continue
+                    node = data.get("node")
+                    if node is None:
+                        break
+                    if node not in seen:
+                        seen.add(node)
+                        p = min(99, int(len(seen) / total_nodes * 100))
+                        if p != last_emit:
+                            last_emit = p
+                            yield p, None
+        finally:
+            ws.close()
+            
         hist = _get_history(prompt_id, req.token)
         info = _extract_video_from_history(hist)
         q = urlencode(info)
@@ -161,4 +176,3 @@ with gr.Blocks(title="T2V UI", theme=gr.themes.Soft(primary_hue="blue")) as demo
 
 if __name__ == "__main__":
     demo.queue().launch(server_name="0.0.0.0", server_port=7860)
-