Update app.py

app.py

@@ -4,7 +4,7 @@ import re
 import time
 import logging
 from datetime import datetime
-from flask import Flask, render_template, request, jsonify, session
+from flask import Flask, render_template, request, jsonify, session, make_response
 from google import genai 
 from dotenv import load_dotenv
 from cachelib import SimpleCache
@@ -27,12 +27,13 @@ client = genai.Client(api_key=api_key)
 cache = SimpleCache(threshold=50, default_timeout=60*60*24*7)
 
 app = Flask(__name__)
-# At the top of app.py after initializing the app
-app.config['SESSION_TYPE'] = 'filesystem'
-app.config['SESSION_PERMANENT'] = True
-app.config['PERMANENT_SESSION_LIFETIME'] = timedelta(days=7)
 # Use secret key from environment or a default for development
 app.secret_key = os.getenv("SECRET_KEY", "dev_secret_key")
+# Set session cookie settings to be more persistent
+app.config['SESSION_COOKIE_SECURE'] = False  # Set to True in HTTPS environments
+app.config['SESSION_COOKIE_HTTPONLY'] = True
+app.config['SESSION_COOKIE_SAMESITE'] = 'Lax'
+app.config['PERMANENT_SESSION_LIFETIME'] = 60*60*24*7  # 7 days
 
 # Supported languages
 LANGUAGES = {
@@ -49,7 +50,7 @@ LANGUAGES = {
     "ml": "മലയാളം (Malayalam)"
 }
 
-# List of pests and diseases (truncated for brevity)
+# List of pests and diseases
 PESTS_DISEASES = [
     {
         "id": 1,
@@ -142,22 +143,33 @@ def index():
     # Set default language if not set
     if 'language' not in session:
         session['language'] = 'en'
-        logger.info("Default language 'en' set in session")
-    else:
-        logger.info(f"Using existing session language: {session['language']}")
+    
+    # Make session persistent
+    session.permanent = True
+    
+    return render_template('index.html', 
+                          pests_diseases=PESTS_DISEASES,
+                          languages=LANGUAGES,
+                          current_language=session['language'])
 
 @app.route('/set_language', methods=['POST'])
 def set_language():
     language = request.form.get('language')
     if language in LANGUAGES:
+        session.permanent = True
         session['language'] = language
-        logger.info(f"Language set to {language} in session")
-        return jsonify({"success": True})
+        # Log the language change
+        logger.info(f"Language changed to {language}")
+        
+        # Create a response with success message
+        response = make_response(jsonify({"success": True}))
+        
+        # Set a cookie to ensure the language persists even if session fails
+        response.set_cookie('user_language', language, max_age=60*60*24*30)  # 30 days
+        
+        return response
     return jsonify({"success": False}), 400
 
-
-
-        
 def extract_json_from_response(content):
     """Extract and parse JSON from API response."""
     try:
@@ -182,8 +194,8 @@ def extract_json_from_response(content):
 
 @app.route('/get_details/<int:pest_id>')
 def get_details(pest_id):
-    # Get current language
-    language = session.get('language', 'en')
+    # Get current language from session or cookie fallback
+    language = session.get('language', request.cookies.get('user_language', 'en'))
     
     # Find the pest/disease by ID
     pest_disease = next((item for item in PESTS_DISEASES if item["id"] == pest_id), None)
@@ -326,6 +338,23 @@ def clear_specific_cache(pest_id):
         cache.delete(cache_key)
     return jsonify({"success": True, "message": f"Cache cleared for pest ID {pest_id}"})
 
+# Add a diagnostics endpoint to help debug session issues
+@app.route('/debug/session')
+def debug_session():
+    # Only enable in development
+    if os.getenv("FLASK_ENV") == "production":
+        return jsonify({"error": "Not available in production"}), 403
+        
+    return jsonify({
+        "session_data": dict(session),
+        "cookies": dict(request.cookies),
+        "session_cookie_name": app.session_cookie_name,
+        "session_cookie_secure": app.config.get('SESSION_COOKIE_SECURE'),
+        "session_cookie_httponly": app.config.get('SESSION_COOKIE_HTTPONLY'),
+        "session_cookie_samesite": app.config.get('SESSION_COOKIE_SAMESITE'),
+        "permanent_session_lifetime": str(app.config.get('PERMANENT_SESSION_LIFETIME'))
+    })
+
 if __name__ == '__main__':
     # Use PORT environment variable if available (for Hugging Face Spaces)
     port = int(os.environ.get("PORT", 7860))