Update app.py

app.py

@@ -5,12 +5,21 @@ import os
 
 app = Flask(__name__)
 
+def reconstruct_code(parts):
+    """Reconstruct the original code from parsed parts."""
+    # Sort parts by start line to ensure correct order
+    sorted_parts = sorted(parts, key=lambda p: p['location'][0])
+    # Concatenate the source strings
+    reconstructed = ''.join(part['source'] for part in sorted_parts)
+    return reconstructed
+
 @app.route('/', methods=['GET', 'POST'])
 def index():
     if request.method == 'POST':
         parts = None
         filename = 'unnamed.py'
 
+        # Handle file upload
         if 'file' in request.files and request.files['file'].filename:
             file = request.files['file']
             if not file.filename.endswith('.py'):
@@ -22,6 +31,7 @@ def index():
                 code = f.read()
             parts = parse_python_code(code)
 
+        # Handle pasted code
         elif 'code' in request.form and request.form['code'].strip():
             code = request.form['code']
             filename = request.form.get('filename', 'unnamed.py') or 'unnamed.py'
@@ -31,7 +41,8 @@ def index():
 
         if parts:
             indexed_parts = [{'index': i + 1, **part} for i, part in enumerate(parts)]
-            return render_template('results.html', parts=indexed_parts, filename=filename)
+            reconstructed_code = reconstruct_code(indexed_parts)
+            return render_template('results.html', parts=indexed_parts, filename=filename, reconstructed_code=reconstructed_code)
         return 'No file or code provided', 400
 
     return render_template('index.html')
@@ -39,4 +50,4 @@ def index():
 if __name__ == '__main__':
     if not os.path.exists('uploads'):
         os.makedirs('uploads')
-    app.run(host="0.0.0.0",port=7860)
+    app.run(port=7860)