Update app.py

app.py

@@ -2,48 +2,40 @@ import gradio as gr
 import subprocess
 import tempfile
 from pathlib import Path
-import platform
+import time
 
 def convert_docx_to_jpg(docx_file):
-    if not docx_file:
-        raise gr.Error("No file uploaded. Please upload a DOCX file.")
-    if not docx_file.name.lower().endswith('.docx'):
-        raise gr.Error("Please upload a valid .docx file.")
-
     try:
         with tempfile.TemporaryDirectory() as tmpdir:
-            # Save uploaded file
-            docx_path = Path(tmpdir) / "input.docx"
+            tmpdir = Path(tmpdir)
+            
+            # 1. Save the uploaded file
+            docx_path = tmpdir / "input.docx"
             with open(docx_path, "wb") as f:
                 f.write(open(docx_file.name, "rb").read())
             
-            # Convert DIRECTLY to JPG using LibreOffice
-            jpg_path = Path(tmpdir) / "input.jpg"  # Match LibreOffice’s naming
-            libreoffice_cmd = "soffice" if platform.system() != "Windows" else "soffice.exe"
-            try:
-                result = subprocess.run(
-                    [libreoffice_cmd, "--headless", "--convert-to", "jpg", "--outdir", tmpdir, str(docx_path)],
-                    check=True,
-                    timeout=60,
-                    capture_output=True,
-                    text=True
-                )
-            except subprocess.CalledProcessError as e:
-                raise gr.Error(f"LibreOffice conversion failed: {e.stderr}")
-            except subprocess.TimeoutExpired:
-                raise gr.Error("Conversion timed out after 60 seconds.")
+            # 2. Convert DOCX to JPG using LibreOffice's export function
+            subprocess.run([
+                "soffice",
+                "--headless",
+                "--convert-to", "jpg",
+                "--outdir", str(tmpdir),
+                str(docx_path)
+            ], check=True, timeout=30)
             
-            # Verify output exists
-            if not jpg_path.exists():
-                raise ValueError("JPG conversion failed")
+            # 3. Find the output file (LibreOffice adds '_1' sometimes)
+            jpg_file = next(tmpdir.glob("*.jpg"), None)
+            if not jpg_file:
+                raise ValueError("No JPG file was created")
             
-            return str(jpg_path)
+            # 4. Return the first page only
+            return str(jpg_file)
             
     except Exception as e:
-        raise gr.Error(f"Failed to convert: {str(e)}")
+        raise gr.Error(f"Conversion failed. Try a simpler DOCX file. Error: {str(e)}")
 
 with gr.Blocks() as app:
-    gr.Markdown("## 📄 Direct DOCX to JPG Converter")
+    gr.Markdown("## 📄 DOCX to JPG (Working Version)")
     file_input = gr.File(label="Upload DOCX", file_types=[".docx"])
     output_image = gr.Image(label="Output JPG", type="filepath")
     
@@ -53,7 +45,4 @@ with gr.Blocks() as app:
         outputs=output_image
     )
 
-try:
-    app.launch()
-except Exception as e:
-    raise gr.Error(f"Failed to launch Gradio app: {str(e)}")
+app.launch()