Update app.py

app.py

@@ -1,35 +1,37 @@
 import os
 import zipfile
-from flask import Flask, jsonify
+from flask import Flask, jsonify, send_from_directory
 
 app = Flask(__name__)
 
 FIG_DIR = "FIG"
 ZIP_FILE = "FIG.zip"
 
-# 自动解压 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("✅ 解压 FIG.zip 成功")
+        print(f"✅ 解压 {ZIP_FILE} 成功，生成 {FIG_DIR}/ 文件夹")
     else:
-        print("⚠️ FIG.zip 不存在，跳过解压")
+        print(f"⚠️ {ZIP_FILE} 不存在，无法解压图像目录")
 
+# 根路由：确认服务运行中
 @app.route("/")
-def home():
-    return "✅ Service running. Try /extract"
+def index():
+    return "✅ Figure extraction service is running. Visit /extract to get JSON."
 
+# 提供 JSON 图像信息
 @app.route("/extract")
 def extract():
-    base_url = "https://sferewqw--poster.hf.space/file/FIG"
     figures = []
+    base_url = "https://sferewqw--poster.hf.space/file/FIG"
 
     if not os.path.exists(FIG_DIR):
-        return jsonify({"error": "FIG folder not found"}), 500
+        return jsonify({"error": "❌ 图像目录 FIG/ 不存在"}), 500
 
     for filename in os.listdir(FIG_DIR):
-        if filename.lower().endswith((".png", ".jpg", ".jpeg")):
+        if filename.lower().endswith((".png", ".jpg", ".jpeg", ".gif")):
             figures.append({
                 "filename": filename,
                 "url": f"{base_url}/{filename}",
@@ -38,6 +40,12 @@ 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)