Update app.py

app.py

@@ -1,5 +1,6 @@
 from flask import Flask, render_template, request, send_file, jsonify
 from imaginepy import Imagine, Style, Ratio
+import os
 
 app = Flask(__name__)
 
@@ -67,12 +68,14 @@ def api_generate_image():
         return jsonify({'error': "An error occurred while upscaling the image."}), 500
 
     try:
-        with open("generated.jpeg", mode="wb") as img_file:
+        image_path = os.path.join(app.root_path, "generated.jpeg")
+        with open(image_path, mode="wb") as img_file:
             img_file.write(img_data)
     except Exception as e:
         return jsonify({'error': f"An error occurred while writing the image to file: {e}"}), 500
 
-    return send_file("generated.jpeg", mimetype='image/jpeg', as_attachment=True) #jsonify({'success': True})
+    return send_file(image_path, mimetype='image/jpeg', as_attachment=True)
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0",port=7860)
+    app.run(host="0.0.0.0", port=7860)
+