forked from open-webui/open-webui
109 lines
3.1 KiB
Python
109 lines
3.1 KiB
Python
from fastapi import APIRouter, UploadFile, File, Response
|
|
from fastapi import Depends, HTTPException, status
|
|
from peewee import SqliteDatabase
|
|
from starlette.responses import StreamingResponse, FileResponse
|
|
from pydantic import BaseModel
|
|
|
|
|
|
from fpdf import FPDF
|
|
import markdown
|
|
|
|
from apps.web.internal.db import DB
|
|
from utils.utils import get_admin_user
|
|
from utils.misc import calculate_sha256, get_gravatar_url
|
|
|
|
from config import OLLAMA_BASE_URLS, DATA_DIR, UPLOAD_DIR, ENABLE_ADMIN_EXPORT
|
|
from constants import ERROR_MESSAGES
|
|
from typing import List
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/gravatar")
|
|
async def get_gravatar(
|
|
email: str,
|
|
):
|
|
return get_gravatar_url(email)
|
|
|
|
|
|
class MarkdownForm(BaseModel):
|
|
md: str
|
|
|
|
|
|
@router.post("/markdown")
|
|
async def get_html_from_markdown(
|
|
form_data: MarkdownForm,
|
|
):
|
|
return {"html": markdown.markdown(form_data.md)}
|
|
|
|
|
|
class ChatForm(BaseModel):
|
|
title: str
|
|
messages: List[dict]
|
|
|
|
|
|
@router.post("/pdf")
|
|
async def download_chat_as_pdf(
|
|
form_data: ChatForm,
|
|
):
|
|
pdf = FPDF()
|
|
pdf.add_page()
|
|
|
|
STATIC_DIR = "./static"
|
|
FONTS_DIR = f"{STATIC_DIR}/fonts"
|
|
|
|
pdf.add_font("NotoSans", "", f"{FONTS_DIR}/NotoSans-Regular.ttf")
|
|
pdf.add_font("NotoSans", "b", f"{FONTS_DIR}/NotoSans-Bold.ttf")
|
|
pdf.add_font("NotoSans", "i", f"{FONTS_DIR}/NotoSans-Italic.ttf")
|
|
pdf.add_font("NotoSansKR", "", f"{FONTS_DIR}/NotoSansKR-Regular.ttf")
|
|
pdf.add_font("NotoSansJP", "", f"{FONTS_DIR}/NotoSansJP-Regular.ttf")
|
|
|
|
pdf.set_font("NotoSans", size=12)
|
|
pdf.set_fallback_fonts(["NotoSansKR", "NotoSansJP"])
|
|
|
|
pdf.set_auto_page_break(auto=True, margin=15)
|
|
|
|
# Adjust the effective page width for multi_cell
|
|
effective_page_width = (
|
|
pdf.w - 2 * pdf.l_margin - 10
|
|
) # Subtracted an additional 10 for extra padding
|
|
|
|
# Add chat messages
|
|
for message in form_data.messages:
|
|
role = message["role"]
|
|
content = message["content"]
|
|
pdf.set_font("NotoSans", "B", size=14) # Bold for the role
|
|
pdf.multi_cell(effective_page_width, 10, f"{role.upper()}", 0, "L")
|
|
pdf.ln(1) # Extra space between messages
|
|
|
|
pdf.set_font("NotoSans", size=10) # Regular for content
|
|
pdf.multi_cell(effective_page_width, 6, content, 0, "L")
|
|
pdf.ln(1.5) # Extra space between messages
|
|
|
|
# Save the pdf with name .pdf
|
|
pdf_bytes = pdf.output()
|
|
|
|
return Response(
|
|
content=bytes(pdf_bytes),
|
|
media_type="application/pdf",
|
|
headers={"Content-Disposition": f"attachment;filename=chat.pdf"},
|
|
)
|
|
|
|
|
|
@router.get("/db/download")
|
|
async def download_db(user=Depends(get_admin_user)):
|
|
if not ENABLE_ADMIN_EXPORT:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
|
|
)
|
|
if not isinstance(DB, SqliteDatabase):
|
|
raise HTTPException(
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
detail=ERROR_MESSAGES.DB_NOT_SQLITE,
|
|
)
|
|
return FileResponse(
|
|
DB.database,
|
|
media_type="application/octet-stream",
|
|
filename="webui.db",
|
|
)
|