Create app.py

app.py

@@ -0,0 +1,43 @@
+import os
+import zipfile
+import json
+from flask import Flask, jsonify
+
+app = Flask(__name__)
+
+FIG_DIR = "FIG"
+ZIP_FILE = "FIG.zip"
+
+# 自动解压 FIG.zip
+if not os.path.exists(FIG_DIR):
+    if os.path.exists(ZIP_FILE):
+        with zipfile.ZipFile(ZIP_FILE, 'r') as zip_ref:
+            zip_ref.extractall(FIG_DIR)
+        print(f"✅ 解压 {ZIP_FILE} 成功，生成 {FIG_DIR}/ 文件夹")
+    else:
+        print(f"⚠️ {ZIP_FILE} 不存在，无法解压图像目录")
+
+@app.route("/")
+def index():
+    return "✅ Figure extraction service is running. Visit /extract to get JSON."
+
+@app.route("/extract")
+def extract():
+    figures = []
+    base_url = "https://hf.space/embed/ASC8384/P2P/file/FIG"
+
+    if not os.path.exists(FIG_DIR):
+        return jsonify({"error": "FIG folder not found."}), 500
+
+    for filename in os.listdir(FIG_DIR):
+        if filename.lower().endswith((".png", ".jpg", ".jpeg", ".gif")):
+            figures.append({
+                "filename": filename,
+                "url": f"{base_url}/{filename}",
+                "caption": f"Figure: {filename}"
+            })
+
+    return jsonify(figures)
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)