Upload app.py

metadata_search/app.py

@@ -91,6 +91,7 @@ from metadata_search.paper_figures import (
     figure_paper_count,
     figures_payload_for_paper,
     figures_dir,
+    resolve_figure_asset,
 )
 from metadata_search.mcp_retrieval import (
     list_saved_papers,
@@ -274,18 +275,17 @@ def paper_pdf_file(filename: str):
 
 @app.get("/figures/<path:paper_key>/<path:filename>")
 def paper_figure_file(paper_key: str, filename: str):
-    root = figures_dir()
-    if root is None:
-        return jsonify({"error": "Figure assets are not configured."}), 404
-    paper_dir = root / paper_key
-    if not paper_dir.is_dir():
-        return jsonify({"error": "Paper figures not found."}), 404
-    target = (paper_dir / filename).resolve()
-    if not str(target).startswith(str(paper_dir.resolve())):
-        return jsonify({"error": "Invalid figure path."}), 400
-    if not target.is_file():
+    safe_paper_key = Path(paper_key).name
+    safe_name = Path(filename).name
+
+    target = resolve_figure_asset(safe_paper_key, safe_name)
+    if target is None:
         return jsonify({"error": "Figure file not found."}), 404
-    return send_from_directory(paper_dir, filename)
+
+    return send_from_directory(
+        target.parent,
+        target.name,
+    )
 
 
 @app.get("/api/papers/figures")