Update app.py

app.py

@@ -5,7 +5,7 @@ import logging
 from werkzeug.middleware.proxy_fix import ProxyFix
 
 # ----------------------------------------------------
-# 1. Logging Setup
+# 1. Logging Setup (Your code is perfect)
 # ----------------------------------------------------
 logging.basicConfig(
     level=logging.INFO,
@@ -13,23 +13,32 @@ logging.basicConfig(
 )
 
 # ----------------------------------------------------
-# 2. Flask App Initialization
+# 2. Flask App Initialization (WITH THE FIXES)
 # ----------------------------------------------------
 app = Flask(__name__)
+
+# A. SECRET_KEY setup (Your logic is fine, just made the error message more explicit)
 app.secret_key = os.environ.get('SECRET_KEY')
 if not app.secret_key:
     logging.error("FATAL: SECRET_KEY environment variable is not set. Sessions will fail.")
     app.secret_key = 'fallback-key-for-dev-only-not-secure'
     logging.warning("⚠️ Using insecure fallback secret key. Set SECRET_KEY in production.")
 
-app.wsgi_app = ProxyFix(app.wsgi_app, x_for=1, x_proto=1, x_host=1, x_prefix=1)
+# B. ProxyFix (Enhanced for robustness on Hugging Face)
+app.wsgi_app = ProxyFix(app.wsgi_app, x_for=2, x_proto=1, x_host=1, x_prefix=1)
+
+# C. Secure cookie must be true
 app.config['SESSION_COOKIE_SECURE'] = True
-app.config['SESSION_COOKIE_SAMESITE'] = 'Lax'
 
-logging.info("✅ Flask App Initialized with Production Settings.")
+# D. *** THE CRITICAL ONE-SHOT FIX ***
+#    Changed from 'Lax' to 'None' to allow the session to work inside the Hugging Face iframe.
+app.config['SESSION_COOKIE_SAMESITE'] = 'None'
+
+logging.info("✅ Flask App Initialized with FINAL Production Settings for iframe environments.")
+
 
 # ----------------------------------------------------
-# 3. Game Content
+# 3. Game Content (Your code is perfect)
 # ----------------------------------------------------
 videos = [
     "static/videos/Animated_Drought_Relief_Video_Creation.mp4",
@@ -66,7 +75,7 @@ task_scenarios = [
      "choices": [
          {"action": "🏺 Underground concrete tank with cover", "description": "Safe long-term storage", "consequence": "✅ MASTER STRATEGY! Stored safely. +100 XP", "xp_reward": 100, "icon": "🏛️"},
          {"action": "🪣 Store in open buckets", "description": "Easy access containers", "consequence": "❌ BREEDING GROUND! Mosquito risk. -20 Health", "xp_reward": 0, "icon": "🦟"},
-         {"action": "🌍 Dig a pit and let water sit", "description": "Natural storage", "consequence": "❌ CONTAMINATION RISK! Polluted. -20 Health", "xp_reward": 0, "icon": "🕳️"}
+         {"action": "🌍 Dig a pit and let water sit", "description": "Natural storage", "consequence": "❌ CONTAMINATION RISK! Polluted. -20 Health", "xp_reward": 0, "icon": "🳳️"}
      ]},
     {"question": "♻️ Dry season! You have 200L left. Neighbor needs help. What do you do?",
      "choices": [
@@ -93,7 +102,7 @@ achievements_list = [
 ]
 
 # ----------------------------------------------------
-# 4. Game Logic
+# 4. Game Logic (Your code is perfect)
 # ----------------------------------------------------
 def calculate_level():
     return min(10, 1 + session.get('xp_points', 0) // 100)
@@ -112,11 +121,14 @@ def unlock_achievement(condition: str):
         resume_or_initialize()
     for a in achievements_list:
         if a["condition"] == condition and a not in session['achievements']:
+            # Use append to modify the list in place
             session['achievements'].append(a)
+            session.modified = True # Mark session as modified
             session['message'] = f"🏆 Achievement Unlocked: {a['name']} - {a['desc']}"
             add_xp(50)
 
 def reset_game_state():
+    session.clear() # Use clear for a full reset
     session["video_index"] = 0
     session["feedback"] = ""
     session["quiz_correct_count"] = 0
@@ -168,7 +180,7 @@ def select_card_logic(choice_index: int):
         session['health'] = max(0, session['health'] - 20)
 
 # ----------------------------------------------------
-# 5. Flask Routes
+# 5. Flask Routes (Your code is perfect)
 # ----------------------------------------------------
 @app.route('/')
 def welcome_page():
@@ -188,7 +200,9 @@ def main_app():
     resume_or_initialize()
 
     if request.method == 'POST':
+        logging.info("ROUTE: /main [POST] - Received POST request. Session is present.")
         if 'mark_watched' in request.form:
+            logging.info("Video marked as watched.")
             session['video_watched'] = True
             session['message'] = ""
         elif 'select_choice' in request.form:
@@ -206,21 +220,14 @@ def main_app():
 
     idx = session['video_index']
     context = {
-        'current_video_index': idx,
-        'total_videos': len(videos),
+        'current_video_index': idx, 'total_videos': len(videos),
         'video_src': url_for('static', filename=videos[idx].replace('static/', '', 1)),
-        'video_description': video_descriptions[idx],
-        'current_scenario': task_scenarios[idx],
-        'player_level': session['player_level'],
-        'xp_points': session['xp_points'],
-        'health': session['health'],
-        'streak': session['streak'],
-        'quiz_correct_count': session['quiz_correct_count'],
-        'quiz_attempts': session['quiz_attempts'],
-        'video_watched': session['video_watched'],
-        'question_answered': session['question_answered'],
-        'feedback': session['feedback'],
-        'achievements': session['achievements'],
+        'video_description': video_descriptions[idx], 'current_scenario': task_scenarios[idx],
+        'player_level': session['player_level'], 'xp_points': session['xp_points'],
+        'health': session['health'], 'streak': session['streak'],
+        'quiz_correct_count': session['quiz_correct_count'], 'quiz_attempts': session['quiz_attempts'],
+        'video_watched': session['video_watched'], 'question_answered': session['question_answered'],
+        'feedback': session['feedback'], 'achievements': session['achievements'],
         'message': session.pop('message', '')
     }
     return render_template('main.html', **context)
@@ -233,13 +240,10 @@ def summary_page():
     if score == total_quizzes and total_quizzes > 0:
         unlock_achievement("perfect")
     context = {
-        'player_level': session['player_level'],
-        'xp_points': session['xp_points'],
-        'quiz_score': score,
-        'total_quizzes': total_quizzes,
+        'player_level': session['player_level'], 'xp_points': session['xp_points'],
+        'quiz_score': score, 'total_quizzes': total_quizzes,
         'accuracy': (score / max(1, session['quiz_attempts'])) * 100,
-        'achievements': session['achievements'],
-        'current_date': time.strftime("%B %d, %Y"),
+        'achievements': session['achievements'], 'current_date': time.strftime("%B %d, %Y"),
         'message': session.pop('message', '')
     }
     return render_template('summary.html', **context)
@@ -250,7 +254,7 @@ def restart_quest():
     return redirect(url_for('welcome_page'))
 
 # ----------------------------------------------------
-# 6. Run App
+# 6. Run App (Your code is perfect)
 # ----------------------------------------------------
 if __name__ == '__main__':
-    app.run(debug=True, port=5000)
+    app.run(debug=True, port=5000)