Update app.py

app.py

@@ -85,9 +85,10 @@ def api_generate_image():
         return jsonify({'error': f"An error occurred while writing the image to file: {e}"}), 500
 
     return send_file(image_path, mimetype='image/jpeg', as_attachment=True)
-@app.route('/api-docs')
-def api_docs():
-    return render_template('api_docs.html')
+@app.errorhandler(404)
+def page_not_found(e):
+    # Render the 404.html template
+    return render_template('404.html'), 404
             
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860,debug=True)