Upload 2 files

app.py

@@ -1,632 +1,625 @@
-import gradio as gr
-import os
-import sqlite3
-import hashlib
-from datetime import datetime
-from pathlib import Path
-from PIL import Image
-import json
-import uuid
-import threading
-import time
-import random
-from fastapi import FastAPI
-from fastapi.responses import FileResponse, JSONResponse
-import uvicorn
-
-# 环境变量
-ACCESS_PASSWORD = os.environ.get("ACCESS_PASSWORD", "changeme")
-HF_USERNAME = os.environ.get("HF_USERNAME", "")
-HF_SPACE_NAME = os.environ.get("HF_SPACE_NAME", "")
-
-# 文件存储路径
-IMAGE_DIR = Path("uploaded_images")
-DB_PATH = "image_database.db"
-
-# 创建图片存储目录
-IMAGE_DIR.mkdir(exist_ok=True)
-
-def init_db():
-    """初始化数据库"""
-    try:
-        conn = sqlite3.connect(DB_PATH)
-        cursor = conn.cursor()
-        
-        # 创建图片表
-        cursor.execute("""
-            CREATE TABLE IF NOT EXISTS images (
-                id INTEGER PRIMARY KEY AUTOINCREMENT,
-                hash TEXT NOT NULL UNIQUE,
-                filename TEXT NOT NULL,
-                original_filename TEXT NOT NULL,
-                file_path TEXT NOT NULL,
-                file_size INTEGER,
-                mime_type TEXT,
-                upload_time TEXT NOT NULL,
-                description TEXT
-            )
-        """)
-        
-        # 创建hash索引以提高查询速度
-        cursor.execute("""
-            CREATE INDEX IF NOT EXISTS idx_hash ON images(hash)
-        """)
-        
-        conn.commit()
-        conn.close()
-        print("数据库初始化成功")
-        return True
-    except Exception as e:
-        print(f"数据库初始化失败: {e}")
-        return False
-
-def get_db_connection():
-    """获取数据库连接"""
-    conn = sqlite3.connect(DB_PATH)
-    conn.row_factory = sqlite3.Row
-    return conn
-
-def check_password(password):
-    """验证密码"""
-    return password == ACCESS_PASSWORD
-
-def generate_filename(original_filename):
-    """生成唯一的文件名"""
-    timestamp = datetime.now().strftime('%Y%m%d_%H%M%S_%f')
-    hash_suffix = hashlib.md5(str(datetime.now().timestamp()).encode()).hexdigest()[:8]
-    ext = Path(original_filename).suffix.lower()
-    if not ext:
-        ext = ".png"
-    return f"{timestamp}_{hash_suffix}{ext}"
-
-def generate_image_hash():
-    """生成唯一的图片hash"""
-    return uuid.uuid4().hex[:12]
-
-def generate_full_url(image_hash):
-    """生成完整的图片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():
-    """防止系统休眠的后台线程"""
-    while True:
-        # 随机等待1-2分钟
-        sleep_time = random.randint(60, 120)
-        time.sleep(sleep_time)
-        print(f"[Keep-Alive] {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
-
-def upload_images(images, description, password):
-    """批量上传图片"""
-    if not check_password(password):
-        return "密码错误", None, ""
-    
-    if not images or len(images) == 0:
-        return "请选择要上传的图片", None, ""
-    
-    results = []
-    success_count = 0
-    fail_count = 0
-    
-    try:
-        for idx, image in enumerate(images):
-            try:
-                # 获取原始文件名
-                if isinstance(image, str):
-                    # 如果是文件路径
-                    original_filename = Path(image).name
-                    img = Image.open(image)
-                else:
-                    # 如果是 PIL Image
-                    original_filename = f"uploaded_image_{idx+1}.png"
-                    img = image
-                
-                # 生成新文件名和hash
-                new_filename = generate_filename(original_filename)
-                image_hash = generate_image_hash()
-                file_path = IMAGE_DIR / new_filename
-                
-                # 保存图片
-                img.save(file_path, quality=95, optimize=True)
-                
-                # 获取文件信息
-                file_size = file_path.stat().st_size
-                mime_type = f"image/{file_path.suffix[1:]}"
-                upload_time = datetime.now().isoformat()
-                
-                # 插入数据库
-                conn = get_db_connection()
-                cursor = conn.cursor()
-                cursor.execute(
-                    """
-                    INSERT INTO images (hash, filename, original_filename, file_path, file_size, mime_type, upload_time, description)
-                    VALUES (?, ?, ?, ?, ?, ?, ?, ?)
-                    """,
-                    (image_hash, new_filename, original_filename, str(file_path), file_size, mime_type, upload_time, description or "")
-                )
-                image_id = cursor.lastrowid
-                conn.commit()
-                conn.close()
-                
-                # 生成访问 URL
-                full_url = generate_full_url(image_hash)
-                
-                results.append({
-                    'hash': image_hash,
-                    'filename': original_filename,
-                    'size': file_size,
-                    'url': full_url,
-                    'status': 'success'
-                })
-                success_count += 1
-                
-            except Exception as e:
-                results.append({
-                    'filename': f"图片 {idx+1}",
-                    'error': str(e),
-                    'status': 'failed'
-                })
-                fail_count += 1
-        
-        # 生成结果文本
-        result_text = f"## 上传结果\n\n"
-        result_text += f"成功: {success_count} 张 | 失败: {fail_count} 张\n\n"
-        
-        for r in results:
-            if r['status'] == 'success':
-                result_text += f"### {r['filename']}\n"
-                result_text += f"- **Hash**: {r['hash']}\n"
-                result_text += f"- **大小**: {r['size'] / 1024:.2f} KB\n"
-                result_text += f"- **URL**: `{r['url']}`\n\n"
-            else:
-                result_text += f"### {r['filename']}\n"
-                result_text += f"- **错误**: {r['error']}\n\n"
-        
-        # 生成URL列表（用于复制）
-        url_list = "\n".join([r['url'] for r in results if r['status'] == 'success'])
-        
-        return result_text, get_image_list_html(password), url_list
-    
-    except Exception as e:
-        return f"上传失败: {str(e)}", None, ""
-
-def get_image_list_html(password):
-    """获取图片列表（HTML格式）"""
-    if not check_password(password):
-        return "<p style='color: red;'>密码错误</p>"
-    
-    try:
-        conn = get_db_connection()
-        cursor = conn.cursor()
-        cursor.execute(
-            """
-            SELECT id, hash, filename, original_filename, file_path, file_size, upload_time, description
-            FROM images
-            ORDER BY upload_time DESC
-            """
-        )
-        rows = cursor.fetchall()
-        conn.close()
-        
-        if not rows:
-            return "<p>暂无图片</p>"
-        
-        # 生成HTML表格
-        html = """
-        <style>
-            .image-table { width: 100%; border-collapse: collapse; margin-top: 10px; }
-            .image-table th, .image-table td { padding: 12px; text-align: left; border-bottom: 1px solid #ddd; }
-            .image-table th { background-color: #f0f0f0; font-weight: bold; }
-            .image-table tr:hover { background-color: #f5f5f5; }
-            .btn-copy { background: #2196F3; color: white; padding: 6px 12px; border: none; border-radius: 4px; cursor: pointer; margin-right: 5px; }
-            .btn-delete { background: #f44336; color: white; padding: 6px 12px; border: none; border-radius: 4px; cursor: pointer; }
-            .image-hash { font-family: monospace; color: #666; font-size: 12px; }
-        </style>
-        <table class="image-table">
-            <thead>
-                <tr>
-                    <th>Hash</th>
-                    <th>文件名</th>
-                    <th>大小</th>
-                    <th>上传时间</th>
-                    <th>描述</th>
-                    <th>操作</th>
-                </tr>
-            </thead>
-            <tbody>
-        """
-        
-        for row in rows:
-            upload_time_str = datetime.fromisoformat(row['upload_time']).strftime('%Y-%m-%d %H:%M:%S')
-            full_url = generate_full_url(row['hash'])
-            
-            html += f"""
-                <tr id="row-{row['hash']}">
-                    <td class="image-hash">{row['hash']}</td>
-                    <td>{row['original_filename']}</td>
-                    <td>{row['file_size'] / 1024:.2f} KB</td>
-                    <td>{upload_time_str}</td>
-                    <td>{row['description'] or '-'}</td>
-                    <td>
-                        <button class="btn-copy" onclick="navigator.clipboard.writeText('{full_url}').then(() => alert('URL已复制'))">复制URL</button>
-                        <button class="btn-delete" data-hash="{row['hash']}" data-filename="{row['original_filename']}">删除</button>
-                    </td>
-                </tr>
-            """
-        
-        html += """
-            </tbody>
-        </table>
-        """
-        
-        return html
-    
-    except Exception as e:
-        return f"<p style='color: red;'>获取列表失败: {str(e)}</p>"
-
-def view_image_by_hash(image_hash, password):
-    """通过hash查看图片"""
-    if not check_password(password):
-        return "密码错误", None, ""
-    
-    if not image_hash:
-        return "请输入图片Hash", None, ""
-    
-    try:
-        conn = get_db_connection()
-        cursor = conn.cursor()
-        cursor.execute(
-            "SELECT * FROM images WHERE hash = ?",
-            (image_hash,)
-        )
-        row = cursor.fetchone()
-        conn.close()
-        
-        if not row:
-            return "图片不存在", None, ""
-        
-        file_path = Path(row['file_path'])
-        
-        if not file_path.exists():
-            return "图片文件已损坏或丢失", None, ""
-        
-        # 生成访问 URL
-        full_url = generate_full_url(row['hash'])
-        
-        info_text = f"""## 图片信息
-
-**Hash**: {row['hash']}
-**原始文件名**: {row['original_filename']}
-**存储文件名**: {row['filename']}
-**大小**: {row['file_size'] / 1024:.2f} KB
-**上传时间**: {row['upload_time']}
-**描述**: {row['description'] or '无'}
-
-**访问URL**:
-```
-{full_url}
-```
-"""
-        
-        return info_text, str(file_path), full_url
-    
-    except Exception as e:
-        return f"查看失败: {str(e)}", None, ""
-
-def delete_image_by_hash(image_hash, password):
-    """通过hash删除图片"""
-    if not check_password(password):
-        return "密码错误", None
-    
-    if not image_hash:
-        return "请输入图片Hash", None
-    
-    try:
-        conn = get_db_connection()
-        cursor = conn.cursor()
-        
-        # 获取图片信息
-        cursor.execute("SELECT * FROM images WHERE hash = ?", (image_hash,))
-        row = cursor.fetchone()
-        
-        if not row:
-            conn.close()
-            return "图片不存在", None
-        
-        file_path = Path(row['file_path'])
-        original_filename = row['original_filename']
-        
-        # 删除数据库记录
-        cursor.execute("DELETE FROM images WHERE hash = ?", (image_hash,))
-        conn.commit()
-        conn.close()
-        
-        # 删除文件
-        if file_path.exists():
-            file_path.unlink()
-        
-        return f"已删除图片: {original_filename}", get_image_list_html(password)
-    
-    except Exception as e:
-        return f"删除失败: {str(e)}", None
-
-def export_data(password):
-    """导出数据（仅元数据）"""
-    if not check_password(password):
-        return "密码错误", None
-    
-    try:
-        conn = get_db_connection()
-        cursor = conn.cursor()
-        cursor.execute("SELECT * FROM images ORDER BY upload_time DESC")
-        rows = cursor.fetchall()
-        conn.close()
-        
-        # 转换为 JSON
-        data = []
-        for row in rows:
-            full_url = generate_full_url(row['hash'])
-            data.append({
-                'hash': row['hash'],
-                'filename': row['filename'],
-                'original_filename': row['original_filename'],
-                'file_size': row['file_size'],
-                'upload_time': row['upload_time'],
-                'description': row['description'],
-                'url': full_url
-            })
-        
-        # 保存为文件
-        export_path = "image_metadata_export.json"
-        with open(export_path, 'w', encoding='utf-8') as f:
-            json.dump(data, f, ensure_ascii=False, indent=2)
-        
-        return f"元数据已导出到: {export_path}", export_path
-    
-    except Exception as e:
-        return f"导出失败: {str(e)}", None
-
-# 初始化数据库
-init_success = init_db()
-
-# 启动防休眠线程
-keep_alive_thread = threading.Thread(target=keep_alive, daemon=True)
-keep_alive_thread.start()
-
-# 创建FastAPI应用
-app = FastAPI(title="私人图床服务")
-
-# 添加图片访问API路由（在Gradio之前）
-@app.get("/api/img/{image_hash}")
-async def get_image(image_hash: str):
-    """通过hash获取图片"""
-    try:
-        conn = get_db_connection()
-        cursor = conn.cursor()
-        cursor.execute("SELECT * FROM images WHERE hash = ?", (image_hash,))
-        row = cursor.fetchone()
-        conn.close()
-        
-        if not row:
-            return JSONResponse(
-                status_code=404,
-                content={"error": "Image not found"}
-            )
-        
-        file_path = Path(row['file_path'])
-        
-        if not file_path.exists():
-            return JSONResponse(
-                status_code=404,
-                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"]}"'
-            }
-        )
-    
-    except Exception as e:
-        return JSONResponse(
-            status_code=500,
-            content={"error": str(e)}
-        )
-
-@app.get("/api/img/{image_hash}/info")
-async def get_image_info(image_hash: str, password: str = None):
-    """获取图片信息（需要密码）"""
-    if not password or not check_password(password):
-        return JSONResponse(
-            status_code=401,
-            content={"error": "Unauthorized"}
-        )
-    
-    try:
-        conn = get_db_connection()
-        cursor = conn.cursor()
-        cursor.execute("SELECT * FROM images WHERE hash = ?", (image_hash,))
-        row = cursor.fetchone()
-        conn.close()
-        
-        if not row:
-            return JSONResponse(
-                status_code=404,
-                content={"error": "Image not found"}
-            )
-        
-        return JSONResponse(content={
-            "hash": row['hash'],
-            "filename": row['filename'],
-            "original_filename": row['original_filename'],
-            "file_size": row['file_size'],
-            "mime_type": row['mime_type'],
-            "upload_time": row['upload_time'],
-            "description": row['description'],
-            "url": generate_full_url(row['hash'])
-        })
-    
-    except Exception as e:
-        return JSONResponse(
-            status_code=500,
-            content={"error": str(e)}
-        )
-
-# 自定义CSS
-custom_css = """
-.compact-gallery .gallery {
-    grid-template-columns: repeat(auto-fill, minmax(150px, 1fr)) !important;
-}
-.url-box textarea {
-    font-family: monospace;
-    font-size: 12px;
-}
-"""
-
-# 创建 Gradio 界面
-with gr.Blocks(title="My图床", theme=gr.themes.Soft(), css=custom_css) as gradio_app:
-    gr.Markdown("# My图床服务")
-    
-    # 全局密码输入
-    with gr.Row():
-        global_password = gr.Textbox(
-            label="访问密码", 
-            type="password", 
-            placeholder="输入密码以使用所有功能",
-            scale=4
-        )
-    
-    if not init_success:
-        gr.Markdown("## 数据库初始化失败")
-    else:
-        with gr.Tabs():
-            # 上传图片标签页
-            with gr.Tab("上传图片"):
-                with gr.Row():
-                    with gr.Column(scale=1):
-                        upload_images_input = gr.File(
-                            label="选择图片（支持多选）",
-                            file_count="multiple",
-                            file_types=["image"]
-                        )
-                        upload_desc_input = gr.Textbox(
-                            label="描述（可选）", 
-                            placeholder="为这批图片添加描述...",
-                            lines=2
-                        )
-                        upload_btn = gr.Button("上传", variant="primary")
-                    
-                    with gr.Column(scale=1):
-                        upload_output = gr.Markdown(label="上传结果")
-                        upload_url_output = gr.Textbox(
-                            label="图片URL列表（可复制）", 
-                            lines=5,
-                            elem_classes=["url-box"]
-                        )
-                
-                # 预览区域
-                with gr.Accordion("图片预览", open=False):
-                    upload_preview = gr.Gallery(
-                        label="上传预览",
-                        show_label=False,
-                        columns=4,
-                        rows=2,
-                        height="auto",
-                        elem_classes=["compact-gallery"]
-                    )
-                
-                # 绑定事件
-                upload_images_input.change(
-                    lambda files: [f.name for f in files] if files else [],
-                    inputs=[upload_images_input],
-                    outputs=[upload_preview]
-                )
-                
-                upload_btn.click(
-                    lambda files, desc, pwd: upload_images([f.name for f in files] if files else [], desc, pwd),
-                    inputs=[upload_images_input, upload_desc_input, global_password],
-                    outputs=[upload_output, gr.HTML(visible=False), upload_url_output]
-                )
-            
-            # 图片列表标签页
-            with gr.Tab("图片列表"):
-                list_refresh_btn = gr.Button("刷新列表", variant="primary")
-                list_output = gr.HTML(label="图片列表")
-                
-                gr.Markdown("---")
-                gr.Markdown("### 查看图片详情")
-                
-                with gr.Row():
-                    with gr.Column():
-                        view_hash_input = gr.Textbox(
-                            label="图片Hash", 
-                            placeholder="输入图片Hash查看详情"
-                        )
-                        view_btn = gr.Button("查看详情", variant="secondary")
-                        view_info_output = gr.Markdown(label="图片信息")
-                        view_url_output = gr.Textbox(
-                            label="图片URL",
-                            lines=2,
-                            elem_classes=["url-box"]
-                        )
-                    
-                    with gr.Column():
-                        view_image_output = gr.Image(label="图片预览", height=400)
-                
-                # 删除图片区域
-                gr.Markdown("---")
-                gr.Markdown("### 删除图片")
-                with gr.Row():
-                    delete_hash_input = gr.Textbox(
-                        label="图片Hash", 
-                        placeholder="输入要删除的图片Hash"
-                    )
-                    delete_btn = gr.Button("删除", variant="stop")
-                delete_output = gr.Textbox(label="删除结果", lines=2)
-                
-                # 绑定事件
-                list_refresh_btn.click(
-                    get_image_list_html,
-                    inputs=[global_password],
-                    outputs=[list_output]
-                )
-                
-                view_btn.click(
-                    view_image_by_hash,
-                    inputs=[view_hash_input, global_password],
-                    outputs=[view_info_output, view_image_output, view_url_output]
-                )
-                
-                delete_btn.click(
-                    delete_image_by_hash,
-                    inputs=[delete_hash_input, global_password],
-                    outputs=[delete_output, list_output]
-                )
-            
-            # 数据导出标签页
-            with gr.Tab("导出数据"):
-                gr.Markdown("导出所有图片的元数据（包含完整URL）为JSON格式")
-                
-                export_btn = gr.Button("导出元数据（JSON）", variant="primary")
-                export_output = gr.Textbox(label="导出结果", lines=2)
-                export_file_output = gr.File(label="下载文件")
-                
-                export_btn.click(
-                    export_data,
-                    inputs=[global_password],
-                    outputs=[export_output, export_file_output]
-                )
-
-# 将Gradio应用挂载到FastAPI
-app = gr.mount_gradio_app(app, gradio_app, path="/")
-
-if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+import gradio as gr
+import os
+import sqlite3
+import hashlib
+from datetime import datetime
+from pathlib import Path
+from PIL import Image
+import json
+import uuid
+import threading
+import time
+import random
+from fastapi.responses import FileResponse, JSONResponse
+
+# 环境变量
+ACCESS_PASSWORD = os.environ.get("ACCESS_PASSWORD", "changeme")
+HF_USERNAME = os.environ.get("HF_USERNAME", "")
+HF_SPACE_NAME = os.environ.get("HF_SPACE_NAME", "")
+
+# 文件存储路径
+IMAGE_DIR = Path("uploaded_images")
+DB_PATH = "image_database.db"
+
+# 创建图片存储目录
+IMAGE_DIR.mkdir(exist_ok=True)
+
+def init_db():
+    """初始化数据库"""
+    try:
+        conn = sqlite3.connect(DB_PATH)
+        cursor = conn.cursor()
+        
+        # 创建图片表
+        cursor.execute("""
+            CREATE TABLE IF NOT EXISTS images (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                hash TEXT NOT NULL UNIQUE,
+                filename TEXT NOT NULL,
+                original_filename TEXT NOT NULL,
+                file_path TEXT NOT NULL,
+                file_size INTEGER,
+                mime_type TEXT,
+                upload_time TEXT NOT NULL,
+                description TEXT
+            )
+        """)
+        
+        # 创建hash索引以提高查询速度
+        cursor.execute("""
+            CREATE INDEX IF NOT EXISTS idx_hash ON images(hash)
+        """)
+        
+        conn.commit()
+        conn.close()
+        print("数据库初始化成功")
+        return True
+    except Exception as e:
+        print(f"数据库初始化失败: {e}")
+        return False
+
+def get_db_connection():
+    """获取数据库连接"""
+    conn = sqlite3.connect(DB_PATH)
+    conn.row_factory = sqlite3.Row
+    return conn
+
+def check_password(password):
+    """验证密码"""
+    return password == ACCESS_PASSWORD
+
+def generate_filename(original_filename):
+    """生成唯一的文件名"""
+    timestamp = datetime.now().strftime('%Y%m%d_%H%M%S_%f')
+    hash_suffix = hashlib.md5(str(datetime.now().timestamp()).encode()).hexdigest()[:8]
+    ext = Path(original_filename).suffix.lower()
+    if not ext:
+        ext = ".png"
+    return f"{timestamp}_{hash_suffix}{ext}"
+
+def generate_image_hash():
+    """生成唯一的图片hash"""
+    return uuid.uuid4().hex[:12]
+
+def generate_full_url(image_hash):
+    """生成完整的图片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():
+    """防止系统休眠的后台线程"""
+    while True:
+        # 随机等待1-2分钟
+        sleep_time = random.randint(60, 120)
+        time.sleep(sleep_time)
+        print(f"[Keep-Alive] {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
+
+def upload_images(images, description, password):
+    """批量上传图片"""
+    if not check_password(password):
+        return "密码错误", None, ""
+    
+    if not images or len(images) == 0:
+        return "请选择要上传的图片", None, ""
+    
+    results = []
+    success_count = 0
+    fail_count = 0
+    
+    try:
+        for idx, image in enumerate(images):
+            try:
+                # 获取原始文件名
+                if isinstance(image, str):
+                    # 如果是文件路径
+                    original_filename = Path(image).name
+                    img = Image.open(image)
+                else:
+                    # 如果是 PIL Image
+                    original_filename = f"uploaded_image_{idx+1}.png"
+                    img = image
+                
+                # 生成新文件名和hash
+                new_filename = generate_filename(original_filename)
+                image_hash = generate_image_hash()
+                file_path = IMAGE_DIR / new_filename
+                
+                # 保存图片
+                img.save(file_path, quality=95, optimize=True)
+                
+                # 获取文件信息
+                file_size = file_path.stat().st_size
+                mime_type = f"image/{file_path.suffix[1:]}"
+                upload_time = datetime.now().isoformat()
+                
+                # 插入数据库
+                conn = get_db_connection()
+                cursor = conn.cursor()
+                cursor.execute(
+                    """
+                    INSERT INTO images (hash, filename, original_filename, file_path, file_size, mime_type, upload_time, description)
+                    VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+                    """,
+                    (image_hash, new_filename, original_filename, str(file_path), file_size, mime_type, upload_time, description or "")
+                )
+                image_id = cursor.lastrowid
+                conn.commit()
+                conn.close()
+                
+                # 生成访问 URL
+                full_url = generate_full_url(image_hash)
+                
+                results.append({
+                    'hash': image_hash,
+                    'filename': original_filename,
+                    'size': file_size,
+                    'url': full_url,
+                    'status': 'success'
+                })
+                success_count += 1
+                
+            except Exception as e:
+                results.append({
+                    'filename': f"图片 {idx+1}",
+                    'error': str(e),
+                    'status': 'failed'
+                })
+                fail_count += 1
+        
+        # 生成结果文本
+        result_text = f"## 上传结果\n\n"
+        result_text += f"成功: {success_count} 张 | 失败: {fail_count} 张\n\n"
+        
+        for r in results:
+            if r['status'] == 'success':
+                result_text += f"### {r['filename']}\n"
+                result_text += f"- **Hash**: {r['hash']}\n"
+                result_text += f"- **大小**: {r['size'] / 1024:.2f} KB\n"
+                result_text += f"- **URL**: `{r['url']}`\n\n"
+            else:
+                result_text += f"### {r['filename']}\n"
+                result_text += f"- **错误**: {r['error']}\n\n"
+        
+        # 生成URL列表（用于复制）
+        url_list = "\n".join([r['url'] for r in results if r['status'] == 'success'])
+        
+        return result_text, get_image_list_html(password), url_list
+    
+    except Exception as e:
+        return f"上传失败: {str(e)}", None, ""
+
+def get_image_list_html(password):
+    """获取图片列表（HTML格式）"""
+    if not check_password(password):
+        return "<p style='color: red;'>密码错误</p>"
+    
+    try:
+        conn = get_db_connection()
+        cursor = conn.cursor()
+        cursor.execute(
+            """
+            SELECT id, hash, filename, original_filename, file_path, file_size, upload_time, description
+            FROM images
+            ORDER BY upload_time DESC
+            """
+        )
+        rows = cursor.fetchall()
+        conn.close()
+        
+        if not rows:
+            return "<p>暂无图片</p>"
+        
+        # 生成HTML表格
+        html = """
+        <style>
+            .image-table { width: 100%; border-collapse: collapse; margin-top: 10px; }
+            .image-table th, .image-table td { padding: 12px; text-align: left; border-bottom: 1px solid #ddd; }
+            .image-table th { background-color: #f0f0f0; font-weight: bold; }
+            .image-table tr:hover { background-color: #f5f5f5; }
+            .btn-copy { background: #2196F3; color: white; padding: 6px 12px; border: none; border-radius: 4px; cursor: pointer; margin-right: 5px; }
+            .btn-delete { background: #f44336; color: white; padding: 6px 12px; border: none; border-radius: 4px; cursor: pointer; }
+            .image-hash { font-family: monospace; color: #666; font-size: 12px; }
+        </style>
+        <table class="image-table">
+            <thead>
+                <tr>
+                    <th>Hash</th>
+                    <th>文件名</th>
+                    <th>大小</th>
+                    <th>上传时间</th>
+                    <th>描述</th>
+                    <th>操作</th>
+                </tr>
+            </thead>
+            <tbody>
+        """
+        
+        for row in rows:
+            upload_time_str = datetime.fromisoformat(row['upload_time']).strftime('%Y-%m-%d %H:%M:%S')
+            full_url = generate_full_url(row['hash'])
+            
+            html += f"""
+                <tr id="row-{row['hash']}">
+                    <td class="image-hash">{row['hash']}</td>
+                    <td>{row['original_filename']}</td>
+                    <td>{row['file_size'] / 1024:.2f} KB</td>
+                    <td>{upload_time_str}</td>
+                    <td>{row['description'] or '-'}</td>
+                    <td>
+                        <button class="btn-copy" onclick="navigator.clipboard.writeText('{full_url}').then(() => alert('URL已复制'))">复制URL</button>
+                        <button class="btn-delete" data-hash="{row['hash']}" data-filename="{row['original_filename']}">删除</button>
+                    </td>
+                </tr>
+            """
+        
+        html += """
+            </tbody>
+        </table>
+        """
+        
+        return html
+    
+    except Exception as e:
+        return f"<p style='color: red;'>获取列表失败: {str(e)}</p>"
+
+def view_image_by_hash(image_hash, password):
+    """通过hash查看图片"""
+    if not check_password(password):
+        return "密码错误", None, ""
+    
+    if not image_hash:
+        return "请输入图片Hash", None, ""
+    
+    try:
+        conn = get_db_connection()
+        cursor = conn.cursor()
+        cursor.execute(
+            "SELECT * FROM images WHERE hash = ?",
+            (image_hash,)
+        )
+        row = cursor.fetchone()
+        conn.close()
+        
+        if not row:
+            return "图片不存在", None, ""
+        
+        file_path = Path(row['file_path'])
+        
+        if not file_path.exists():
+            return "图片文件已损坏或丢失", None, ""
+        
+        # 生成访问 URL
+        full_url = generate_full_url(row['hash'])
+        
+        info_text = f"""## 图片信息
+
+**Hash**: {row['hash']}
+**原始文件名**: {row['original_filename']}
+**存储文件名**: {row['filename']}
+**大小**: {row['file_size'] / 1024:.2f} KB
+**上传时间**: {row['upload_time']}
+**描述**: {row['description'] or '无'}
+
+**访问URL**:
+```
+{full_url}
+```
+"""
+        
+        return info_text, str(file_path), full_url
+    
+    except Exception as e:
+        return f"查看失败: {str(e)}", None, ""
+
+def delete_image_by_hash(image_hash, password):
+    """通过hash删除图片"""
+    if not check_password(password):
+        return "密码错误", None
+    
+    if not image_hash:
+        return "请输入图片Hash", None
+    
+    try:
+        conn = get_db_connection()
+        cursor = conn.cursor()
+        
+        # 获取图片信息
+        cursor.execute("SELECT * FROM images WHERE hash = ?", (image_hash,))
+        row = cursor.fetchone()
+        
+        if not row:
+            conn.close()
+            return "图片不存在", None
+        
+        file_path = Path(row['file_path'])
+        original_filename = row['original_filename']
+        
+        # 删除数据库记录
+        cursor.execute("DELETE FROM images WHERE hash = ?", (image_hash,))
+        conn.commit()
+        conn.close()
+        
+        # 删除文件
+        if file_path.exists():
+            file_path.unlink()
+        
+        return f"已删除图片: {original_filename}", get_image_list_html(password)
+    
+    except Exception as e:
+        return f"删除失败: {str(e)}", None
+
+def export_data(password):
+    """导出数据（仅元数据）"""
+    if not check_password(password):
+        return "密码错误", None
+    
+    try:
+        conn = get_db_connection()
+        cursor = conn.cursor()
+        cursor.execute("SELECT * FROM images ORDER BY upload_time DESC")
+        rows = cursor.fetchall()
+        conn.close()
+        
+        # 转换为 JSON
+        data = []
+        for row in rows:
+            full_url = generate_full_url(row['hash'])
+            data.append({
+                'hash': row['hash'],
+                'filename': row['filename'],
+                'original_filename': row['original_filename'],
+                'file_size': row['file_size'],
+                'upload_time': row['upload_time'],
+                'description': row['description'],
+                'url': full_url
+            })
+        
+        # 保存为文件
+        export_path = "image_metadata_export.json"
+        with open(export_path, 'w', encoding='utf-8') as f:
+            json.dump(data, f, ensure_ascii=False, indent=2)
+        
+        return f"元数据已导出到: {export_path}", export_path
+    
+    except Exception as e:
+        return f"导出失败: {str(e)}", None
+
+# 初始化数据库
+init_success = init_db()
+
+# 启动防休眠线程
+keep_alive_thread = threading.Thread(target=keep_alive, daemon=True)
+keep_alive_thread.start()
+
+# 自定义CSS
+custom_css = """
+.compact-gallery .gallery {
+    grid-template-columns: repeat(auto-fill, minmax(150px, 1fr)) !important;
+}
+.url-box textarea {
+    font-family: monospace;
+    font-size: 12px;
+}
+"""
+
+# 创建 Gradio 界面
+with gr.Blocks(title="My图床", theme=gr.themes.Soft(), css=custom_css) as gradio_app:
+    gr.Markdown("# My图床服务")
+    
+    # 全局密码输入
+    with gr.Row():
+        global_password = gr.Textbox(
+            label="访问密码", 
+            type="password", 
+            placeholder="输入密码以使用所有功能",
+            scale=4
+        )
+    
+    if not init_success:
+        gr.Markdown("## 数据库初始化失败")
+    else:
+        with gr.Tabs():
+            # 上传图片标签页
+            with gr.Tab("上传图片"):
+                with gr.Row():
+                    with gr.Column(scale=1):
+                        upload_images_input = gr.File(
+                            label="选择图片（支持多选）",
+                            file_count="multiple",
+                            file_types=["image"]
+                        )
+                        upload_desc_input = gr.Textbox(
+                            label="描述（可选）", 
+                            placeholder="为这批图片添加描述...",
+                            lines=2
+                        )
+                        upload_btn = gr.Button("上传", variant="primary")
+                    
+                    with gr.Column(scale=1):
+                        upload_output = gr.Markdown(label="上传结果")
+                        upload_url_output = gr.Textbox(
+                            label="图片URL列表（可复制）", 
+                            lines=5,
+                            elem_classes=["url-box"]
+                        )
+                
+                # 预览区域
+                with gr.Accordion("图片预览", open=False):
+                    upload_preview = gr.Gallery(
+                        label="上传预览",
+                        show_label=False,
+                        columns=4,
+                        rows=2,
+                        height="auto",
+                        elem_classes=["compact-gallery"]
+                    )
+                
+                # 绑定事件
+                upload_images_input.change(
+                    lambda files: [f.name for f in files] if files else [],
+                    inputs=[upload_images_input],
+                    outputs=[upload_preview]
+                )
+                
+                upload_btn.click(
+                    lambda files, desc, pwd: upload_images([f.name for f in files] if files else [], desc, pwd),
+                    inputs=[upload_images_input, upload_desc_input, global_password],
+                    outputs=[upload_output, gr.HTML(visible=False), upload_url_output]
+                )
+            
+            # 图片列表标签页
+            with gr.Tab("图片列表"):
+                list_refresh_btn = gr.Button("刷新列表", variant="primary")
+                list_output = gr.HTML(label="图片列表")
+                
+                gr.Markdown("---")
+                gr.Markdown("### 查看图片详情")
+                
+                with gr.Row():
+                    with gr.Column():
+                        view_hash_input = gr.Textbox(
+                            label="图片Hash", 
+                            placeholder="输入图片Hash查看详情"
+                        )
+                        view_btn = gr.Button("查看详情", variant="secondary")
+                        view_info_output = gr.Markdown(label="图片信息")
+                        view_url_output = gr.Textbox(
+                            label="图片URL",
+                            lines=2,
+                            elem_classes=["url-box"]
+                        )
+                    
+                    with gr.Column():
+                        view_image_output = gr.Image(label="图片预览", height=400)
+                
+                # 删除图片区域
+                gr.Markdown("---")
+                gr.Markdown("### 删除图片")
+                with gr.Row():
+                    delete_hash_input = gr.Textbox(
+                        label="图片Hash", 
+                        placeholder="输入要删除的图片Hash"
+                    )
+                    delete_btn = gr.Button("删除", variant="stop")
+                delete_output = gr.Textbox(label="删除结果", lines=2)
+                
+                # 绑定事件
+                list_refresh_btn.click(
+                    get_image_list_html,
+                    inputs=[global_password],
+                    outputs=[list_output]
+                )
+                
+                view_btn.click(
+                    view_image_by_hash,
+                    inputs=[view_hash_input, global_password],
+                    outputs=[view_info_output, view_image_output, view_url_output]
+                )
+                
+                delete_btn.click(
+                    delete_image_by_hash,
+                    inputs=[delete_hash_input, global_password],
+                    outputs=[delete_output, list_output]
+                )
+            
+            # 数据导出标签页
+            with gr.Tab("导出数据"):
+                gr.Markdown("导出所有图片的元数据（包含完整URL）为JSON格式")
+                
+                export_btn = gr.Button("导出元数据（JSON）", variant="primary")
+                export_output = gr.Textbox(label="导出结果", lines=2)
+                export_file_output = gr.File(label="下载文件")
+                
+                export_btn.click(
+                    export_data,
+                    inputs=[global_password],
+                    outputs=[export_output, export_file_output]
+                )
+
+# 在Gradio的底层FastAPI上添加自定义API路由
+@gradio_app.fastapi_app.get("/api/img/{image_hash}")
+async def get_image(image_hash: str):
+    """通过hash获取图片"""
+    try:
+        conn = get_db_connection()
+        cursor = conn.cursor()
+        cursor.execute("SELECT * FROM images WHERE hash = ?", (image_hash,))
+        row = cursor.fetchone()
+        conn.close()
+        
+        if not row:
+            return JSONResponse(
+                status_code=404,
+                content={"error": "Image not found"}
+            )
+        
+        file_path = Path(row['file_path'])
+        
+        if not file_path.exists():
+            return JSONResponse(
+                status_code=404,
+                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"]}"'
+            }
+        )
+    
+    except Exception as e:
+        return JSONResponse(
+            status_code=500,
+            content={"error": str(e)}
+        )
+
+@gradio_app.fastapi_app.get("/api/img/{image_hash}/info")
+async def get_image_info(image_hash: str, password: str = None):
+    """获取图片信息（需要密码）"""
+    if not password or not check_password(password):
+        return JSONResponse(
+            status_code=401,
+            content={"error": "Unauthorized"}
+        )
+    
+    try:
+        conn = get_db_connection()
+        cursor = conn.cursor()
+        cursor.execute("SELECT * FROM images WHERE hash = ?", (image_hash,))
+        row = cursor.fetchone()
+        conn.close()
+        
+        if not row:
+            return JSONResponse(
+                status_code=404,
+                content={"error": "Image not found"}
+            )
+        
+        return JSONResponse(content={
+            "hash": row['hash'],
+            "filename": row['filename'],
+            "original_filename": row['original_filename'],
+            "file_size": row['file_size'],
+            "mime_type": row['mime_type'],
+            "upload_time": row['upload_time'],
+            "description": row['description'],
+            "url": generate_full_url(row['hash'])
+        })
+    
+    except Exception as e:
+        return JSONResponse(
+            status_code=500,
+            content={"error": str(e)}
+        )
+
+if __name__ == "__main__":
+    # 直接启动Gradio，它会使用自己的FastAPI实例
+    gradio_app.launch(server_name="0.0.0.0", server_port=7860)

requirements.txt

@@ -1,5 +1,3 @@
 gradio==4.44.0
 Pillow==10.4.0
-fastapi>=0.104.0
-uvicorn[standard]>=0.24.0