Update app.py

app.py

@@ -10,6 +10,7 @@ from pygments import highlight
 from pygments.lexers import get_lexer_by_name
 from pygments.formatters import HtmlFormatter
 from pygments.styles import get_all_styles
+import uuid # <--- REQUIRED CHANGE 1: Import uuid
 
 app = Flask(__name__)
 
@@ -126,29 +127,43 @@ def build_full_html(markdown_text, styles, include_fontawesome):
 @app.route('/convert', methods=['POST'])
 def convert_endpoint():
     data = request.json
+    temp_html_path = None # Initialize path
     try:
         full_html = build_full_html(
             markdown_text=data.get('markdown_text', ''),
             styles=data.get('styles', {}),
             include_fontawesome=data.get('include_fontawesome', False)
         )
-        # Define options here to avoid repetition and add the required fix
-        options = {"quiet": "", 'encoding': "UTF-8", "--no-cache": ""}
+        
+        # Define options for wkhtmltoimage, removing the unsupported --no-cache
+        options = {"quiet": "", 'encoding': "UTF-8"}
+
+        # --- REQUIRED CHANGE 2: Manually create and use a temporary file ---
+        temp_html_path = os.path.join(TEMP_DIR, f"{uuid.uuid4()}.html")
+        with open(temp_html_path, "w", encoding="utf-8") as f:
+            f.write(full_html)
         
         if data.get('download', False):
             download_type = data.get('download_type', 'png')
             if download_type == 'html':
                 return send_file(BytesIO(full_html.encode("utf-8")), as_attachment=True, download_name="output.html", mimetype="text/html")
             else:
-                png_bytes = imgkit.from_string(full_html, False, options=options)
+                # Use from_file instead of from_string
+                png_bytes = imgkit.from_file(temp_html_path, False, options=options)
                 return send_file(BytesIO(png_bytes), as_attachment=True, download_name="output.png", mimetype="image/png")
         else:
-            png_bytes = imgkit.from_string(full_html, False, options=options)
+            # Use from_file instead of from_string
+            png_bytes = imgkit.from_file(temp_html_path, False, options=options)
             png_base64 = base64.b64encode(png_bytes).decode('utf-8')
             return jsonify({'preview_html': full_html, 'preview_png_base64': png_base64})
+            
     except Exception as e:
         traceback.print_exc()
         return jsonify({'error': f'Failed to convert content: {str(e)}'}), 500
+    finally:
+        # --- REQUIRED CHANGE 3: Clean up the temporary file ---
+        if temp_html_path and os.path.exists(temp_html_path):
+            os.remove(temp_html_path)
 
 # --- MAIN PAGE RENDERER (with corrected CSS) ---
 @app.route('/')
@@ -305,4 +320,4 @@ def index():
 if __name__ == "__main__":
     # Ensure you have installed the required libraries:
     # pip install Flask markdown imgkit pygments
-    app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))
+    app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))