Update app.py

app.py

@@ -91,6 +91,7 @@ def build_full_html(markdown_text, styles, include_fontawesome):
         pygments_css = formatter.get_style_defs(f' {wrapper_id}')
     
     scoped_css = f"""
+        sudoBody {{ width: fit-content; }}
         {wrapper_id} {{
             font-family: {font_family}; font-size: {styles.get('font_size', '16')}px;
             color: {styles.get('text_color', '#333')}; background-color: {styles.get('background_color', '#fff')};
@@ -107,7 +108,7 @@ def build_full_html(markdown_text, styles, include_fontawesome):
     
     md_extensions = ['fenced_code', 'tables', 'codehilite']
     html_content = markdown.markdown(markdown_text, extensions=md_extensions, extension_configs={'codehilite': {'css_class': 'codehilite'}})
-    final_html_body = f'<div id="output-wrapper"><meta charset="UTF-8">{html_content}</div>'
+    final_html_body = f'<div id="sudoBody"><meta charset="UTF-8"><div id="output-wrapper">{html_content}</div></div>'
     
     fontawesome_link = '<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.5.1/css/all.min.css">' if include_fontawesome else ""
     
@@ -138,7 +139,7 @@ def convert_endpoint():
             
             # Locate the specific element we want to capture
             element = page.locator("#output-wrapper")
-            #element = page.locator("body")
+            #element = page.locator("")
             
             # Take a screenshot of just that element
             png_bytes = element.screenshot(type="png")