Spaces:

Moonfanz
/

PDF_MANAGER

Paused

App Files Files Community

Moonfanz commited on Dec 15, 2024

Commit

ca2bf17

verified ·

1 Parent(s): b275b3a

Upload 5 files

Browse files

Files changed (5) hide show

Dockerfile +31 -0
app.py +81 -0
requirements.txt +3 -0
templates/index.html +46 -0
templates/view.html +118 -0

Dockerfile ADDED Viewed

	@@ -0,0 +1,31 @@

+# 使用官方 Python 镜像作为基础镜像
+FROM python:3.9-slim-buster
+# 设置工作目录
+WORKDIR /app
+# 复制 requirements.txt 到容器中
+COPY requirements.txt /app/
+# 安装依赖
+RUN pip install --no-cache-dir -r requirements.txt
+# 安装必要的系统包
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    gcc \
+    libc-dev \
+    libmupdf-dev \
+    libmupdf-tools \
+    && rm -rf /var/lib/apt/lists/*
+# 复制当前目录的所有文件到容器中
+COPY . /app/
+# 暴露端口
+EXPOSE 7860
+# 定义环境变量
+ENV NAME pdf_manager
+# 运行应用程序
+CMD ["python", "app.py"]

app.py ADDED Viewed

	@@ -0,0 +1,81 @@

+import os
+import shutil
+from flask import Flask, render_template, request, send_from_directory, redirect, url_for
+from werkzeug.utils import secure_filename
+import fitz  # PyMuPDF
+app = Flask(__name__)
+# 配置
+app.config['UPLOAD_FOLDER'] = '/temp/uploads'
+app.config['ALLOWED_EXTENSIONS'] = {'pdf'}
+app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024  # 16MB 上传限制
+app.config['PORT'] = 7860
+# 确保上传目录存在
+os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
+def allowed_file(filename):
+    return '.' in filename and \
+           filename.rsplit('.', 1)[1].lower() in app.config['ALLOWED_EXTENSIONS']
+@app.route('/')
+def index():
+    pdf_files = [f for f in os.listdir(app.config['UPLOAD_FOLDER']) if f.endswith('.pdf')]
+    return render_template('index.html', pdf_files=pdf_files)
+@app.route('/upload', methods=['POST'])
+def upload_file():
+    if 'file' not in request.files:
+        return redirect(request.url)
+    file = request.files['file']
+    if file.filename == '':
+        return redirect(request.url)
+    if file and allowed_file(file.filename):
+        filename = secure_filename(file.filename)
+        file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
+        return redirect(url_for('index'))
+    return redirect(request.url)
+@app.route('/view/<filename>')
+def view_pdf(filename):
+    return render_template('view.html', filename=filename)
+@app.route('/get_pdf/<filename>')
+def get_pdf(filename):
+    return send_from_directory(app.config['UPLOAD_FOLDER'], filename)
+@app.route('/delete/<filename>')
+def delete_pdf(filename):
+    file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
+    if os.path.exists(file_path):
+        os.remove(file_path)
+    return redirect(url_for('index'))
+@app.route('/delete_all')
+def delete_all_pdfs():
+    folder = app.config['UPLOAD_FOLDER']
+    for filename in os.listdir(folder):
+        file_path = os.path.join(folder, filename)
+        try:
+            if os.path.isfile(file_path) or os.path.islink(file_path):
+                os.unlink(file_path)
+            elif os.path.isdir(file_path):
+                shutil.rmtree(file_path)
+        except Exception as e:
+            print(f'Failed to delete {file_path}. Reason: {e}')
+    return redirect(url_for('index'))
+@app.route('/get_page/<filename>/<int:page_num>')
+def get_page(filename, page_num):
+    try:
+        doc = fitz.open(os.path.join(app.config['UPLOAD_FOLDER'], filename))
+        page = doc.load_page(page_num - 1)  # page numbers are zero-based
+        pix = page.get_pixmap()
+        img_data = pix.tobytes("png")
+        return img_data, 200, {'Content-Type': 'image/png'}
+    except Exception as e:
+        return str(e), 500
+if __name__ == '__main__':
+    app.run(debug=True, host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))

requirements.txt ADDED Viewed

	@@ -0,0 +1,3 @@

+Flask==2.2.2
+Werkzeug==2.2.2
+PyMuPDF==1.21.0

templates/index.html ADDED Viewed

	@@ -0,0 +1,46 @@

+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <title>PDF Manager</title>
+    <style>
+        body {
+            font-family: sans-serif;
+        }
+        .pdf-item {
+            margin-bottom: 10px;
+        }
+        .pdf-item a {
+            margin-right: 10px;
+        }
+    </style>
+</head>
+<body>
+    <h1>PDF Manager</h1>
+    <h2>Upload PDF</h2>
+    <form method="POST" action="/upload" enctype="multipart/form-data">
+        <input type="file" name="file">
+        <button type="submit">Upload</button>
+    </form>
+    <h2>Uploaded PDFs</h2>
+    {% if pdf_files %}
+    {% for filename in pdf_files %}
+    <div class="pdf-item">
+        <a href="{{ url_for('view_pdf', filename=filename) }}">{{ filename }}</a>
+        <a href="{{ url_for('delete_pdf', filename=filename) }}" onclick="return confirm('Are you sure?')">Delete</a>
+    </div>
+    {% endfor %}
+    {% else %}
+    <p>No PDFs uploaded yet.</p>
+    {% endif %}
+    <a href="{{ url_for('delete_all_pdfs') }}"
+        onclick="return confirm('Are you sure you want to delete all PDFs?')">Delete All</a>
+</body>
+</html>

templates/view.html ADDED Viewed

	@@ -0,0 +1,118 @@

+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <title>View PDF - {{ filename }}</title>
+    <style>
+        body {
+            font-family: sans-serif;
+            text-align: center;
+        }
+        #pdf-container {
+            width: 100%;
+            height: 800px;
+            overflow: auto;
+        }
+        .page-img {
+            width: 100%;
+        }
+        .nav-btn {
+            padding: 10px;
+            margin: 5px;
+            cursor: pointer;
+            background-color: #4CAF50;
+            color: white;
+            border: none;
+        }
+    </style>
+</head>
+<body>
+    <h1>{{ filename }}</h1>
+    <button class="nav-btn" onclick="prevPage()">Previous</button>
+    <span id="page-num">1</span> / <span id="page-count"></span>
+    <button class="nav-btn" onclick="nextPage()">Next</button>
+    <div id="pdf-container">
+        <!-- PDF pages will be loaded here -->
+    </div>
+    <script>
+        const pdfFilename = "{{ filename }}";
+        let currentPage = 1;
+        let pageCount = 0;
+        function loadPage(pageNum) {
+            fetch(`/get_page/${pdfFilename}/${pageNum}`)
+                .then(response => {
+                    if (!response.ok) {
+                        throw new Error('Network response was not ok');
+                    }
+                    return response.blob();
+                })
+                .then(blob => {
+                    const imageUrl = URL.createObjectURL(blob);
+                    const img = document.createElement('img');
+                    img.src = imageUrl;
+                    img.className = 'page-img';
+                    const pdfContainer = document.getElementById('pdf-container');
+                    pdfContainer.innerHTML = ''; // 清空之前的页面
+                    pdfContainer.appendChild(img);
+                    document.getElementById('page-num').textContent = pageNum;
+                })
+                .catch(error => {
+                    console.error('Error:', error);
+                });
+        }
+        function prevPage() {
+            if (currentPage > 1) {
+                currentPage--;
+                loadPage(currentPage);
+            }
+        }
+        function nextPage() {
+            if (currentPage < pageCount) {
+                currentPage++;
+                loadPage(currentPage);
+            }
+        }
+        // 初始化时获取总页数
+        fetch(`/get_page/${pdfFilename}/1`)
+            .then(response => {
+                if (!response.ok) {
+                    throw new Error('Network response was not ok');
+                }
+                // 假设第一页加载成功，通过 PyMuPDF 获取总页数
+                return fetch(`/get_page/${pdfFilename}/1`);
+            })
+            .then(() => {
+                // 这里没法直接获取总页数, 需要通过循环尝试, 直到一个错误返回
+                // 此方法效率较低，仅为演示用
+                function fetchPageCount(page = 1) {
+                    fetch(`/get_page/${pdfFilename}/${page}`)
+                        .then(response => {
+                            if (!response.ok) {
+                                pageCount = page - 1;
+                                document.getElementById('page-count').textContent = pageCount;
+                                loadPage(currentPage);
+                            } else {
+                                fetchPageCount(page + 1);
+                            }
+                        })
+                }
+                fetchPageCount();
+            })
+            .catch(error => {
+                console.error('Error:', error);
+            });
+    </script>
+</body>
+</html>