Update app.py

app.py

@@ -1,7 +1,6 @@
 # app.py
 from flask import Flask, request, render_template
 from parser import parse_python_code
-import enumerate
 import os
 
 app = Flask(__name__)
@@ -33,7 +32,9 @@ def index():
             parts = parse_python_code(code)
 
         if parts:
-            return render_template('results.html', parts=parts, filename=filename)
+            # Add index to each part for Jinja
+            indexed_parts = [{'index': i + 1, **part} for i, part in enumerate(parts)]
+            return render_template('results.html', parts=indexed_parts, filename=filename)
         return 'No file or code provided', 400
 
     return render_template('index.html')