Update app.py

app.py

@@ -1,60 +1,53 @@
 import gradio as gr
-import os
 import subprocess
 import tempfile
 from pathlib import Path
 
-def convert_file(input_file, output_format):
+def convert_to_image(docx_file, output_format):
     try:
-        # Create temp directory
         with tempfile.TemporaryDirectory() as tmpdir:
             # Step 1: Save uploaded file
-            input_path = Path(input_file.name)
-            temp_input = Path(tmpdir) / input_path.name
-            with open(temp_input, "wb") as f:
-                f.write(input_file.read())
+            docx_path = Path(tmpdir) / "input.docx"
+            with open(docx_path, "wb") as f:
+                f.write(docx_file.read())
             
-            # Step 2: Convert using soffice
-            output_path = Path(tmpdir) / f"{temp_input.stem}.{output_format}"
-            cmd = [
+            # Step 2: Convert DOCX → PDF
+            pdf_path = Path(tmpdir) / "output.pdf"
+            subprocess.run([
                 "soffice",
                 "--headless",
-                "--convert-to", output_format,
+                "--convert-to", "pdf",
                 "--outdir", tmpdir,
-                str(temp_input)
-            ]
-            subprocess.run(cmd, check=True, timeout=30)
+                str(docx_path)
+            ], check=True, timeout=30)
             
-            # Verify output
-            if not output_path.exists():
-                raise FileNotFoundError("Conversion failed - no output file")
+            # Step 3: Convert PDF → Image
+            img_path = Path(tmpdir) / f"output.{output_format}"
+            subprocess.run([
+                "pdftoppm",
+                "-singlefile",
+                "-jpeg" if output_format == "jpg" else "-png",
+                "-r", "150",  # DPI (150 for free tier)
+                str(pdf_path),
+                str(img_path.with_suffix(''))
+            ], check=True)
             
-            return str(output_path)
+            return str(img_path) if img_path.exists() else "Conversion failed"
             
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Supported formats (LibreOffice 7.4+)
-supported_formats = [
-    "pdf", "docx", "odt", "html", 
-    "txt", "rtf", "pptx", "xlsx"
-]
-
-with gr.Blocks(title="File Converter") as app:
-    gr.Markdown("## 📁 File Converter (LibreOffice)")
+with gr.Blocks() as app:
+    gr.Markdown("## 📄 DOCX to Image Converter")
     with gr.Row():
-        file_input = gr.File(label="Upload File")
-        format_dropdown = gr.Dropdown(
-            supported_formats, 
-            label="Output Format",
-            value="pdf"
-        )
-        output_file = gr.File(label="Download Converted File")
+        file_input = gr.File(label="Upload DOCX", file_types=[".docx"])
+        format_radio = gr.Radio(["jpg", "png"], label="Output Format", value="jpg")
+    output_image = gr.Image(label="Result", type="filepath")
     
     gr.Button("Convert").click(
-        fn=convert_file,
-        inputs=[file_input, format_dropdown],
-        outputs=output_file
+        fn=convert_to_image,
+        inputs=[file_input, format_radio],
+        outputs=output_image
     )
 
 app.launch()