Update app.py

app.py

@@ -330,15 +330,25 @@ def image_to_video():
     except ValueError as e:
         return jsonify({'error': str(e)}), 400
 
+    
+    # Wait for workflow execution to complete
+    last_ping = time.time()  # Track the last ping time
+    PING_INTERVAL = 30  # Send a ping every 30 seconds
+
     # Wait for workflow execution to complete
     while True:
         message = json.loads(ws.recv())
         if message.get('type') == 'executing' and message['data']['node'] is None and message['data']['prompt_id'] == prompt_id:
             break
-
+        
+        # Send a ping if PING_INTERVAL has passed
+        if time.time() - last_ping > PING_INTERVAL:
+            ws.send('ping')
+            last_ping = time.time()
+            
    # Fetch the complete history for the provided prompt_id
     history = get_history(prompt_id, token).get(prompt_id, {})
-    print(f"History for prompt {prompt_id}: {history}")
+    # print(f"History for prompt {prompt_id}: {history}")
 
     video_data = None  # Initialize video data