Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -44,29 +44,16 @@ FONT_DIR.mkdir(exist_ok=True)
|
|
| 44 |
# --- Font & Emoji Handling ---
|
| 45 |
|
| 46 |
def download_and_register_fonts():
|
| 47 |
-
"""
|
| 48 |
print("--- Font Registration Process Starting ---")
|
| 49 |
-
fonts_to_download = {
|
| 50 |
-
"DejaVuSans.ttf": "https://github.com/dejavu-fonts/dejavu-fonts/blob/main/ttf/DejaVuSans.ttf?raw=true",
|
| 51 |
-
"NotoColorEmoji.ttf": "https://github.com/googlefonts/noto-emoji/blob/main/fonts/NotoColorEmoji.ttf?raw=true"
|
| 52 |
-
}
|
| 53 |
-
for filename, url in fonts_to_download.items():
|
| 54 |
-
font_path = FONT_DIR / filename
|
| 55 |
-
if not font_path.exists():
|
| 56 |
-
print(f"Downloading {filename}...")
|
| 57 |
-
try:
|
| 58 |
-
r = requests.get(url, allow_redirects=True, timeout=20)
|
| 59 |
-
r.raise_for_status()
|
| 60 |
-
with open(font_path, "wb") as f: f.write(r.content)
|
| 61 |
-
print(f"{filename} downloaded.")
|
| 62 |
-
except Exception as e:
|
| 63 |
-
print(f"Failed to download {filename}: {e}")
|
| 64 |
-
|
| 65 |
font_names = []
|
| 66 |
print(f"Scanning for fonts in: {FONT_DIR.absolute()}")
|
| 67 |
font_files = list(FONT_DIR.glob("*.ttf"))
|
| 68 |
print(f"Found {len(font_files)} .ttf files: {[f.name for f in font_files]}")
|
| 69 |
|
|
|
|
|
|
|
|
|
|
| 70 |
for font_path in font_files:
|
| 71 |
try:
|
| 72 |
font_name = font_path.stem
|
|
@@ -79,7 +66,9 @@ def download_and_register_fonts():
|
|
| 79 |
pdfmetrics.registerFont(TTFont(f"{font_name}-BoldItalic", str(font_path)))
|
| 80 |
pdfmetrics.registerFontFamily(font_name, normal=font_name, bold=f"{font_name}-Bold", italic=f"{font_name}-Italic", boldItalic=f"{font_name}-BoldItalic")
|
| 81 |
|
| 82 |
-
|
|
|
|
|
|
|
| 83 |
font_names.append(font_name)
|
| 84 |
except Exception as e:
|
| 85 |
print(f"Could not register font {font_path.name}: {e}")
|
|
@@ -89,12 +78,15 @@ def download_and_register_fonts():
|
|
| 89 |
|
| 90 |
def apply_emoji_font(text: str) -> str:
|
| 91 |
"""Wraps emoji characters in a <font> tag to use the dedicated emoji font."""
|
|
|
|
|
|
|
|
|
|
| 92 |
emoji_pattern = re.compile(f"([{re.escape(''.join(map(chr, range(0x1f600, 0x1f650))))}"
|
| 93 |
f"{re.escape(''.join(map(chr, range(0x1f300, 0x1f5ff))))}"
|
| 94 |
f"{re.escape(''.join(map(chr, range(0x1f900, 0x1f9ff))))}"
|
| 95 |
f"{re.escape(''.join(map(chr, range(0x2600, 0x26ff))))}"
|
| 96 |
f"{re.escape(''.join(map(chr, range(0x2700, 0x27bf))))}]+)")
|
| 97 |
-
return emoji_pattern.sub(fr'<font name="{
|
| 98 |
|
| 99 |
|
| 100 |
# --- PDF Generation & Handling ---
|
|
@@ -226,10 +218,23 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Advanced PDF Generator") as demo:
|
|
| 226 |
"""Function to run on app load to generate a sample PDF."""
|
| 227 |
print("Running initial demo generation...")
|
| 228 |
# Create a dummy file object for Gradio
|
| 229 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 230 |
sample_bytes = f.read()
|
| 231 |
|
| 232 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 233 |
previews, logs, files = generate_pdfs_api(demo_files, ["A4 Portrait"], [AVAILABLE_FONTS[0]] if AVAILABLE_FONTS else [], 1)
|
| 234 |
return previews, logs, files
|
| 235 |
|
|
|
|
| 44 |
# --- Font & Emoji Handling ---
|
| 45 |
|
| 46 |
def download_and_register_fonts():
|
| 47 |
+
"""Finds and registers all .ttf files from the local 'fonts' directory."""
|
| 48 |
print("--- Font Registration Process Starting ---")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 49 |
font_names = []
|
| 50 |
print(f"Scanning for fonts in: {FONT_DIR.absolute()}")
|
| 51 |
font_files = list(FONT_DIR.glob("*.ttf"))
|
| 52 |
print(f"Found {len(font_files)} .ttf files: {[f.name for f in font_files]}")
|
| 53 |
|
| 54 |
+
if not font_files:
|
| 55 |
+
print("WARNING: No .ttf files found in the 'fonts' directory. Please add font files to use them in the application.")
|
| 56 |
+
|
| 57 |
for font_path in font_files:
|
| 58 |
try:
|
| 59 |
font_name = font_path.stem
|
|
|
|
| 66 |
pdfmetrics.registerFont(TTFont(f"{font_name}-BoldItalic", str(font_path)))
|
| 67 |
pdfmetrics.registerFontFamily(font_name, normal=font_name, bold=f"{font_name}-Bold", italic=f"{font_name}-Italic", boldItalic=f"{font_name}-BoldItalic")
|
| 68 |
|
| 69 |
+
# Exclude specific emoji fonts from the main selection list as they are handled automatically.
|
| 70 |
+
# We use a more general check to catch variants like NotoEmoji-Bold etc.
|
| 71 |
+
if "notoemoji" not in font_name.lower():
|
| 72 |
font_names.append(font_name)
|
| 73 |
except Exception as e:
|
| 74 |
print(f"Could not register font {font_path.name}: {e}")
|
|
|
|
| 78 |
|
| 79 |
def apply_emoji_font(text: str) -> str:
|
| 80 |
"""Wraps emoji characters in a <font> tag to use the dedicated emoji font."""
|
| 81 |
+
# Assuming 'NotoColorEmoji-Regular' is the intended font for color emojis.
|
| 82 |
+
# The font name here must match the one registered from the filename.
|
| 83 |
+
emoji_font_to_use = "NotoColorEmoji-Regular"
|
| 84 |
emoji_pattern = re.compile(f"([{re.escape(''.join(map(chr, range(0x1f600, 0x1f650))))}"
|
| 85 |
f"{re.escape(''.join(map(chr, range(0x1f300, 0x1f5ff))))}"
|
| 86 |
f"{re.escape(''.join(map(chr, range(0x1f900, 0x1f9ff))))}"
|
| 87 |
f"{re.escape(''.join(map(chr, range(0x2600, 0x26ff))))}"
|
| 88 |
f"{re.escape(''.join(map(chr, range(0x2700, 0x27bf))))}]+)")
|
| 89 |
+
return emoji_pattern.sub(fr'<font name="{emoji_font_to_use}">\1</font>', text)
|
| 90 |
|
| 91 |
|
| 92 |
# --- PDF Generation & Handling ---
|
|
|
|
| 218 |
"""Function to run on app load to generate a sample PDF."""
|
| 219 |
print("Running initial demo generation...")
|
| 220 |
# Create a dummy file object for Gradio
|
| 221 |
+
sample_md_path = CWD / "sample.md"
|
| 222 |
+
if not sample_md_path.exists():
|
| 223 |
+
return [], "Sample.md not found.", []
|
| 224 |
+
|
| 225 |
+
with open(sample_md_path, "rb") as f:
|
| 226 |
sample_bytes = f.read()
|
| 227 |
|
| 228 |
+
# This part is tricky with Gradio's file handling on load.
|
| 229 |
+
# For simplicity, we'll pass the path and have the API function handle it.
|
| 230 |
+
# A more robust solution might involve writing to a temp file.
|
| 231 |
+
# This is a conceptual fix for the demo.
|
| 232 |
+
class TempFile:
|
| 233 |
+
def __init__(self, name):
|
| 234 |
+
self.name = name
|
| 235 |
+
|
| 236 |
+
demo_files = [TempFile(name=str(sample_md_path))]
|
| 237 |
+
|
| 238 |
previews, logs, files = generate_pdfs_api(demo_files, ["A4 Portrait"], [AVAILABLE_FONTS[0]] if AVAILABLE_FONTS else [], 1)
|
| 239 |
return previews, logs, files
|
| 240 |
|