Update app.py

app.py

@@ -4,8 +4,8 @@ import os
 import base64
 import uuid
 
-def process_conversion(file_input, base64_input):
-    # Use /tmp for reliability on Hugging Face
+def process_conversion(base64_input):
+    # Setup paths in /tmp for Hugging Face environment
     unique_id = uuid.uuid4().hex
     temp_docx = f"/tmp/input_{unique_id}.docx"
     output_dir = "/tmp/outputs"
@@ -14,55 +14,72 @@ def process_conversion(file_input, base64_input):
         os.makedirs(output_dir)
 
     try:
-        # 1. Handle Input
-        if file_input is not None:
-            input_path = file_input.name
-        elif base64_input and base64_input.strip():
-            # Clean base64
-            if "," in base64_input:
-                base64_input = base64_input.split(",")[1]
-            with open(temp_docx, "wb") as f:
-                f.write(base64.b64decode(base64_input))
-            input_path = temp_docx
-        else:
-            return None, "No input provided."
+        # 1. Handle Base64 Input
+        if not base64_input or not base64_input.strip():
+            return None, "Error: No Base64 data provided."
 
-        # 2. Run Conversion
-        # Adding timeout and capturing errors
+        # Clean base64 string (remove data header if present)
+        if "," in base64_input:
+            base64_input = base64_input.split(",")[1]
+        
+        # Write the binary data to a temporary .docx file
+        # LibreOffice needs a physical file to perform the conversion
+        with open(temp_docx, "wb") as f:
+            f.write(base64.b64decode(base64_input))
+        
+        # 2. Run LibreOffice Conversion
         result = subprocess.run([
             'libreoffice', '--headless', '--convert-to', 'pdf', 
-            '--outdir', output_dir, input_path
+            '--outdir', output_dir, temp_docx
         ], capture_output=True, text=True, timeout=30)
 
         if result.returncode != 0:
             return None, f"LibreOffice Error: {result.stderr}"
 
-        # 3. Get Result
-        filename = os.path.basename(input_path)
-        pdf_filename = os.path.splitext(filename)[0] + ".pdf"
+        # 3. Locate and Read the Resulting PDF
+        pdf_filename = f"input_{unique_id}.pdf"
         pdf_path = os.path.join(output_dir, pdf_filename)
 
         if os.path.exists(pdf_path):
             with open(pdf_path, "rb") as f:
-                pdf_base64 = base64.b64encode(f.read()).decode('utf-8')
+                pdf_bytes = f.read()
+                pdf_base64 = base64.b64encode(pdf_bytes).decode('utf-8')
+            
+            # Cleanup the temporary docx file
+            if os.path.exists(temp_docx):
+                os.remove(temp_docx)
+                
+            # Return File object for download and the Base64 string for API use
             return gr.FileData(path=pdf_path, mime_type="application/pdf"), pdf_base64
         
-        return None, "Conversion failed - file not found."
+        return None, "Error: Conversion failed - PDF not generated."
 
     except Exception as e:
-        # This prevents 'event: error' and shows the real problem
         return None, f"System Error: {str(e)}"
 
+# UI Layout
 with gr.Blocks() as demo:
-    gr.Markdown("# DOCX to PDF")
-    with gr.Row():
-        file_in = gr.File(label="Upload", file_types=[".docx"])
-        text_in = gr.Textbox(label="Base64 Input")
-    btn = gr.Button("Convert")
-    with gr.Row():
-        file_out = gr.File(label="PDF File")
-        text_out = gr.Textbox(label="Base64 Output")
+    gr.Markdown("# 📄 DOCX to PDF (Base64 Only)")
+    gr.Markdown("Paste the Base64 string of a `.docx` file to convert it to `.pdf`.")
     
-    btn.click(process_conversion, [file_in, text_in], [file_out, text_out])
+    with gr.Row():
+        with gr.Column():
+            text_in = gr.Textbox(
+                label="Paste Base64 DOCX", 
+                placeholder="Input base64 string here...", 
+                lines=8
+            )
+            submit_btn = gr.Button("Convert to PDF", variant="primary")
+            
+        with gr.Column():
+            file_out = gr.File(label="Download PDF File")
+            text_out = gr.Textbox(label="Result Base64 PDF", lines=8, show_copy_button=True)
+
+    submit_btn.click(
+        fn=process_conversion,
+        inputs=[text_in],
+        outputs=[file_out, text_out]
+    )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()