Added OmniGen Image to Image

app.py

@@ -197,10 +197,13 @@ def omnigen_image_to_image():
         return jsonify({'error': 'Text prompt is required'}), 400
 
     steps = data.get('steps')
-
     if not steps:
         steps = 50
 
+    image_url = data.get('image_url')
+    if not image_url:
+        return jsonify({'error': '"image_url" is required'}), 400
+
     # Handle uploaded image or base64 image
     image_file = request.files.get('image')
     base64_image = data.get('base64_image')
@@ -232,8 +235,9 @@ def omnigen_image_to_image():
         elif base64_image:
             # Save base64 image
             try:
+                pass
                 image_path, image_url = save_base64_image(base64_image)
-                # image_url = "https://drive.google.com/uc?id=1JEHEy0zCVWOob4421hLQIPMbO_ebeCPS&export=download"
+                image_url = "https://drive.google.com/uc?id=1JEHEy0zCVWOob4421hLQIPMbO_ebeCPS&export=download"
             except Exception as e:
                 raise ValueError(f'Invalid base64 image data: {str(e)}')
 
@@ -291,6 +295,7 @@ def get_image_file(filename):
 # Route to serve images
 @app.route('/static/<path:filename>')
 def serve_static(filename):
+    print(f"Request for static file: {filename}", flush=True)
     return send_from_directory('static', filename)
 
 # Make a request route
@@ -596,11 +601,6 @@ def video_tasks(prompt_id):
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
-
-@app.route('/static/<path:filename>')
-def static_files(filename):
-    return send_from_directory('static', filename)
-
 # Route: Image to Video old
 @app.route('/image_to_video', methods=['POST'])
 def image_to_video():