Update app.py

app.py

@@ -87,6 +87,18 @@ def generate(
 def root():
     return "Welcome to the Fashion Outfit "
 
+# Route pour récupérer l'image générée
+@app.route('/api/get_image/<image_id>', methods=['GET'])
+def get_image(image_id):
+    # Construire le chemin complet de l'image
+    image_path = f"{image_id}.png"  # Assurez-vous que le nom de fichier correspond à celui que vous avez utilisé lors de la sauvegarde
+
+    # Renvoyer l'image
+    try:
+        return send_file(image_path, mimetype='image/png')
+    except FileNotFoundError:
+        return jsonify({'error': 'Image not found'}), 404
+
 # Route pour l'API REST
 @app.route('/api/run', methods=['POST'])
 def run():