Added: Generate image to video

app.py

@@ -515,8 +515,6 @@ def image_to_video():
         # Queue the prompt and wait for completion
         prompt_id = queue_prompt(workflow, token)
 
-        return jsonify({'prompt_id': prompt_id, 'message': 'Prompt queued successfully', 'get_video_url': f'https://gosign-de-comfyui-api.hf.space/get_video/{prompt_id}'}), 202
-
 
         # Wait for workflow execution to complete
         # last_ping = time.time()
@@ -524,10 +522,13 @@ def image_to_video():
 
         # 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
+            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
 
             # Send a ping if PING_INTERVAL has passed
             # if time.time() - last_ping > PING_INTERVAL:
@@ -583,11 +584,10 @@ def image_to_video():
         return jsonify({'error': str(e)}), 500
 
     finally:
-        pass
         # Always delete the image if it was saved
-        # if image_path and os.path.exists(image_path):
-        #     os.remove(image_path)
-        #     print(f"Deleted temporary image: {image_path}", flush=True)
+        if image_path and os.path.exists(image_path):
+            os.remove(image_path)
+            print(f"Deleted temporary image: {image_path}", flush=True)
 
 
 if __name__ == '__main__':