Update app.py

app.py

@@ -5,73 +5,45 @@ import os
 from pdf2image import convert_from_path
 import logging
 
-# Configure logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-def convert_docx_to_image(docx_path: str) -> str:
-    """
-    Converts DOCX to image via LibreOffice and pdf2image.
-    Returns path to the generated image.
-    """
+# 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")
+
+def convert_docx_to_image(docx_path: str):
     with tempfile.TemporaryDirectory() as tmpdir:
         try:
-            # Step 1: DOCX → PDF using LibreOffice
+            # DOCX → PDF
             pdf_path = os.path.join(tmpdir, "output.pdf")
             cmd = [
                 "libreoffice", "--headless", "--convert-to", "pdf",
                 "--outdir", tmpdir, docx_path
             ]
-            subprocess.run(cmd, check=True, capture_output=True)
-            
-            if not os.path.exists(pdf_path):
-                raise FileNotFoundError("PDF conversion failed")
-
-            # Step 2: PDF → Image
-            images = convert_from_path(
-                pdf_path,
-                dpi=300,  # High resolution
-                output_folder=tmpdir,
-                fmt="png",
-                single_file=True
-            )
-            output_path = os.path.join(tmpdir, "output.png")
-            images[0].save(output_path, "PNG")
+            subprocess.run(cmd, check=True, timeout=30)
             
-            return output_path
-
-        except subprocess.CalledProcessError as e:
-            logger.error(f"LibreOffice error: {e.stderr.decode()}")
-            raise
+            # 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")
         except Exception as e:
-            logger.error(f"Conversion failed: {str(e)}")
-            raise
+            logging.error(f"Conversion failed: {str(e)}")
+            raise gr.Error(f"Error: {str(e)}")
 
-# Gradio Interface
-with gr.Blocks(title="DOCX to Image (Pixel-Perfect)") as app:
-    gr.Markdown("## 📄→🖼️ Convert Word to Image (Exact Visual Copy)")
+with gr.Blocks() as app:
+    gr.Markdown("# 📄 screenshot my unit planner")
     with gr.Row():
-        input_file = gr.File(
-            label="Upload DOCX",
-            file_types=[".docx"],
-            type="filepath"
-        )
-        output_image = gr.Image(label="Rendered Image", type="filepath")
-    
-    convert_btn = gr.Button("Convert", variant="primary")
-    status = gr.Textbox(label="Status", interactive=False)
-
-    def process(file_path: str) -> tuple[str, str]:
-        try:
-            image_path = convert_docx_to_image(file_path)
-            return image_path, "✅ Conversion successful!"
-        except Exception as e:
-            return None, f"❌ Error: {str(e)}"
-
-    convert_btn.click(
-        fn=process,
+        input_file = gr.File(label="Upload DOCX", file_types=[".docx"])
+        output_image = gr.Image(label="Result")
+    gr.Button("Convert").click(
+        convert_docx_to_image,
         inputs=input_file,
-        outputs=[output_image, status]
+        outputs=output_image
     )
 
-app.launch(server_name="0.0.0.0")
+app.launch()