Spaces:

broadfield-dev
/

md2png

Paused

App Files Files Community

broadfield-dev commited on Oct 15, 2025

Commit

48891a9

verified ·

1 Parent(s): 80a56ff

Update app.py

Browse files

Files changed (1) hide show

app.py +37 -13

app.py CHANGED Viewed

@@ -77,14 +77,14 @@ def parse_endpoint():
     except Exception as e:
         return jsonify({'error': f'Failed to parse: {e}'}), 500
-# --- HTML & PNG BUILDER (CHANGED) ---
 def build_full_html(markdown_text, styles, include_fontawesome):
     wrapper_id = "#output-wrapper"
     font_family = styles.get('font_family', "'Arial', sans-serif")
-    # *** THIS IS THE FIX: Forcefully disable ALL external network calls ***
     google_font_link = ""
-    fontawesome_link = ""
     highlight_theme = styles.get('highlight_theme', 'default')
     pygments_css = ""
@@ -111,6 +111,8 @@ def build_full_html(markdown_text, styles, include_fontawesome):
     html_content = markdown.markdown(markdown_text, extensions=md_extensions, extension_configs={'codehilite': {'css_class': 'codehilite'}})
     final_html_body = f'<div id="output-wrapper">{html_content}</div>'
     full_html = f"""<!DOCTYPE html>
     <html><head><meta charset="UTF-8">{google_font_link}{fontawesome_link}<style>
     #ouput-wrapper {{ background-color: {styles.get('background_color', '#fff')}; padding: 25px; display: inline-block;}}
@@ -131,7 +133,7 @@ def convert_endpoint():
             include_fontawesome=data.get('include_fontawesome', False)
         )
-        # We need xvfb because wkhtmltoimage is a headless browser
         options = {
             "quiet": "",
             'encoding': "UTF-8",
@@ -161,7 +163,7 @@ def convert_endpoint():
         if temp_html_path and os.path.exists(temp_html_path):
             os.remove(temp_html_path)
-# --- MAIN PAGE RENDERER (Unchanged) ---
 @app.route('/')
 def index():
     highlight_styles = sorted(list(get_all_styles()))
@@ -236,7 +238,7 @@ def index():
                 if (componentsFieldset.style.display === 'block') {
                     const parts = [];
                     componentsContainer.querySelectorAll('.component-container').forEach(div => {
-                        if (div.querySelector('.component-checkbox').checked) { parts.push(div.dataset.reconstructed || div.dataset.content); }
                     });
                     finalMarkdown = parts.join('\\n\\n---\\n\\n');
                 } else { finalMarkdown = markdownTextInput.value; }
@@ -293,11 +295,33 @@ def index():
                     if (result.format !== 'Unknown') {
                         componentsFieldset.style.display = 'block';
                         result.components.forEach((comp, index) => {
-                            const div = document.createElement('div'); div.className = 'component-container'; div.dataset.type = comp.type; div.dataset.filename = comp.filename; div.dataset.content = comp.content;
-                            let reconstructedContent = comp.content;
-                            if (comp.is_code_block) { div.dataset.isCodeBlock = 'true'; div.dataset.language = comp.language || ''; reconstructedContent = "```" + (comp.language || '') + "\\n" + comp.content + "\\n```"; }
-                            if (comp.type === 'section') div.dataset.reconstructed = `## ${comp.filename}\\n${comp.content}`; if (comp.type === 'version') div.dataset.reconstructed = `## ${comp.filename}\\n${comp.content}`;
-                            div.innerHTML = `<div class="component-header"><input type="checkbox" id="comp-check-${index}" class="component-checkbox" checked><label for="comp-check-${index}">${comp.filename}</label></div><div class="component-content"><textarea readonly>${comp.content}</textarea></div>`;
                             componentsContainer.appendChild(div);
                         });
                     }
@@ -310,4 +334,4 @@ def index():
     """, highlight_styles=highlight_styles)
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))

     except Exception as e:
         return jsonify({'error': f'Failed to parse: {e}'}), 500
+# --- HTML & PNG BUILDER (Unchanged) ---
 def build_full_html(markdown_text, styles, include_fontawesome):
     wrapper_id = "#output-wrapper"
     font_family = styles.get('font_family', "'Arial', sans-serif")
+    google_font_name = font_family.split(',')[0].strip("'\"")
     google_font_link = ""
+    if " " in google_font_name and google_font_name not in ["Times New Roman", "Courier New"]:
+        google_font_link = f'<link href="https://fonts.googleapis.com/css2?family={google_font_name.replace(" ", "+")}:wght@400;700&display=swap" rel="stylesheet">'
     highlight_theme = styles.get('highlight_theme', 'default')
     pygments_css = ""
     html_content = markdown.markdown(markdown_text, extensions=md_extensions, extension_configs={'codehilite': {'css_class': 'codehilite'}})
     final_html_body = f'<div id="output-wrapper">{html_content}</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 ""
     full_html = f"""<!DOCTYPE html>
     <html><head><meta charset="UTF-8">{google_font_link}{fontawesome_link}<style>
     #ouput-wrapper {{ background-color: {styles.get('background_color', '#fff')}; padding: 25px; display: inline-block;}}
             include_fontawesome=data.get('include_fontawesome', False)
         )
+        # Use xvfb, which is required for stable headless rendering.
         options = {
             "quiet": "",
             'encoding': "UTF-8",
         if temp_html_path and os.path.exists(temp_html_path):
             os.remove(temp_html_path)
+# --- MAIN PAGE RENDERER (with corrected JavaScript) ---
 @app.route('/')
 def index():
     highlight_styles = sorted(list(get_all_styles()))
                 if (componentsFieldset.style.display === 'block') {
                     const parts = [];
                     componentsContainer.querySelectorAll('.component-container').forEach(div => {
+                        if (div.querySelector('.component-checkbox').checked) { parts.push(div.dataset.reconstructed); }
                     });
                     finalMarkdown = parts.join('\\n\\n---\\n\\n');
                 } else { finalMarkdown = markdownTextInput.value; }
                     if (result.format !== 'Unknown') {
                         componentsFieldset.style.display = 'block';
                         result.components.forEach((comp, index) => {
+                            const div = document.createElement('div');
+                            div.className = 'component-container';
+                            // *** THIS IS THE FIX ***
+                            // We create the full markdown for reconstruction here, ensuring headings are preserved.
+                            let full_content_for_reconstruction;
+                            let display_content = comp.content; // Content for the <textarea>
+                            let content_with_fences = comp.content;
+                            if (comp.is_code_block) {
+                                content_with_fences = "```" + (comp.language || '') + "\\n" + comp.content + "\\n```";
+                            }
+                            if (comp.type === 'file') {
+                                // For repo2md, format each file as a Level 3 heading with its content
+                                full_content_for_reconstruction = `### File: ${comp.filename}\\n${content_with_fences}`;
+                            } else if (comp.type === 'section' || comp.type === 'version') {
+                                // For other formats, use a Level 2 heading
+                                full_content_for_reconstruction = `## ${comp.filename}\\n${comp.content}`;
+                            } else {
+                                // For intro text, use the content as-is
+                                full_content_for_reconstruction = comp.content;
+                            }
+                            div.dataset.reconstructed = full_content_for_reconstruction;
+                            div.innerHTML = `<div class="component-header"><input type="checkbox" id="comp-check-${index}" class="component-checkbox" checked><label for="comp-check-${index}">${comp.filename}</label></div><div class="component-content"><textarea readonly>${display_content}</textarea></div>`;
                             componentsContainer.appendChild(div);
                         });
                     }
     """, highlight_styles=highlight_styles)
 if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))```