Update app.py

app.py

@@ -7,17 +7,15 @@ 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
+    return ''.join(part['source'] for part in sorted_parts)
 
 @app.route('/', methods=['GET', 'POST'])
 def index():
     if request.method == 'POST':
         parts = None
         filename = 'unnamed.py'
+        code_input = None
 
         # Handle file upload
         if 'file' in request.files and request.files['file'].filename:
@@ -28,26 +26,34 @@ def index():
             file_path = os.path.join('uploads', filename)
             file.save(file_path)
             with open(file_path, 'r') as f:
-                code = f.read()
-            parts = parse_python_code(code)
+                code_input = f.read()
+            parts = parse_python_code(code_input)
 
         # Handle pasted code
         elif 'code' in request.form and request.form['code'].strip():
-            code = request.form['code']
+            code_input = request.form['code']
             filename = request.form.get('filename', 'unnamed.py') or 'unnamed.py'
             if not filename.endswith('.py'):
                 filename += '.py'
-            parts = parse_python_code(code)
+            parts = parse_python_code(code_input)
 
         if parts:
             indexed_parts = [{'index': i + 1, **part} for i, part in enumerate(parts)]
             reconstructed_code = reconstruct_code(indexed_parts)
-            return render_template('results.html', parts=indexed_parts, filename=filename, reconstructed_code=reconstructed_code)
+            # Return partial HTML for htmx to update the results section
+            return render_template(
+                'results_partial.html',
+                parts=indexed_parts,
+                filename=filename,
+                reconstructed_code=reconstructed_code,
+                code_input=code_input
+            )
         return 'No file or code provided', 400
 
-    return render_template('index.html')
+    # Initial page load
+    return render_template('index.html', parts=None, filename=None, reconstructed_code=None, code_input=None)
 
 if __name__ == '__main__':
     if not os.path.exists('uploads'):
         os.makedirs('uploads')
-    app.run(host="0.0.0.0", port=7860)
+    app.run(host="0.0.0.0",port=7860)