Update app.py

app.py

@@ -10,7 +10,6 @@ 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__)
 
@@ -127,43 +126,29 @@ 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 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)
+        # Define options here to avoid repetition and add the required fix
+        options = {"quiet": "", 'encoding': "UTF-8", "--no-cache": ""}
         
         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:
-                # Use from_file instead of from_string
-                png_bytes = imgkit.from_file(temp_html_path, False, options=options)
+                png_bytes = imgkit.from_string(full_html, False, options=options)
                 return send_file(BytesIO(png_bytes), as_attachment=True, download_name="output.png", mimetype="image/png")
         else:
-            # Use from_file instead of from_string
-            png_bytes = imgkit.from_file(temp_html_path, False, options=options)
+            png_bytes = imgkit.from_string(full_html, 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('/')