some more info and some difficulty added!

app.py

@@ -41,13 +41,14 @@ def index():
     db_path = Path(__file__).parent / 'problems.sqlite3'
     conn = sqlite3.connect(db_path)
     c = conn.cursor()
-    c.execute('SELECT folder, description, test_code FROM problems')
+    #<!-- The query was fucked up so it fetched the fucking testcode -->
+    c.execute('SELECT folder, description, test_code, difficulty FROM problems')
     problems = c.fetchall()
     conn.close()
     # Get leaderboard entries
     leaderboard = get_leaderboard()
     # Map folder to title for display
-    problem_titles = {folder: folder.replace('_', ' ').title() for folder, _, _ in problems}
+    problem_titles = {folder: folder.replace('_', ' ').title() for folder, _, _, _ in problems}
     return render_template('index.html', problems=problems, leaderboard=leaderboard, problem_titles=problem_titles)
 
 @app.route('/problem/new', methods=['GET', 'POST'])
@@ -67,17 +68,21 @@ def view_problem(folder):
     db_path = Path(__file__).parent / 'problems.sqlite3'
     conn = sqlite3.connect(db_path)
     c = conn.cursor()
-    c.execute('SELECT folder, description, test_code FROM problems WHERE folder = ?', (folder,))
+    c.execute('SELECT folder, description,test_code , difficulty FROM problems WHERE folder = ?', (folder,))
     row = c.fetchone()
     conn.close()
+    
     if not row:
         return 'Problem not found', 404
+        
     problem = {
         'folder': row[0],
         'description': row[1],
-        'test_code': row[2],
+        'difficulty': row[2],
+        'test_code': row[3],  # This is used internally, not displayed
         'title': row[0].replace('_', ' ').title()
     }
+
     result = None
     if request.method == 'POST':
         user_code = request.form['user_code']