Added: Generate image to video

app.py

@@ -169,7 +169,8 @@ def generate_image():
     ws = websocket.WebSocket()
 
     try:
-        ws.connect(f"{ws_address}?clientId={client_id}&token={token}")
+        ws.connect(f"{ws_address}?clientId={client_id}&token={token}", header=
+        {"Authorization": f"Bearer {token}"})
     except websocket.WebSocketException as e:
         return jsonify({'error': f'WebSocket connection failed: {str(e)}'}), 500
 
@@ -198,9 +199,12 @@ def image_to_video():
 
     # Extract token from headers
     token = request.headers.get('Authorization')
-    if not token or not token.startswith("Bearer "):
-        return jsonify({'error': 'Invalid or missing token'}), 400
-    token = base64.b64decode(token.split(" ")[1]).decode("utf-8")
+    if token is None:
+        return jsonify({'error': 'No token provided'}), 400
+    if token.startswith("Bearer "):
+        token = token.split(" ")[1]
+
+    # token = base64.b64decode(token).decode("utf-8")
 
     # Extract text prompt
     text_prompt = data.get('text_prompt')
@@ -244,7 +248,8 @@ def image_to_video():
 
     # WebSocket connection to trigger workflow
     ws = websocket.WebSocket()
-    ws.connect(f"{ws_address}?clientId={client_id}&token={token}")
+    ws.connect(f"{ws_address}?clientId={client_id}&token={token}", header=
+        {"Authorization": f"Bearer {token}"})
     ws.send(json.dumps({"workflow": workflow}))  # Send the modified workflow
 
     # Receive video processing result