Upload 2 files

app.py

@@ -1,11 +1,10 @@
-import gradio as gr
+from flask import Flask, render_template_string, request, send_file, jsonify, redirect, url_for, session
 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
@@ -17,22 +16,22 @@ ACCESS_PASSWORD = os.environ.get("ACCESS_PASSWORD", "changeme")
 HF_USERNAME = os.environ.get("HF_USERNAME", "")
 HF_SPACE_NAME = os.environ.get("HF_SPACE_NAME", "")
 
+# Flask应用
+app = Flask(__name__)
+app.secret_key = os.environ.get("ACCESS_PASSWORD", "your-secret-key-change-this")
+
 # 文件存储路径
 IMAGE_DIR = Path("uploaded_images")
-PUBLIC_DIR = Path("public")  # 公共访问目录，使用hash作为文件名
 DB_PATH = "image_database.db"
 
-# 创建图片存储目录
+# 创建目录
 IMAGE_DIR.mkdir(exist_ok=True)
-PUBLIC_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,
@@ -46,12 +45,9 @@ def init_db():
                 description TEXT
             )
         """)
-        
-        # 创建hash索引以提高查询速度
         cursor.execute("""
             CREATE INDEX IF NOT EXISTS idx_hash ON images(hash)
         """)
-        
         conn.commit()
         conn.close()
         print("数据库初始化成功")
@@ -61,17 +57,17 @@ def init_db():
         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_image_hash():
+    return uuid.uuid4().hex[:12]
+
 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()
@@ -79,324 +75,399 @@ def generate_filename(original_filename):
         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（直接访问public目录）"""
-    # 查找对应的文件扩展名
-    try:
-        conn = get_db_connection()
-        cursor = conn.cursor()
-        cursor.execute("SELECT file_path FROM images WHERE hash = ?", (image_hash,))
-        row = cursor.fetchone()
-        conn.close()
-        
-        if row:
-            ext = Path(row['file_path']).suffix
-            public_path = PUBLIC_DIR / f"{image_hash}{ext}"
-            file_path = f"/file={public_path}"
-            
-            if HF_USERNAME and HF_SPACE_NAME:
-                return f"https://{HF_USERNAME}-{HF_SPACE_NAME}.hf.space{file_path}"
-            else:
-                return file_path
-    except:
-        pass
-    
-    # 默认返回
-    return f"/file=public/{image_hash}"
+    """生成完整的图片URL"""
+    if HF_USERNAME and HF_SPACE_NAME:
+        return f"https://{HF_USERNAME}-{HF_SPACE_NAME}.hf.space/img/{image_hash}"
+    else:
+        return f"/img/{image_hash}"
 
 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)
