fix SameSite attr

app.py

@@ -1,8 +1,10 @@
-from flask import Flask, render_template, request, jsonify, make_response
+from flask import Flask, render_template, request, jsonify, make_response, session
+from flask.sessions import SecureCookieSessionInterface
 from modules.model import summarize
 import __main__
 
 app = Flask(__name__)
+session_cookie = SecureCookieSessionInterface().get_signing_serializer(app)
 # shortTokenizer = BartTokenizer.from_pretrained('sshleifer/distilbart-xsum-12-6')
 # shortModel = BartForConditionalGeneration.from_pretrained('sshleifer/distilbart-xsum-12-6')
 
@@ -15,6 +17,7 @@ def home():
 
 
 @app.route('/summarize', methods=['GET', 'POST'])
+@app.after_request
 def recommend():
     if request.method == "POST":
         # Get form data
@@ -26,11 +29,8 @@ def recommend():
         try:
             short_output_summary, long_output_summary = summarize(input_text)
             response = jsonify({'short': short_output_summary.strip(), 'long': long_output_summary.strip()})
-            cookie = f'{cookie}; SameSite=None'
-            response.headers.add(
-                'Set-Cookie',
-                cookie
-            )
+            same_cookie = session_cookie.dumps(dict(session))
+            response.headers.add("Set-Cookie", f"my_cookie={same_cookie}; Secure; HttpOnly; SameSite=None; Path=/;")
             print(response)
             # Pass output summary to the output template
             return response