Create generate_image

generate_image

@@ -0,0 +1,25 @@
+@app.route('/generate_image', methods=['POST'])
+def generate_image():
+    try:
+        # Get text input from the request
+        text_description = request.json['text_description']
+
+        # Generate an image from the text description
+        output = model.generate_images(text_description)
+
+        # Get the image data
+        image_data = requests.get(output[0]["image"]).content
+
+        # Check if the user wants to save the image
+        save_image = request.json.get('save_image', False)
+
+        if save_image:
+            # Save the image to a file
+            with open("generated_image.jpg", "wb") as img_file:
+                img_file.write(image_data)
+            return jsonify({'message': 'Image generated and saved successfully'})
+        else:
+            # Return the image as a response
+            return jsonify({'image_data': image_data})
+    except Exception as e:
+        return jsonify({'error': str(e)})