Update app.py

app.py

@@ -1,34 +1,31 @@
 import os
 import zipfile
-from flask import Flask, jsonify, send_from_directory
+import json
+from flask import Flask, jsonify
 
 app = Flask(__name__)
 
 FIG_DIR = "FIG"
 ZIP_FILE = "FIG.zip"
 
-# 自动解压 FIG.zip（如果 FIG 文件夹不存在）
-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} 不存在，无法解压图像目录")
+# 解压 FIG.zip（仅首次）
+if not os.path.exists(FIG_DIR) and os.path.exists(ZIP_FILE):
+    with zipfile.ZipFile(ZIP_FILE, 'r') as zip_ref:
+        zip_ref.extractall(FIG_DIR)
+    print("✅ 解压完成，生成 FIG/ 文件夹")
+else:
+    print("⚠️ FIG 文件夹已存在或 ZIP 不存在")
 
-# 根路由：确认服务运行中
 @app.route("/")
 def index():
-    return "✅ Figure extraction service is running. Visit /extract to get JSON."
+    return "✅ Service running. Try /extract"
 
-# 提供 JSON 图像信息
 @app.route("/extract")
 def extract():
     figures = []
     base_url = "https://sferewqw--poster.hf.space/file/FIG"
-
     if not os.path.exists(FIG_DIR):
-        return jsonify({"error": "❌ 图像目录 FIG/ 不存在"}), 500
+        return jsonify({"error": "FIG folder not found."}), 500
 
     for filename in os.listdir(FIG_DIR):
         if filename.lower().endswith((".png", ".jpg", ".jpeg", ".gif")):
@@ -40,12 +37,5 @@ def extract():
 
     return jsonify(figures)
 
-# 显式文件访问（可选）
-@app.route("/file/FIG/<path:filename>")
-def serve_figure(filename):
-    return send_from_directory(FIG_DIR, filename)
-
-# 启动 Flask
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)
-