Update app.py

app.py

@@ -5,45 +5,66 @@ import os
 from pdf2image import convert_from_path
 import logging
 
-# Verify LibreOffice exists
-try:
-    subprocess.run(["libreoffice", "--version"], check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-    logging.info("✅ LibreOffice is installed")
-except Exception as e:
-    logging.error("❌ LibreOffice not found. Check your .dockerfile!")
-    raise RuntimeError("LibreOffice not installed")
+# ===== HEALTH CHECK =====
+def verify_libreoffice():
+    try:
+        subprocess.run(["libreoffice", "--version"], check=True, capture_output=True)
+        return True
+    except (subprocess.CalledProcessError, FileNotFoundError):
+        return False
 
-def convert_docx_to_image(docx_path: str):
+if not verify_libreoffice():
+    raise RuntimeError("❌ LibreOffice not installed. Check your .dockerfile!")
+
+# ===== MAIN CONVERSION FUNCTION =====
+def convert_docx_to_image(docx_path: str) -> str:
     with tempfile.TemporaryDirectory() as tmpdir:
         try:
-            # DOCX → PDF
+            # Step 1: DOCX → PDF
             pdf_path = os.path.join(tmpdir, "output.pdf")
             cmd = [
-                "libreoffice", "--headless", "--convert-to", "pdf",
-                "--outdir", tmpdir, docx_path
+                "libreoffice", 
+                "--headless",
+                "--convert-to", "pdf",
+                "--outdir", tmpdir,
+                docx_path
             ]
             subprocess.run(cmd, check=True, timeout=30)
             
-            # PDF → PNG
-            images = convert_from_path(pdf_path, dpi=300, fmt="png")
-            if images:
-                output_path = os.path.join(tmpdir, "output.png")
-                images[0].save(output_path)
-                return output_path
-            raise ValueError("No images generated")
+            if not os.path.exists(pdf_path):
+                raise FileNotFoundError("PDF was not generated")
+            
+            # Step 2: PDF → PNG
+            images = convert_from_path(
+                pdf_path,
+                dpi=300,
+                output_folder=tmpdir,
+                fmt="png",
+                single_file=True
+            )
+            output_path = os.path.join(tmpdir, "output.png")
+            images[0].save(output_path, "PNG")
+            return output_path
+
         except Exception as e:
-            logging.error(f"Conversion failed: {str(e)}")
-            raise gr.Error(f"Error: {str(e)}")
+            logging.error(f"Conversion failed: {str(e)}", exc_info=True)
+            raise gr.Error(f"Conversion error: {str(e)}")
 
+# ===== GRADIO UI =====
 with gr.Blocks() as app:
-    gr.Markdown("# 📄 screenshot my unit planner")
+    gr.Markdown("## 📄 DOCX to Image (Pixel-Perfect)")
     with gr.Row():
-        input_file = gr.File(label="Upload DOCX", file_types=[".docx"])
-        output_image = gr.Image(label="Result")
+        input_file = gr.File(
+            label="Upload DOCX",
+            file_types=[".docx"],
+            type="filepath"
+        )
+        output_image = gr.Image(label="Output Image", type="filepath")
+    
     gr.Button("Convert").click(
-        convert_docx_to_image,
+        fn=convert_docx_to_image,
         inputs=input_file,
         outputs=output_image
     )
 
-app.launch()
+app.launch(server_name="0.0.0.0")