Upload 2 files

app.py

@@ -11,6 +11,7 @@ import threading
 import time
 import random
 from fastapi.responses import FileResponse, JSONResponse
+from starlette.responses import RedirectResponse
 
 # 环境变量
 ACCESS_PASSWORD = os.environ.get("ACCESS_PASSWORD", "changeme")
@@ -82,12 +83,11 @@ def generate_image_hash():
     return uuid.uuid4().hex[:12]
 
 def generate_full_url(image_hash):
-    """生成完整的图片URL（使用专用API端点）"""
+    """生成完整的图片URL（使用API端点，会重定向到实际文件）"""
     api_path = f"/api/img/{image_hash}"
     if HF_USERNAME and HF_SPACE_NAME:
         return f"https://{HF_USERNAME}-{HF_SPACE_NAME}.hf.space{api_path}"
     else:
-        # 返回相对路径，用户访问时会自动补全域名
         return api_path
 
 def keep_alive():
@@ -543,11 +543,11 @@ with gr.Blocks(title="My图床", theme=gr.themes.Soft(), css=custom_css) as grad
 # 在Gradio的底层FastAPI上添加自定义API路由
 @gradio_app.app.get("/api/img/{image_hash}")
 async def get_image(image_hash: str):
-    """通过hash获取图片"""
+    """通过hash重定向到图片文件"""
     try:
         conn = get_db_connection()
         cursor = conn.cursor()
-        cursor.execute("SELECT * FROM images WHERE hash = ?", (image_hash,))
+        cursor.execute("SELECT file_path FROM images WHERE hash = ?", (image_hash,))
         row = cursor.fetchone()
         conn.close()
         
@@ -565,15 +565,8 @@ async def get_image(image_hash: str):
                 content={"error": "Image file not found"}
             )
         
-        # 返回图片文件
-        return FileResponse(
-            file_path,
-            media_type=row['mime_type'],
-            headers={
-                "Cache-Control": "public, max-age=31536000",
-                "Content-Disposition": f'inline; filename="{row["original_filename"]}"'
-            }
-        )
+        # 重定向到Gradio的文件服务路径
+        return RedirectResponse(url=f"/file={file_path}", status_code=302)
     
     except Exception as e:
         return JSONResponse(