Imgmaker / app.py2.bak
Threscomma's picture
Rename app.py to app.py2.bak
f720f3c verified
import gradio as gr
import subprocess
import tempfile
import os
from pdf2image import convert_from_path
import logging
# Configure logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
def verify_libreoffice():
"""Check if LibreOffice exists before conversion"""
try:
result = subprocess.run(
["/usr/bin/libreoffice", "--version"],
check=True,
capture_output=True,
text=True
)
logger.info(f"LibreOffice verified: {result.stdout.strip()}")
return True
except Exception as e:
logger.error(f"LibreOffice check failed: {str(e)}")
return False
def convert_docx_to_image(docx_path):
if not verify_libreoffice():
raise gr.Error("System Error: LibreOffice not installed!")
with tempfile.TemporaryDirectory() as tmpdir:
try:
# Conversion steps...
pdf_path = os.path.join(tmpdir, "output.pdf")
subprocess.run([
"/usr/bin/libreoffice",
"--headless",
"--convert-to", "pdf",
"--outdir", tmpdir,
docx_path
], check=True, timeout=30)
images = convert_from_path(
pdf_path,
dpi=100,
poppler_path="/usr/bin"
)
return images[0] if images else None
except Exception as e:
logger.error(f"Conversion error: {str(e)}")
raise gr.Error(f"Failed: {str(e)}")
# Gradio Interface
with gr.Blocks() as app:
gr.Markdown("## 🖥️ DOCX to Image Converter")
with gr.Row():
input_file = gr.File(
label="Upload DOCX",
file_types=[".docx"],
type="filepath"
)
output_image = gr.Image(label="Output Image")
gr.Button("Convert").click(
fn=convert_docx_to_image,
inputs=input_file,
outputs=output_image
)
app.launch()