app.py

from flask import Flask, request, jsonify
import os
import tempfile
from markitdown import MarkItDown
from werkzeug.utils import secure_filename

app = Flask(__name__)

@app.route('/health', methods=['GET'])
def health_check():
    return jsonify({"status": "healthy"}), 200

@app.route('/convert', methods=['POST'])
def convert_file():
    # Check if file is present in the request
    if 'file' not in request.files:
        return jsonify({"error": "No file provided"}), 400
    
    file = request.files['file']
    
    # Check if the file has a name
    if file.filename == '':
        return jsonify({"error": "No file selected"}), 400
    
    # Create a temporary file to save the uploaded file
    temp_dir = tempfile.mkdtemp()
    file_path = os.path.join(temp_dir, secure_filename(file.filename))
    
    try:
        # Save the file temporarily
        file.save(file_path)
        
        # Process with MarkItDown
        md = MarkItDown()
        result = md.convert(file_path)
        
        # Prepare the response
        response = {
            "markdown": result.text_content
        }
        
        # Add any other relevant data from the result object if needed
        # Example: if result has metadata, add it to the response
        if hasattr(result, 'metadata'):
            response["metadata"] = result.metadata
        
        return jsonify(response), 200
    
    except Exception as e:
        return jsonify({"error": str(e)}), 500
    
    finally:
        # Clean up the temporary file
        if os.path.exists(file_path):
            os.remove(file_path)
        
        # Clean up the temporary directory
        if os.path.exists(temp_dir):
            os.rmdir(temp_dir)