Spaces:

broadfield-dev
/

markdown2PNG-dev

Paused

App Files Files Community

broadfield-dev commited on Jun 2, 2025

Commit

d5dc3fa

verified ·

1 Parent(s): 4b7ec42

Update app.py

Browse files

Files changed (1) hide show

app.py +45 -29

app.py CHANGED Viewed

@@ -2,26 +2,30 @@ from flask import Flask, request, render_template_string, send_file
 import markdown
 import imgkit
 import os
 from io import BytesIO
-import base64
 app = Flask(__name__)
-# Ensure a temporary directory exists for saving files
-os.makedirs("temp", exist_ok=True)
 @app.route("/", methods=["GET", "POST"])
 def index():
     preview_html = None
     download_available = False
     download_type = "png"
-    file_path = None
-    if request.method == "POST":
-        markdown_text = request.form.get("markdown_text")
-        download_type = request.form.get("download_type", "png")
-        if markdown_text:
             # Convert Markdown to HTML
             html_content = markdown.markdown(markdown_text, extensions=['fenced_code', 'tables'])
@@ -45,30 +49,37 @@ def index():
             """
             # Save HTML to a temporary file
-            html_path = "temp/output.html"
-            with open(html_path, "w") as f:
                 f.write(full_html)
             # Generate preview HTML
             preview_html = full_html
             download_available = True
-            if download_type == "html":
-                file_path = html_path
-            else:  # PNG
-                # Convert HTML to PNG using imgkit
-                png_path = "temp/output.png"
-                imgkit.from_string(full_html, png_path, options={"quiet": ""})
-                file_path = png_path
-            # Handle download if requested
             if "download" in request.form:
-                return send_file(
-                    file_path,
-                    as_attachment=True,
-                    download_name=f"output.{download_type}",
-                    mimetype="text/html" if download_type == "html" else "image/png"
-                )
     return render_template_string("""
     <!DOCTYPE html>
@@ -79,15 +90,16 @@ def index():
             body { font-family: Arial, sans-serif; max-width: 800px; margin: 0 auto; padding: 20px; }
             textarea { width: 100%; height: 300px; margin-bottom: 10px; }
             select, button { padding: 10px; margin: 5px; }
-            .preview { border: 1px solid #ddd; padding: 12px; margin-top: 20px; }
             .download-btn { background-color: #4CAF50; color: white; border: none; cursor: pointer; }
             .download-btn:hover { background-color: #45a049; }
         </style>
     </head>
     <body>
         <h1>Markdown to PNG/HTML Converter</h1>
         <form method="post">
-            <textarea name="markdown_text" placeholder="Paste your Markdown here...">{{ request.form.get('markdown_text', '') }}</textarea><br>
             <label for="download_type">Output format:</label>
             <select name="download_type">
                 <option value="png" {% if download_type == 'png' %}selected{% endif %}>PNG</option>
@@ -98,6 +110,9 @@ def index():
                 <button type="submit" name="download" value="true" class="download-btn">Download {{ download_type.upper() }}</button>
             {% endif %}
         </form>
         {% if preview_html %}
             <h2>Preview</h2>
             <div class="preview">
@@ -106,7 +121,8 @@ def index():
         {% endif %}
     </body>
     </html>
-    """, preview_html=preview_html, download_available=download_available, download_type=download_type)
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))

 import markdown
 import imgkit
 import os
+import traceback
 from io import BytesIO
 app = Flask(__name__)
+# Use a directory within the app's working directory to avoid permission issues
+TEMP_DIR = os.path.join(os.getcwd(), "temp")
+# Create temporary directory if it doesn't exist
+try:
+    os.makedirs(TEMP_DIR, exist_ok=True)
+except Exception as e:
+    print(f"Error creating temp directory: {e}")
 @app.route("/", methods=["GET", "POST"])
 def index():
     preview_html = None
     download_available = False
     download_type = "png"
+    error_message = None
+    markdown_text = request.form.get("markdown_text", "") if request.method == "POST" else ""
+    if request.method == "POST" and markdown_text:
+        try:
             # Convert Markdown to HTML
             html_content = markdown.markdown(markdown_text, extensions=['fenced_code', 'tables'])
             """
             # Save HTML to a temporary file
+            html_path = os.path.join(TEMP_DIR, "output.html")
+            with open(html_path, "w", encoding="utf-8") as f:
                 f.write(full_html)
             # Generate preview HTML
             preview_html = full_html
             download_available = True
+            download_type = request.form.get("download_type", "png")
             if "download" in request.form:
+                if download_type == "html":
+                    return send_file(
+                        html_path,
+                        as_attachment=True,
+                        download_name="output.html",
+                        mimetype="text/html"
+                    )
+                else:  # PNG
+                    # Convert HTML to PNG using imgkit
+                    png_path = os.path.join(TEMP_DIR, "output.png")
+                    imgkit.from_string(full_html, png_path, options={"quiet": ""})
+                    return send_file(
+                        png_path,
+                        as_attachment=True,
+                        download_name="output.png",
+                        mimetype="image/png"
+                    )
+        except Exception as e:
+            error_message = f"Error processing request: {str(e)}"
+            print(f"Error: {traceback.format_exc()}")
     return render_template_string("""
     <!DOCTYPE html>
             body { font-family: Arial, sans-serif; max-width: 800px; margin: 0 auto; padding: 20px; }
             textarea { width: 100%; height: 300px; margin-bottom: 10px; }
             select, button { padding: 10px; margin: 5px; }
+            .preview { border: 1px solid #ddd; padding: 15px; margin-top: 20px; }
             .download-btn { background-color: #4CAF50; color: white; border: none; cursor: pointer; }
             .download-btn:hover { background-color: #45a049; }
+            .error { color: red; margin-top: 10px; }
         </style>
     </head>
     <body>
         <h1>Markdown to PNG/HTML Converter</h1>
         <form method="post">
+            <textarea name="markdown_text" placeholder="Paste your Markdown here...">{{ markdown_text }}</textarea><br>
             <label for="download_type">Output format:</label>
             <select name="download_type">
                 <option value="png" {% if download_type == 'png' %}selected{% endif %}>PNG</option>
                 <button type="submit" name="download" value="true" class="download-btn">Download {{ download_type.upper() }}</button>
             {% endif %}
         </form>
+        {% if error_message %}
+            <p class="error">{{ error_message }}</p>
+        {% endif %}
         {% if preview_html %}
             <h2>Preview</h2>
             <div class="preview">
         {% endif %}
     </body>
     </html>
+    """, preview_html=preview_html, download_available=download_available,
+       download_type=download_type, error_message=error_message, markdown_text=markdown_text)
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))