Upload 2 files

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 import os
 import sqlite3
 import hashlib
-import shutil
 from datetime import datetime
 from pathlib import Path
 from PIL import Image
@@ -10,6 +9,8 @@ import json
 
 # 环境变量
 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")
@@ -58,90 +59,125 @@ def check_password(password):
 
 def generate_filename(original_filename):
     """生成唯一的文件名"""
-    timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
+    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
+    ext = Path(original_filename).suffix.lower()
+    if not ext:
+        ext = ".png"
     return f"{timestamp}_{hash_suffix}{ext}"
 
-def upload_image(image, description, password):
-    """上传图片"""
+def generate_full_url(file_path):
+    """生成完整的图片URL"""
+    relative_path = f"/file={file_path}"
+    if HF_USERNAME and HF_SPACE_NAME:
+        return f"https://{HF_USERNAME}-{HF_SPACE_NAME}.hf.space{relative_path}"
+    else:
+        return relative_path
+
+def upload_images(images, description, password):
+    """批量上传图片"""
     if not check_password(password):
-        return "❌ 密码错误！", None, None
+        return "❌ 密码错误！", None, ""
+    
+    if not images or len(images) == 0:
+        return "❌ 请选择要上传的图片！", None, ""
     
-    if image is None:
-        return "❌ 请选择要上传的图片！", None, None
+    results = []
+    success_count = 0
+    fail_count = 0
     
     try:
