Update app.py

app.py

@@ -3,30 +3,37 @@ import os
 from datetime import datetime
 
 app = Flask(__name__)
-
-# Ensure output folder exists
-os.makedirs("markdown_files", exist_ok=True)
+BASE_DIR = "markdown_files"
+os.makedirs(BASE_DIR, exist_ok=True)
 
 @app.route("/create-md", methods=["POST"])
 def create_markdown():
-    data = request.get_json()
-    if not data or "title" not in data or "content" not in data:
-        return jsonify({"error": "Missing 'title' or 'content'"}), 400
-
-    title = data["title"].strip()
-    content = data["content"]
-
-    # Sanitize filename
+    try:
+        data = request.get_json(force=True)
+        title = data["title"].strip()
+        content = data["content"]
+        tag = data.get("tag", "untagged").strip().replace(" ", "_")
+    except Exception as e:
+        return jsonify({"error": f"Invalid request body or missing fields: {str(e)}"}), 400
+
+    # Create subfolder for tag
+    tag_dir = os.path.join(BASE_DIR, tag)
+    os.makedirs(tag_dir, exist_ok=True)
+
+    # Generate a timestamped filename
     filename = f"{title.replace(' ', '_')}_{int(datetime.utcnow().timestamp())}.md"
-    filepath = os.path.join("markdown_files", filename)
+    filepath = os.path.join(tag_dir, filename)
 
-    # Write markdown content
-    with open(filepath, "w", encoding="utf-8") as f:
-        f.write(f"# {title}\n\n{content}")
+    try:
+        with open(filepath, "w", encoding="utf-8") as f:
+            f.write(f"# {title}\n\n{content}")
+    except Exception as e:
+        return jsonify({"error": f"Could not write file: {str(e)}"}), 500
 
     return jsonify({
         "message": "Markdown file created",
         "filename": filename,
+        "tag": tag,
         "path": filepath
     }), 201