Added: Generate image to video

app.py

@@ -274,6 +274,7 @@ def v1_image_to_video():
     # Validate text prompt
     text_prompt = data.get('text_prompt')
     frame_rate = data.get('frame_rate')
+    steps = data.get('steps')
     if not text_prompt or not text_prompt.strip():
         return jsonify({'error': 'Text prompt is required'}), 400
 
@@ -288,6 +289,8 @@ def v1_image_to_video():
         except ValueError:
             return jsonify({'error': 'Frame rate must be a valid number (8, 12, or 24).'}), 400
 
+    if not steps:
+        steps = 50
     # Handle uploaded image or base64 image
     image_file = request.files.get('image')
     base64_image = data.get('base64_image')
@@ -335,6 +338,7 @@ def v1_image_to_video():
         # Modify workflow with inputs
         workflow["30"]["inputs"]["prompt"] = text_prompt
         workflow["44"]["inputs"]["frame_rate"] = frame_rate
+        workflow["57"]["inputs"]["steps"] = steps
         workflow["73"]["inputs"]["url"] = image_url
         workflow["31"]["inputs"]["prompt"] = "Low quality, watermark, strange motion, blur"
 
@@ -450,7 +454,7 @@ def image_to_video():
             return jsonify({'error': 'Frame rate must be a valid number (8, 12, or 24).'}), 400
 
     if not steps:
-        steps=50
+        steps = 50
 
     # Handle uploaded image or base64 image
     image_file = request.files.get('image')
@@ -499,6 +503,7 @@ def image_to_video():
         # Modify workflow with inputs
         workflow["30"]["inputs"]["prompt"] = text_prompt
         workflow["44"]["inputs"]["frame_rate"] = frame_rate
+        workflow["57"]["inputs"]["steps"] = steps
         workflow["73"]["inputs"]["url"] = image_url
         workflow["31"]["inputs"]["prompt"] = "Low quality, watermark, strange motion"