-        # 获取原始文件名
-        if isinstance(image, str):
-            # 如果是文件路径
-            original_filename = Path(image).name
-            img = Image.open(image)
-        else:
-            # 如果是 PIL Image
-            original_filename = "uploaded_image.png"
-            img = image
-        
-        # 生成新文件名
-        new_filename = generate_filename(original_filename)
-        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 (filename, original_filename, file_path, file_size, mime_type, upload_time, description)
-            VALUES (?, ?, ?, ?, ?, ?, ?)
-            """,
-            (new_filename, original_filename, str(file_path), file_size, mime_type, upload_time, description or "")
-        )
-        image_id = cursor.lastrowid
-        conn.commit()
-        conn.close()
-        
-        # 生成访问 URL
-        url = f"/file={file_path}"
-        
-        result_text = f"""✅ 上传成功！
-
-**图片ID**: {image_id}
-**文件名**: {original_filename}
-**大小**: {file_size / 1024:.2f} KB
-**上传时间**: {upload_time}
-
-**访问URL**: 
-```
-{url}
-```
-
-💡 提示：可以在"查看图片"标签页中输入 ID: {image_id} 来查看图片
-"""
-        
-        return result_text, get_image_list(password), url
+        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
+                
+                # 生成新文件名
+                new_filename = generate_filename(original_filename)
+                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 (filename, original_filename, file_path, file_size, mime_type, upload_time, description)
+                    VALUES (?, ?, ?, ?, ?, ?, ?)
+                    """,
+                    (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(file_path)
+                
+                results.append({
+                    'id': image_id,
+                    '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"- **ID**: {r['id']}\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, None
+        return f"❌ 上传失败: {str(e)}", None, ""
 
-def get_image_list(password):
-    """获取图片列表"""
+def get_image_list_html(password):
+    """获取图片列表（HTML格式）"""
     if not check_password(password):
-        return "❌ 密码错误！"
+        return "<p style='color: red;'>❌ 密码错误！</p>"
     
     try:
         conn = get_db_connection()
         cursor = conn.cursor()
         cursor.execute(
             """
-            SELECT id, filename, original_filename, file_size, upload_time, description
+            SELECT id, filename, original_filename, file_path, file_size, upload_time, description
             FROM images
             ORDER BY upload_time DESC
             """
@@ -150,31 +186,69 @@ def get_image_list(password):
         conn.close()
         
         if not rows:
-            return "暂无图片"
+            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-view { background: #4CAF50; color: white; padding: 6px 12px; border: none; border-radius: 4px; cursor: pointer; margin-right: 5px; }
+            .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-id { font-weight: bold; color: #2196F3; }
+        </style>
+        <table class="image-table">
+            <thead>
+                <tr>
+                    <th>ID</th>
+                    <th>文件名</th>
+                    <th>大小</th>
+                    <th>上传时间</th>
+                    <th>描述</th>
+                    <th>操作</th>
+                </tr>
+            </thead>
+            <tbody>
+        """
         
-        # 格式化输出
-        output = "## 📸 图片列表\n\n"
         for row in rows:
-            output += f"**ID: {row['id']}** | {row['original_filename']}\n"
-            output += f"- 存储文件名: {row['filename']}\n"
-            output += f"- 大小: {row['file_size'] / 1024:.2f} KB\n"
-            output += f"- 上传时间: {row['upload_time']}\n"
-            if row['description']:
-                output += f"- 描述: {row['description']}\n"
-            output += "\n---\n\n"
-        
-        return output
+            upload_time_str = datetime.fromisoformat(row['upload_time']).strftime('%Y-%m-%d %H:%M:%S')
+            full_url = generate_full_url(row['file_path'])
+            
+            html += f"""
+                <tr>
+                    <td class="image-id">{row['id']}</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-view" onclick="window.viewImage_{row['id']}()">查看</button>
+                        <button class="btn-copy" onclick="navigator.clipboard.writeText('{full_url}').then(() => alert('URL已复制到剪贴板！'))">复制URL</button>
+                    </td>
+                </tr>
+            """
+        
+        html += """
+            </tbody>
+        </table>
+        """
+        
+        return html
     
     except Exception as e:
-        return f"❌ 获取列表失败: {str(e)}"
+        return f"<p style='color: red;'>❌ 获取列表失败: {str(e)}</p>"
 
-def view_image(image_id, password):
-    """查看图片"""
+def view_image_by_id(image_id, password):
+    """通过ID查看图片"""
     if not check_password(password):
-        return "❌ 密码错误！", None, None
+        return "❌ 密码错误！", None, ""
     
     if not image_id:
-        return "❌ 请输入图片ID！", None, None
+        return "❌ 请输入图片ID！", None, ""
     
     try:
         conn = get_db_connection()
@@ -187,17 +261,17 @@ def view_image(image_id, password):
         conn.close()
         
         if not row:
-            return "❌ 图片不存在！", None, None
+            return "❌ 图片不存在！", None, ""
         
         file_path = Path(row['file_path'])
         
         if not file_path.exists():
-            return "❌ 图片文件已损坏或丢失！", None, None
+            return "❌ 图片文件已损坏或丢失！", None, ""
         
         # 生成访问 URL
-        url = f"/file={file_path}"
+        full_url = generate_full_url(file_path)
         
-        info_text = f"""✅ 图片信息
+        info_text = f"""## 📷 图片信息
 
 **ID**: {row['id']}
 **原始文件名**: {row['original_filename']}
@@ -208,14 +282,14 @@ def view_image(image_id, password):
 
 **访问URL**:
 ```
-{url}
+{full_url}
 ```
 """
         
-        return info_text, str(file_path), url
+        return info_text, str(file_path), full_url
     
     except Exception as e:
-        return f"❌ 查看失败: {str(e)}", None, None
+        return f"❌ 查看失败: {str(e)}", None, ""
 
 def delete_image(image_id, password):
     """删除图片"""
@@ -249,7 +323,7 @@ def delete_image(image_id, password):
         if file_path.exists():
             file_path.unlink()
         
-        return f"✅ 已删除图片: {original_filename} (ID: {image_id})", get_image_list(password)
+        return f"✅ 已删除图片: {original_filename} (ID: {image_id})", get_image_list_html(password)
     
     except Exception as e:
         return f"❌ 删除失败: {str(e)}", None
@@ -277,12 +351,22 @@ def get_stats(password):
         
         conn.close()
         
+        # 检查配置状态
+        config_status = "✅ 已配置" if (HF_USERNAME and HF_SPACE_NAME) else "⚠️ 未配置（仅显示相对路径）"
+        
         stats = f"""## 📊 统计信息
 
 **总图片数**: {total_count} 张
 **总存储大小**: {total_size / 1024 / 1024:.2f} MB
 **最近上传**: {last_upload['upload_time'] if last_upload else '暂无'}
-**存储路径**: {IMAGE_DIR.absolute()}
+
+### 🔧 配置信息
+**HF用户名**: {HF_USERNAME or '未设置'}
+**Space名称**: {HF_SPACE_NAME or '未设置'}
+**URL生成**: {config_status}
+
+### 📁 存储信息
+**图片目录**: {IMAGE_DIR.absolute()}
 **数据库**: {Path(DB_PATH).absolute()}
 """
         
@@ -306,13 +390,15 @@ def export_data(password):
         # 转换为 JSON
         data = []
         for row in rows:
+            full_url = generate_full_url(row['file_path'])
             data.append({
                 'id': row['id'],
                 'filename': row['filename'],
                 'original_filename': row['original_filename'],
                 'file_size': row['file_size'],
                 'upload_time': row['upload_time'],
-                'description': row['description']
+                'description': row['description'],
+                'url': full_url
             })
         
         # 保存为文件
@@ -328,10 +414,21 @@ def export_data(password):
 # 初始化数据库
 init_success = init_db()
 
+# 自定义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="私人图床服务", theme=gr.themes.Soft()) as app:
+with gr.Blocks(title="私人图床服务", theme=gr.themes.Soft(), css=custom_css) as app:
     gr.Markdown("# 🖼️ 私人图床服务")
-    gr.Markdown("⚠️ 请输入密码���使用此服务 | 图片存储在服务器本地文件系统")
+    gr.Markdown("⚠️ 请输入密码以使用此服务 | 支持批量上传 | 图片存储在服务器本地")
     
     if not init_success:
         gr.Markdown("## ❌ 数据库初始化失败！")
@@ -339,66 +436,137 @@ with gr.Blocks(title="私人图床服务", theme=gr.themes.Soft()) as app:
         with gr.Tabs():
             # 上传图片标签页
             with gr.Tab("📤 上传图片"):
+                gr.Markdown("### 批量上传图片（支持多张）")
+                
                 with gr.Row():
-                    with gr.Column():
-                        upload_image_input = gr.Image(label="选择图片", type="pil")
-                        upload_desc_input = gr.Textbox(label="描述（可选）", placeholder="给图片添加描述...")
-                        upload_password_input = gr.Textbox(label="密码", type="password", placeholder="输入访问密码")
-                        upload_btn = gr.Button("上传", variant="primary")
+                    with gr.Column(scale=1):
+                        upload_password_input = gr.Textbox(
+                            label="🔐 密码", 
+                            type="password", 
+                            placeholder="输入访问密码"
+                        )
+                        upload_images_input = gr.File(
+                            label="选择图片（支持多选）",
+                            file_count="multiple",
+                            file_types=["image"]
+                        )
+                        upload_desc_input = gr.Textbox(
+                            label="描述（可选）", 
+                            placeholder="为这批图片添加描述...",
+                            lines=2
+                        )
+                        
+                        with gr.Row():
+                            upload_btn = gr.Button("📤 上传", variant="primary", scale=2)
+                            toggle_preview_btn = gr.Button("👁️ 显示/隐藏预览", scale=1)
                     
-                    with gr.Column():
-                        upload_output = gr.Textbox(label="上传结果", lines=10)
-                        upload_url_output = gr.Textbox(label="图片URL", lines=2)
-                        upload_list_output = gr.Markdown(label="图片列表")
+                    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) as preview_accordion:
+                    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(
-                    upload_image,
-                    inputs=[upload_image_input, upload_desc_input, upload_password_input],
-                    outputs=[upload_output, upload_list_output, upload_url_output]
+                    lambda files, desc, pwd: upload_images([f.name for f in files] if files else [], desc, pwd),
+                    inputs=[upload_images_input, upload_desc_input, upload_password_input],
+                    outputs=[upload_output, gr.HTML(visible=False), upload_url_output]
                 )
             
-            # 查看图片标签页
-            with gr.Tab("👀 查看图片"):
+            # 图片列表标签页
+            with gr.Tab("📋 图片列表"):
+                gr.Markdown("### 查看和管理所有图片")
+                
+                with gr.Row():
+                    list_password_input = gr.Textbox(
+                        label="🔐 密码", 
+                        type="password", 
+                        placeholder="输入访问密码",
+                        scale=3
+                    )
+                    list_refresh_btn = gr.Button("🔄 刷新列表", variant="primary", scale=1)
+                
+                list_output = gr.HTML(label="图片列表")
+                
+                gr.Markdown("---")
+                gr.Markdown("### 📷 查看图片详情")
+                
                 with gr.Row():
                     with gr.Column():
-                        view_id_input = gr.Textbox(label="图片ID", placeholder="输入图片ID")
-                        view_password_input = gr.Textbox(label="密码", type="password", placeholder="输入访问密码")
-                        view_btn = gr.Button("查看", variant="primary")
-                        view_output = gr.Textbox(label="图片信息", lines=10)
-                        view_url_output = gr.Textbox(label="图片URL", lines=2)
+                        view_id_input = gr.Textbox(
+                            label="图片ID", 
+                            placeholder="输入图片ID查看详情"
+                        )
+                        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="图片预览")
+                        view_image_output = gr.Image(label="图片预览", height=400)
                 
-                view_btn.click(
-                    view_image,
-                    inputs=[view_id_input, view_password_input],
-                    outputs=[view_output, view_image_output, view_url_output]
-                )
-            
-            # 图片列表标签页
-            with gr.Tab("📋 图片列表"):
-                list_password_input = gr.Textbox(label="密码", type="password", placeholder="输入访问密码")
-                list_btn = gr.Button("刷新列表", variant="primary")
-                list_output = gr.Markdown(label="图片列表")
-                
-                list_btn.click(
-                    get_image_list,
+                # 绑定事件
+                list_refresh_btn.click(
+                    get_image_list_html,
                     inputs=[list_password_input],
                     outputs=[list_output]
                 )
+                
+                view_btn.click(
+                    view_image_by_id,
+                    inputs=[view_id_input, list_password_input],
+                    outputs=[view_info_output, view_image_output, view_url_output]
+                )
             
             # 删除图片标签页
             with gr.Tab("🗑️ 删除图片"):
+                gr.Markdown("### ⚠️ 删除操作不可恢复，请谨慎操作")
+                
                 with gr.Row():
                     with gr.Column():
-                        delete_id_input = gr.Textbox(label="图片ID", placeholder="输入要删除的图片ID")
-                        delete_password_input = gr.Textbox(label="密码", type="password", placeholder="输入访问密码")
-                        delete_btn = gr.Button("删除", variant="stop")
+                        delete_id_input = gr.Textbox(
+                            label="图片ID", 
+                            placeholder="输入要删除的图片ID"
+                        )
+                        delete_password_input = gr.Textbox(
+                            label="🔐 密码", 
+                            type="password", 
+                            placeholder="输入访问密码"
+                        )
+                        delete_btn = gr.Button("🗑️ 确认删除", variant="stop")
+                        delete_output = gr.Textbox(label="删除结果", lines=3)
                     
                     with gr.Column():
-                        delete_output = gr.Textbox(label="删除结果", lines=3)
-                        delete_list_output = gr.Markdown(label="剩余图片列表")
+                        gr.Markdown("### 💡 提示")
+                        gr.Markdown("""
+                        - 删除操作会同时删除数据库记录和文件
+                        - 删除后无法恢复
+                        - 可以在图片列表中查看要删除的图片ID
+                        """)
+                
+                delete_list_output = gr.HTML(label="剩余图片列表")
                 
                 delete_btn.click(
                     delete_image,
@@ -408,13 +576,19 @@ with gr.Blocks(title="私人图床服务", theme=gr.themes.Soft()) as app:
             
             # 统计信息标签页
             with gr.Tab("📊 统计信息"):
-                stats_password_input = gr.Textbox(label="密码", type="password", placeholder="输入访问密码")
-                stats_btn = gr.Button("查看统计", variant="primary")
+                stats_password_input = gr.Textbox(
+                    label="🔐 密码", 
+                    type="password", 
+                    placeholder="输入访问密码"
+                )
+                stats_btn = gr.Button("📊 查看统计", variant="primary")
                 stats_output = gr.Markdown(label="统计信息")
                 
                 gr.Markdown("---")
-                gr.Markdown("### 导出数据")
-                export_btn = gr.Button("导出元数据（JSON）", variant="secondary")
+                gr.Markdown("### 📦 导出数据")
+                gr.Markdown("导出所有图片的元数据（包含完整URL）为JSON格式")
+                
+                export_btn = gr.Button("📦 导出元数据（JSON）", variant="secondary")
                 export_output = gr.Textbox(label="导出结果", lines=2)
                 export_file_output = gr.File(label="下载文件")
                 
@@ -431,8 +605,18 @@ with gr.Blocks(title="私人图床服务", theme=gr.themes.Soft()) as app:
                 )
     
     gr.Markdown("---")
-    gr.Markdown("💡 提示：图片存储在服务器本地，URL可以直接在外部访问（需要 Space 为 Public）")
-    gr.Markdown("🔒 安全提示：建议将 Space 设为 Private 以保护隐私")
+    gr.Markdown("""
+    ### 💡 使用提示
+    - **完整URL**: 在环境变量中配置 `HF_USERNAME` 和 `HF_SPACE_NAME` 即可生成完整的外部访问URL
+    - **批量上传**: 支持一次选择多张图片上传
+    - **快速复制**: 在图片列表中点击"复制URL"按钮即可复制到剪贴板
+    - **数据安全**: 定期导出元数据进行备份
+    
+    ### 🔒 安全提示
+    - 建议将 Space 设为 Private 以保护隐私
+    - 定期更换访问密码
+    - 不要分享完整URL给不信任的人
+    """)
 
 if __name__ == "__main__":
     app.launch()