Added: Generate image to video

app.py

@@ -24,7 +24,7 @@ load_dotenv()
 app = Flask(__name__)
 
 # Increase timeout limit
-WSGIRequestHandler.timeout = 3600  # Set to 60 minutes
+# WSGIRequestHandler.timeout = 3600
 
 ALLOWED_EXTENSIONS = {'jpg', 'jpeg', 'png', 'webp'}  # Define supported image types
 
@@ -517,23 +517,20 @@ def image_to_video():
 
 
         # Wait for workflow execution to complete
-        # last_ping = time.time()
-        # PING_INTERVAL = 30
+        last_ping = time.time()
+        PING_INTERVAL = 30
 
         # Wait for workflow execution to complete
         while True:
-            out = ws.recv()
-            if isinstance(out, str):
-                message = json.loads(out)
-                if message['type'] == 'executing':
-                    data = message['data']
-                    if data['node'] is None and data['prompt_id'] == prompt_id:
-                        break  # Execution is done
+            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()
+            if time.time() - last_ping > PING_INTERVAL:
+                ws.send('ping')
+                last_ping = time.time()
 
         # Fetch the history of the workflow
         history = get_history(prompt_id, token).get(prompt_id, {})