Deploy Document Format Converter

README.md

@@ -1,3 +1,16 @@
+---
+title: Document Format Converter
+emoji: 🙊
+colorFrom: green
+colorTo: indigo
+sdk: gradio
+sdk_version: 5.13.2
+app_file: app.py
+pinned: false
+license: apache-2.0
+short_description: Upload document and select target format for conversion.
+---
+
 # Document Format Converter
 
 Upload document and select target format for conversion.
@@ -8,7 +21,11 @@ This application provides a web interface for document conversion. Simply upload
 
 ## API Endpoints
 
-The application also provides API endpoints for programmatic access:
+The application provides API endpoints for programmatic access. **API key authentication is required.**
+
+### Authentication
+
+All API requests require an API key to be sent in the `X-API-Key` header.
 
 ### 1. DOCX to JSON Conversion
 
@@ -18,13 +35,18 @@ import requests
 # URL of the API endpoint
 url = "https://huggingface.co/spaces/ObiJuanCodenobi/docgen/api/docx-to-json"
 
+# API key for authentication
+headers = {
+    'X-API-Key': 'YOUR_API_KEY'
+}
+
 # Prepare the file for upload
 files = {
     'file': ('document.docx', open('path/to/your/document.docx', 'rb'), 'application/vnd.openxmlformats-officedocument.wordprocessingml.document')
 }
 
 # Send the request
-response = requests.post(url, files=files)
+response = requests.post(url, files=files, headers=headers)
 
 # Get the JSON result
 if response.status_code == 200:
@@ -42,13 +64,18 @@ import requests
 # URL of the API endpoint
 url = "https://huggingface.co/spaces/ObiJuanCodenobi/docgen/api/json-to-docx"
 
+# API key for authentication
+headers = {
+    'X-API-Key': 'YOUR_API_KEY',
+    'Content-Type': 'application/json'
+}
+
 # Your JSON document data
 json_data = {
     # Your document structure here
 }
 
 # Send the request
-headers = {'Content-Type': 'application/json'}
 response = requests.post(url, json=json_data, headers=headers)
 
 # Save the DOCX file

app.py

@@ -14,6 +14,7 @@ import sys
 import tempfile
 from flask import Flask, request, jsonify, send_file
 import threading
+import secrets
 
 os.system('sudo apt-get install texlive')
 
@@ -521,11 +522,26 @@ if __name__ == "__main__":
         parity_check(sys.argv[2])
         sys.exit(0)
     
+    # Generate a random API key if one doesn't exist in environment variables
+    API_KEY = os.environ.get('API_KEY', secrets.token_urlsafe(32))
+    print(f"API Key: {API_KEY}")  # Print the API key when the app starts
+
     # Create Flask app for API endpoints
     app = Flask(__name__)
     
+    def check_api_key():
+        """Check if the API key is valid."""
+        provided_key = request.headers.get('X-API-Key')
+        if not provided_key or provided_key != API_KEY:
+            return False
+        return True
+    
     @app.route('/api/docx-to-json', methods=['POST'])
     def api_docx_to_json():
+        # Check API key
+        if not check_api_key():
+            return jsonify({"error": "Invalid or missing API key"}), 401
+            
         if 'file' not in request.files:
             return jsonify({"error": "No file part"}), 400
         
@@ -558,6 +574,10 @@ if __name__ == "__main__":
     
     @app.route('/api/json-to-docx', methods=['POST'])
     def api_json_to_docx():
+        # Check API key
+        if not check_api_key():
+            return jsonify({"error": "Invalid or missing API key"}), 401
+            
         if not request.is_json:
             return jsonify({"error": "Request must be JSON"}), 400
         

samples/sample.json

@@ -0,0 +1 @@
+{"title": "Sample Document", "content": "This is a sample JSON document for testing the converter."}