open-webui/backend/apps/web/routers/utils.py

110 lines
3.1 KiB
Python
Raw Normal View History

2024-04-06 10:54:59 +02:00
from fastapi import APIRouter, UploadFile, File, Response
2023-12-24 00:38:52 +01:00
from fastapi import Depends, HTTPException, status
from peewee import SqliteDatabase
2024-03-02 09:33:20 +01:00
from starlette.responses import StreamingResponse, FileResponse
2023-12-24 00:38:52 +01:00
from pydantic import BaseModel
2024-04-04 05:35:32 +02:00
2024-04-06 10:54:59 +02:00
from fpdf import FPDF
2024-04-04 05:35:32 +02:00
import markdown
2023-12-24 08:40:14 +01:00
from apps.web.internal.db import DB
2024-03-02 09:33:20 +01:00
from utils.utils import get_admin_user
from utils.misc import calculate_sha256, get_gravatar_url
2023-12-24 08:40:14 +01:00
2024-04-25 03:24:17 +02:00
from config import OLLAMA_BASE_URLS, DATA_DIR, UPLOAD_DIR, ENABLE_ADMIN_EXPORT
2024-01-09 22:25:42 +01:00
from constants import ERROR_MESSAGES
2024-04-06 10:54:59 +02:00
from typing import List
2023-12-24 00:38:52 +01:00
router = APIRouter()
@router.get("/gravatar")
async def get_gravatar(
email: str,
):
return get_gravatar_url(email)
2024-03-02 09:33:20 +01:00
2024-04-04 05:35:32 +02:00
class MarkdownForm(BaseModel):
md: str
@router.post("/markdown")
async def get_html_from_markdown(
form_data: MarkdownForm,
):
return {"html": markdown.markdown(form_data.md)}
2024-04-06 10:54:59 +02:00
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"]
2024-04-06 11:04:28 +02:00
pdf.set_font("NotoSans", "B", size=14) # Bold for the role
2024-04-06 10:54:59 +02:00
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
2024-04-06 11:04:28 +02:00
pdf.multi_cell(effective_page_width, 6, content, 0, "L")
pdf.ln(1.5) # Extra space between messages
2024-04-06 10:54:59 +02:00
# 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"},
)
2024-03-02 09:33:20 +01:00
@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,
)
2024-03-02 09:33:20 +01:00
return FileResponse(
DB.database,
2024-03-02 09:33:20 +01:00
media_type="application/octet-stream",
filename="webui.db",
)