Update app.py

app.py

@@ -1,7 +1,9 @@
 # app.py
-from flask import Flask, request, render_template
+from flask import Flask, request, render_template, jsonify, send_file
 from parser import parse_python_code
 import os
+import json
+import io
 
 app = Flask(__name__)
 
@@ -17,7 +19,6 @@ def index():
         filename = 'unnamed.py'
         code_input = None
 
-        # Handle file upload
         if 'file' in request.files and request.files['file'].filename:
             file = request.files['file']
             if not file.filename.endswith('.py'):
@@ -28,8 +29,6 @@ def index():
             with open(file_path, 'r') as f:
                 code_input = f.read()
             parts = parse_python_code(code_input)
-
-        # Handle pasted code
         elif 'code' in request.form and request.form['code'].strip():
             code_input = request.form['code']
             filename = request.form.get('filename', 'unnamed.py') or 'unnamed.py'
@@ -40,7 +39,6 @@ def index():
         if parts:
             indexed_parts = [{'index': i + 1, **part} for i, part in enumerate(parts)]
             reconstructed_code = reconstruct_code(indexed_parts)
-            # Return partial HTML for htmx to update the results section
             return render_template(
                 'results_partial.html',
                 parts=indexed_parts,
@@ -50,9 +48,22 @@ def index():
             )
         return 'No file or code provided', 400
 
-    # Initial page load
     return render_template('index.html', parts=None, filename=None, reconstructed_code=None, code_input=None)
 
+@app.route('/export_json', methods=['POST'])
+def export_json():
+    parts = request.json.get('parts', [])
+    export_data = [{'vector': part['vector'], 'source': part['source']} for part in parts]
+    json_str = json.dumps(export_data, indent=2)
+    buffer = io.BytesIO(json_str.encode('utf-8'))
+    buffer.seek(0)
+    return send_file(
+        buffer,
+        as_attachment=True,
+        download_name='code_vectors.json',
+        mimetype='application/json'
+    )
+
 if __name__ == '__main__':
     if not os.path.exists('uploads'):
         os.makedirs('uploads')