Update app.py

app.py

@@ -1,30 +1,35 @@
-from flask import Flask, jsonify
 import os
-import zipfile
+import json
+from flask import Flask, jsonify
+from PIL import Image
+from zipfile import ZipFile
 
 app = Flask(__name__)
 
+# FIX: 使用 /data/ 路径来避免权限错误
 FIG_DIR = "/data/FIG"
-ZIP_FILE = "/data/FIG.zip"  # 你的 zip 应该传到了这个路径
+ZIP_PATH = "/app/FIG.zip"  # 上传的图像包路径
 
-# 解压缩逻辑（只解压一次）
+# 解压 FIG.zip 到 FIG_DIR（只运行一次）
 if not os.path.exists(FIG_DIR):
     os.makedirs(FIG_DIR, exist_ok=True)
-    with zipfile.ZipFile(ZIP_FILE, 'r') as zip_ref:
+    with ZipFile(ZIP_PATH, 'r') as zip_ref:
         zip_ref.extractall(FIG_DIR)
 
-@app.route("/")
-def index():
-    return "Figure extraction service is running. Visit /extract to get JSON."
+@app.route('/')
+def home():
+    return "✅ Figure extraction service is running. Visit /extract to get JSON."
 
-@app.route("/extract")
-def extract():
+@app.route('/extract')
+def extract_figures():
     results = []
-    for filename in sorted(os.listdir(FIG_DIR)):
-        if filename.lower().endswith((".png", ".jpg", ".jpeg")):
+    for filename in os.listdir(FIG_DIR):
+        if filename.lower().endswith(('.png', '.jpg', '.jpeg', '.gif')):
+            image_url = f"https://{os.environ.get('SPACE_ID')}.hf.space/file/FIG/{filename}"
+            caption = os.path.splitext(filename)[0]
             results.append({
-                "image_url": f"https://{os.environ['HOSTNAME']}.hf.space/file/FIG/{filename}",
-                "caption": filename.rsplit(".", 1)[0].replace("_", " ")
+                "image": image_url,
+                "caption": caption
             })
     return jsonify(results)