-                
-                # 在public目录创建副本，文件名为hash
-                ext = file_path.suffix
-                public_file = PUBLIC_DIR / f"{image_hash}{ext}"
-                shutil.copy2(file_path, public_file)
-                
-                # 获取文件信息
-                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()
+# 初始化
+init_db()
+keep_alive_thread = threading.Thread(target=keep_alive, daemon=True)
+keep_alive_thread.start()
+
+# HTML模板
+HTML_TEMPLATE = """
+<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>My图床服务</title>
+    <style>
+        * { margin: 0; padding: 0; box-sizing: border-box; }
+        body { font-family: Arial, sans-serif; padding: 20px; background: #f5f5f5; }
+        .container { max-width: 1200px; margin: 0 auto; background: white; padding: 30px; border-radius: 8px; box-shadow: 0 2px 10px rgba(0,0,0,0.1); }
+        h1 { margin-bottom: 10px; }
+        .auth-box { margin: 20px 0; padding: 15px; background: #f9f9f9; border-radius: 5px; }
+        .auth-box input { padding: 10px; width: 300px; border: 1px solid #ddd; border-radius: 4px; }
+        .tabs { display: flex; gap: 10px; margin: 20px 0; border-bottom: 2px solid #ddd; }
+        .tab { padding: 10px 20px; cursor: pointer; border: none; background: none; font-size: 16px; }
+        .tab.active { border-bottom: 3px solid #007bff; color: #007bff; }
+        .tab-content { display: none; padding: 20px 0; }
+        .tab-content.active { display: block; }
+        .upload-area { border: 2px dashed #ddd; padding: 40px; text-align: center; border-radius: 8px; margin: 20px 0; }
+        .upload-area:hover { border-color: #007bff; background: #f9f9f9; }
+        input[type="file"] { display: none; }
+        .btn { padding: 10px 20px; background: #007bff; color: white; border: none; border-radius: 4px; cursor: pointer; font-size: 14px; }
+        .btn:hover { background: #0056b3; }
+        .btn-danger { background: #dc3545; }
+        .btn-danger:hover { background: #c82333; }
+        textarea, input[type="text"] { width: 100%; padding: 10px; border: 1px solid #ddd; border-radius: 4px; margin: 10px 0; }
+        .result { margin: 20px 0; padding: 15px; background: #e7f3ff; border-radius: 5px; white-space: pre-wrap; word-break: break-all; }
+        table { width: 100%; border-collapse: collapse; margin: 20px 0; }
+        th, td { padding: 12px; text-align: left; border-bottom: 1px solid #ddd; }
+        th { background: #f0f0f0; font-weight: bold; }
+        tr:hover { background: #f9f9f9; }
+        .hash { font-family: monospace; font-size: 12px; color: #666; }
+        .btn-small { padding: 6px 12px; font-size: 12px; margin: 0 2px; }
+    </style>
+</head>
+<body>
+    <div class="container">
+        <h1>My图床服务</h1>
+        
+        <div class="auth-box">
+            <label>访问密码：</label>
+            <input type="password" id="password" placeholder="输入密码以使用所有功能">
+        </div>
+
+        <div class="tabs">
+            <button class="tab active" onclick="showTab('upload')">上传图片</button>
+            <button class="tab" onclick="showTab('list')">图片列表</button>
+            <button class="tab" onclick="showTab('export')">导出数据</button>
+        </div>
+
+        <div id="upload" class="tab-content active">
+            <h2>上传图片</h2>
+            <div class="upload-area" onclick="document.getElementById('fileInput').click()">
+                <p>点击选择图片（支持多选）</p>
+                <input type="file" id="fileInput" multiple accept="image/*">
+            </div>
+            <div>
+                <label>描述（可选）：</label>
+                <textarea id="description" rows="3" placeholder="为这批图片添加描述..."></textarea>
+            </div>
+            <button class="btn" onclick="uploadImages()">上传</button>
+            <div id="uploadResult" class="result" style="display:none;"></div>
+        </div>
+
+        <div id="list" class="tab-content">
+            <h2>图片列表</h2>
+            <button class="btn" onclick="loadImages()">刷新列表</button>
+            <div id="imageList"></div>
+        </div>
+
+        <div id="export" class="tab-content">
+            <h2>导出数据</h2>
+            <p>导出所有图片的��数据（包含完整URL）为JSON格式</p>
+            <button class="btn" onclick="exportData()">导出元数据</button>
+            <div id="exportResult" class="result" style="display:none;"></div>
+        </div>
+    </div>
+
+    <script>
+        function showTab(tabName) {
+            document.querySelectorAll('.tab-content').forEach(el => el.classList.remove('active'));
+            document.querySelectorAll('.tab').forEach(el => el.classList.remove('active'));
+            document.getElementById(tabName).classList.add('active');
+            event.target.classList.add('active');
+        }
+
+        async function uploadImages() {
+            const password = document.getElementById('password').value;
+            const files = document.getElementById('fileInput').files;
+            const description = document.getElementById('description').value;
+            
+            if (!password) {
+                alert('请输入密码');
+                return;
+            }
+            if (files.length === 0) {
+                alert('请选择图片');
+                return;
+            }
+
+            const formData = new FormData();
+            formData.append('password', password);
+            formData.append('description', description);
+            for (let file of files) {
+                formData.append('images', file);
+            }
+
+            try {
+                const response = await fetch('/api/upload', {
+                    method: 'POST',
+                    body: formData
+                });
+                const result = await response.json();
+                const resultDiv = document.getElementById('uploadResult');
+                resultDiv.style.display = 'block';
                 
-                # 生成访问 URL
-                full_url = generate_full_url(image_hash)
+                if (result.success) {
+                    let html = `成功上传 ${result.data.length} 张图片:\\n\\n`;
+                    result.data.forEach(img => {
+                        html += `Hash: ${img.hash}\\nURL: ${img.url}\\n\\n`;
+                    });
+                    resultDiv.textContent = html;
+                } else {
+                    resultDiv.textContent = '错误: ' + result.error;
+                }
+            } catch (error) {
+                alert('上传失败: ' + error);
+            }
+        }
+
+        async function loadImages() {
+            const password = document.getElementById('password').value;
+            if (!password) {
+                alert('请输入密码');
+                return;
+            }
+
+            try {
+                const response = await fetch(`/api/images?password=${password}`);
+                const result = await response.json();
+                const listDiv = document.getElementById('imageList');
                 
-                results.append({
-                    'hash': image_hash,
-                    'filename': original_filename,
-                    'size': file_size,
-                    'url': full_url,
-                    'status': 'success'
-                })
-                success_count += 1
+                if (result.success) {
+                    if (result.data.length === 0) {
+                        listDiv.innerHTML = '<p>暂无图片</p>';
+                        return;
+                    }
+                    
+                    let html = '<table><thead><tr><th>Hash</th><th>文件名</th><th>大小</th><th>上传时间</th><th>操作</th></tr></thead><tbody>';
+                    result.data.forEach(img => {
+                        html += `<tr>
+                            <td class="hash">${img.hash}</td>
+                            <td>${img.original_filename}</td>
+                            <td>${(img.file_size / 1024).toFixed(2)} KB</td>
+                            <td>${img.upload_time}</td>
+                            <td>
+                                <button class="btn btn-small" onclick="copyUrl('${img.url}')">复制URL</button>
+                                <button class="btn btn-small btn-danger" onclick="deleteImage('${img.hash}')">删除</button>
+                            </td>
+                        </tr>`;
+                    });
+                    html += '</tbody></table>';
+                    listDiv.innerHTML = html;
+                } else {
+                    listDiv.innerHTML = '<p>错误: ' + result.error + '</p>';
+                }
+            } catch (error) {
+                alert('加载失败: ' + error);
+            }
+        }
+
+        function copyUrl(url) {
+            navigator.clipboard.writeText(url).then(() => alert('URL已复制到剪贴板'));
+        }
+
+        async function deleteImage(hash) {
+            if (!confirm('确定要删除这张图片吗？')) return;
+            
+            const password = document.getElementById('password').value;
+            try {
+                const response = await fetch(`/api/delete/${hash}`, {
+                    method: 'POST',
+                    headers: {'Content-Type': 'application/json'},
+                    body: JSON.stringify({password})
+                });
+                const result = await response.json();
+                if (result.success) {
+                    alert('删除成功');
+                    loadImages();
+                } else {
+                    alert('删除失败: ' + result.error);
+                }
+            } catch (error) {
+                alert('删除失败: ' + error);
+            }
+        }
+
+        async function exportData() {
+            const password = document.getElementById('password').value;
+            if (!password) {
+                alert('请输入密码');
+                return;
+            }
+
+            try {
+                const response = await fetch(`/api/export?password=${password}`);
+                const blob = await response.blob();
+                const url = window.URL.createObjectURL(blob);
+                const a = document.createElement('a');
+                a.href = url;
+                a.download = 'image_metadata.json';
+                a.click();
                 
-            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, ""
+                document.getElementById('exportResult').style.display = 'block';
+                document.getElementById('exportResult').textContent = '导出成功！';
+            } catch (error) {
+                alert('导出失败: ' + error);
+            }
+        }
+    </script>
+</body>
+</html>
+"""
 
