Added: Generate image to video

app.py

@@ -30,6 +30,44 @@ ws_address = os.getenv("WS_ADDRESS")
 # Generate a unique client ID
 client_id = str(uuid.uuid4())
 
+
+def allowed_file(filename):
+    """Check if the uploaded file has an allowed extension."""
+    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+
+def save_base64_image(b64_string):
+    """Decode a base64 string and save it as an image in the static folder."""
+    try:
+        # Handle Data URI scheme if present
+        if ',' in b64_string:
+            header, encoded = b64_string.split(',', 1)
+            ext = header.split('/')[1].split(';')[0] if '/' in header else 'png'
+        else:
+            encoded = b64_string
+            ext = 'png'
+
+        # Decode the image data
+        image_data = base64.b64decode(encoded)
+
+        # Generate a unique path for the image in the static folder
+        image_path = f"static/{uuid.uuid4()}.{ext}"
+
+        # Ensure directory exists
+        os.makedirs('static', exist_ok=True)
+
+        # Save the image
+        with open(image_path, 'wb') as f:
+            f.write(image_data)
+
+        print(f"Image saved at: {image_path}")
+
+        # Return the path and URL of the saved image
+        image_url = f"https://gosign-de-comfyui-api.hf.space/{image_path}"
+        return image_path, image_url
+
+    except Exception as e:
+        raise ValueError(f"Failed to save image: {e}")
+
 def get_image(filename, subfolder, image_type, token):
     url_values = {'filename': filename, 'subfolder': subfolder, 'type': image_type}
     url = f"{server_address}/view?{urllib.parse.urlencode(url_values)}"
@@ -73,6 +111,10 @@ def get_images(ws, workflow, token):
 def home():
     return render_template('home.html')
 
+                ####################################################
+                # Generate text to image using FLUX1.DEV Checkpoint#
+                ####################################################
+
 # Generate image route
 @app.route('/generate_image', methods=['POST'])
 def generate_image():
@@ -108,10 +150,6 @@ def generate_image():
     # seednum = random.randint(1, 9999999999999)
     # workflow["3"]["inputs"]["seed"] = seednum
 
-                #######################
-                # For model Flux1.dev #
-                #######################
-
     # Generate a random 15-digit seed as an integer
     seednum = random.randint(100000000000000, 999999999999999)
     workflow["31"]["inputs"]["seed"] = seednum
@@ -145,41 +183,6 @@ def get_image_file(filename):
     return send_file(filename, mimetype='image/png')
 
 
-def allowed_file(filename):
-    """Check if the uploaded file has an allowed extension."""
-    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
-
-def save_base64_image(b64_string):
-    """Decode a base64 string and save it as an image in the static folder."""
-    try:
-        # Handle Data URI scheme if present
-        if ',' in b64_string:
-            header, encoded = b64_string.split(',', 1)
-            ext = header.split('/')[1].split(';')[0] if '/' in header else 'png'
-        else:
-            encoded = b64_string
-            ext = 'png'
-
-        # Decode the image data
-        image_data = base64.b64decode(encoded)
-
-        # Generate a unique path for the image in the static folder
-        image_path = f"static/{uuid.uuid4()}.{ext}"
-
-        # Ensure directory exists
-        os.makedirs('static', exist_ok=True)
-
-        # Save the image
-        with open(image_path, 'wb') as f:
-            f.write(image_data)
-
-        print(f"Image saved at: {image_path}")
-
-        # Return the URL or path of the saved image
-        return f"https://gosign-de-comfyui-api.hf.space/{image_path}"
-
-    except Exception as e:
-        raise ValueError(f"Failed to save image: {e}")
 
 # Route to serve images
 @app.route('/static/<path:filename>')
@@ -259,6 +262,9 @@ def get_video_data(filename, subfolder, token):
         print(f"URL Error: {e.reason}")
         raise
 
+                ##################################################
+                # Generate image to video using CogVideoX-5B-12V #
+                ##################################################
 
 # Route: Image to Video
 @app.route('/image_to_video', methods=['POST'])
@@ -304,7 +310,7 @@ def image_to_video():
     elif base64_image:
         # Save base64 image
         try:
-            image_url = save_base64_image(base64_image)
+            image_path, image_url = save_base64_image(base64_image)
             # return jsonify({'image_url': image_url})
 
         except Exception as e: