Spaces:
Running
Running
File size: 2,718 Bytes
cb6e75a bb03e66 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
import gradio as gr
import subprocess
from PIL import Image
import re
def convert_and_compile(input_file):
docx_file = input_file.name
latex_file = "output.tex"
pdf_file = "output.pdf"
png_file = "output.png"
# Define the helper functions
def separate_text_equations(text):
text = re.sub(r"\\\(", "\n\\(", text)
text = re.sub(r"\\\)", "\\)\n", text)
return text
def remove_unnecessary_latex_commands(text):
lines = text.split("\n")
for i in range(len(lines)):
line = lines[i]
if "\\RL" in line:
line = re.sub(r"\\RL\{([^}]*)\}", r"\1", line)
if "\\text" in line:
line = re.sub(r"\\text\{([^}]*)\}", r"\1", line)
lines[i] = line
return "\n".join(lines).strip()
# Convert .docx to .md
subprocess.run(["pandoc", "-s", docx_file, "-t", "markdown", "-o", "intermediate.md"], check=True)
# Convert .md to .tex
subprocess.run(["pandoc", "-s", "intermediate.md", "-t", "latex", "-o", "intermediate.tex"], check=True)
# Extract content between \begin{document} and \end{document}
with open("intermediate.tex", "r") as f:
content = f.read()
content = re.search(r"\\begin\{document\}([\s\S]*)\\end\{document\}", content).group(1)
# Separate text and equations, and remove unnecessary LaTeX commands
content = separate_text_equations(content)
content = remove_unnecessary_latex_commands(content)
# Generate a new .tex file
with open(latex_file, "w") as f:
f.write("\\documentclass[12pt]{standalone}\n")
f.write("\\input{etc/cmd}\n")
f.write("\\usepackage[utf8]{inputenc}\n")
f.write("\\usepackage{amsmath}\n")
f.write("\\usepackage{amssymb}\n")
f.write("\\usepackage{hyperref}\n")
f.write("\\usepackage{varwidth}\n")
f.write("\\usepackage{adjustbox}\n")
f.write("\\begin{document}\n")
f.write("\\begin{adjustbox}{margin=5mm}\n")
f.write("\\begin{varwidth}{\\linewidth}\n")
f.write(content)
f.write("\n\\end{varwidth}\n")
f.write("\\end{adjustbox}\n")
f.write("\\end{document}")
# Compile LaTeX to PDF with XeLaTeX
subprocess.run(["xelatex", "-interaction=nonstopmode", latex_file], check=True)
# Convert PDF to PNG with pdftoppm
subprocess.run(["pdftoppm", "-png", "-singlefile", pdf_file, "output"], check=True)
# Load the PNG image and return it
return Image.open(png_file)
iface = gr.Interface(
fn=convert_and_compile,
inputs=gr.File(label="Upload .DocX"),
outputs=gr.Image(type="pil", label="Latex Output", show_label=False),
)
iface.launch()
|