-def get_image_list_html(password):
-    """获取图片列表（HTML格式）"""
-    if not check_password(password):
-        return "<p style='color: red;'>密码错误</p>"
-    
+@app.route('/')
+def index():
+    return render_template_string(HTML_TEMPLATE)
+
+@app.route('/api/upload', methods=['POST'])
+def upload():
     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()
+        password = request.form.get('password')
+        if not check_password(password):
+            return jsonify({'success': False, 'error': '密码错误'})
         
-        if not rows:
-            return "<p>暂无图片</p>"
+        files = request.files.getlist('images')
+        description = request.form.get('description', '')
         
-        # 生成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>
-        """
+        if not files:
+            return jsonify({'success': False, 'error': '没有文件'})
         
-        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'])
+        results = []
+        for file in files:
+            if file.filename == '':
+                continue
             
-            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>
-        """
+            # 生成hash和文件名
+            image_hash = generate_image_hash()
+            original_filename = file.filename
+            new_filename = generate_filename(original_filename)
+            file_path = IMAGE_DIR / new_filename
+            
+            # 保存文件
+            file.save(file_path)
+            
+            # 获取信息
+            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)
+            )
+            conn.commit()
+            conn.close()
+            
+            results.append({
+                'hash': image_hash,
+                'url': generate_full_url(image_hash),
+                'filename': original_filename
+            })
         
