Update app.py

app.py

@@ -1,7 +1,8 @@
-from flask import Flask, render_template, request, jsonify, send_file
+from flask import Flask, render_template, request, jsonify, send_file, Response
 from parser import parse_source_to_graph
-from dataset_gen import create_dataset_entry, get_dataset_stats, upload_to_hub, OUTPUT_FILE
+from dataset_gen import create_dataset_entry, get_dataset_stats, upload_to_hub, build_dataset_entry, OUTPUT_FILE
 import os
+import json
 
 app = Flask(__name__)
 
@@ -14,6 +15,35 @@ def parse():
     code = request.json.get('code', '')
     return jsonify(parse_source_to_graph(code))
 
+# --- New API Endpoint ---
+@app.route('/api/process', methods=['POST'])
+def api_process():
+    """
+    Direct API for external tools.
+    Input: JSON { "code": "...", "format": "json" | "jsonl" }
+    Output: The dataset entry object or a raw JSONL string.
+    """
+    data = request.get_json()
+    if not data or 'code' not in data:
+        return jsonify({"error": "No code provided"}), 400
+        
+    code = data['code']
+    fmt = data.get('format', 'json') # Default to JSON object
+    
+    # Generate the entry without saving to disk
+    entry = build_dataset_entry(code)
+    
+    if "error" in entry:
+        return jsonify(entry), 400
+    
+    if fmt == 'jsonl':
+        # Return as a string suitable for appending to a file
+        return Response(json.dumps(entry) + '\n', mimetype='text/plain')
+    
+    # Return as standard JSON object
+    return jsonify(entry)
+
+
 @app.route('/dataset/add', methods=['POST'])
 def add_entry():
     code = request.json.get('code', '')