-        return html
-    
+        return jsonify({'success': True, 'data': results})
     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, ""
-    
+        return jsonify({'success': False, 'error': str(e)})
+
+@app.route('/img/<image_hash>')
+def serve_image(image_hash):
     try:
         conn = get_db_connection()
         cursor = conn.cursor()
-        cursor.execute(
-            "SELECT * FROM images WHERE hash = ?",
-            (image_hash,)
-        )
+        cursor.execute("SELECT file_path, mime_type, original_filename FROM images WHERE hash = ?", (image_hash,))
         row = cursor.fetchone()
         conn.close()
         
         if not row:
-            return "图片不存在", None, ""
+            return "Image not found", 404
         
         file_path = Path(row['file_path'])
-        
         if not file_path.exists():
-            return "图片文件已损坏或丢失", None, ""
+            return "File not found", 404
         
-        # 生成访问 URL
-        full_url = generate_full_url(row['hash'])
+        return send_file(file_path, mimetype=row['mime_type'])
+    except Exception as e:
+        return str(e), 500
+
+@app.route('/api/images')
+def get_images():
+    try:
+        password = request.args.get('password')
+        if not check_password(password):
+            return jsonify({'success': False, 'error': '密码错误'})
         
-        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}
-```
-"""
+        conn = get_db_connection()
+        cursor = conn.cursor()
+        cursor.execute("SELECT hash, original_filename, file_size, upload_time, description FROM images ORDER BY upload_time DESC")
+        rows = cursor.fetchall()
+        conn.close()
         
-        return info_text, str(file_path), full_url
-    
+        images = []
+        for row in rows:
+            images.append({
+                'hash': row['hash'],
+                'original_filename': row['original_filename'],
+                'file_size': row['file_size'],
+                'upload_time': row['upload_time'],
+                'description': row['description'],
+                'url': generate_full_url(row['hash'])
+            })
+        
+        return jsonify({'success': True, 'data': images})
     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
-    
+        return jsonify({'success': False, 'error': str(e)})
+
+@app.route('/api/delete/<image_hash>', methods=['POST'])
+def delete_image(image_hash):
     try:
+        data = request.get_json()
+        password = data.get('password')
+        if not check_password(password):
+            return jsonify({'success': False, 'error': '密码错误'})
+        
         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()
         
         if not row:
             conn.close()
-            return "图片不存在", None
+            return jsonify({'success': False, 'error': '图片不存在'})
         
         file_path = Path(row['file_path'])
-        original_filename = row['original_filename']
-        
-        # 删除public目录的副本
-        ext = file_path.suffix
-        public_file = PUBLIC_DIR / f"{image_hash}{ext}"
-        if public_file.exists():
-            public_file.unlink()
-        
-        # 删除数据库记录
         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)
-    
+        return jsonify({'success': True})
     except Exception as e:
-        return f"删除失败: {str(e)}", None
+        return jsonify({'success': False, 'error': str(e)})
 
-def export_data(password):
-    """导出数据（仅元数据）"""
-    if not check_password(password):
-        return "密码错误", None
-    
+@app.route('/api/export')
+def export_data():
     try:
+        password = request.args.get('password')
+        if not check_password(password):
+            return jsonify({'success': False, 'error': '密码错误'})
+        
         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'],
@@ -404,171 +475,18 @@ def export_data(password):
                 'file_size': row['file_size'],
                 'upload_time': row['upload_time'],
                 'description': row['description'],
-                'url': full_url
+                'url': generate_full_url(row['hash'])
             })
         
-        # 保存为文件
-        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
-    
+        import json
+        from flask import Response
+        return Response(
+            json.dumps(data, ensure_ascii=False, indent=2),
+            mimetype='application/json',
+            headers={'Content-Disposition': 'attachment; filename=image_metadata.json'}
+        )
     except Exception as e:
-        return f"导出失败: {str(e)}", None
+        return jsonify({'success': False, 'error': str(e)})
 
-# 初始化数据库
-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]
-                )
-
-if __name__ == "__main__":
-    # 直接启动Gradio，它会使用自己的FastAPI实例
-    gradio_app.launch(server_name="0.0.0.0", server_port=7860)
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860, debug=False)

requirements.txt

@@ -1,3 +1,4 @@
-gradio==4.44.0
+Flask==3.0.0
 Pillow==10.4.0
+Werkzeug==3.0.1