Update app.py

app.py

@@ -6,6 +6,7 @@ import time
 from huggingface_hub import hf_hub_download
 from handwriting_api import InputData, validate_input
 from hand import Hand
+import Cairosvg
 
 # Create img directory if it doesn't exist
 os.makedirs("img", exist_ok=True)
@@ -78,7 +79,7 @@ def generate_handwriting(
         return f"Error: {str(e)}"
 
 def export_to_png(svg_content):
-    """Convert SVG to PNG using CairoSVG"""
+    """Convert SVG to transparent PNG using CairoSVG"""
     try:
         import cairosvg
         
@@ -89,11 +90,17 @@ def export_to_png(svg_content):
         with open("img/temp.svg", "w") as f:
             f.write(svg_content)
             
-        # Convert SVG to PNG
-        cairosvg.svg2png(url="img/temp.svg", write_to="img/output.png", scale=2.0)
+        # Convert SVG to PNG with transparency
+        cairosvg.svg2png(
+            url="img/temp.svg", 
+            write_to="img/output.png", 
+            scale=2.0,
+            background_color="none"  # This ensures transparency
+        )
         
         return "img/output.png"
     except Exception as e:
+        print(f"Error converting to PNG: {str(e)}")
         return None
 
 def generate_lyrics_sample():
@@ -110,7 +117,8 @@ def generate_handwriting_wrapper(
     multiline=True
 ):
     svg = generate_handwriting(text, style, bias, color, stroke_width, multiline)
-    return svg
+    png_path = export_to_png(svg)
+    return svg, png_path
 
 css = """
 .container {max-width: 900px; margin: auto;}
@@ -173,10 +181,12 @@ with gr.Blocks(css=css) as demo:
                 sample_btn = gr.Button("Insert Sample Text")
             
         with gr.Column(scale=3):
-            output_svg = gr.HTML(label="Generated Handwriting", elem_classes=["output-container"])
+            output_svg = gr.HTML(label="Generated Handwriting (SVG)", elem_classes=["output-container"])
+            output_png = gr.Image(type="filepath", label="Generated Handwriting (PNG)", elem_classes=["output-container"])
             
             with gr.Row():
-                download_btn = gr.Button("Download SVG")
+                download_svg_btn = gr.Button("Download SVG")
+                download_png_btn = gr.Button("Download PNG")
     
     gr.Markdown("""
     ### Tips:
@@ -185,6 +195,7 @@ with gr.Blocks(css=css) as demo:
     - Each line should be 75 characters or less
     - The model works best for English text
     - Forward slashes (/) and backslashes (\\) will be replaced with dashes (-)
+    - PNG output has transparency for easy integration into other documents
     """)
     
     gr.Markdown("""
@@ -197,7 +208,7 @@ with gr.Blocks(css=css) as demo:
     generate_btn.click(
         fn=generate_handwriting_wrapper,
         inputs=[text_input, style_select, bias_slider, color_picker, stroke_width],
-        outputs=[output_svg]
+        outputs=[output_svg, output_png]
     )
     
     clear_btn.click(
@@ -212,13 +223,28 @@ with gr.Blocks(css=css) as demo:
         outputs=[text_input]
     )
     
-    download_btn.click(
+    download_svg_btn.click(
         fn=lambda x: x,
         inputs=[output_svg],
         outputs=[gr.File(label="Download SVG", file_count="single", file_types=[".svg"])]
     )
+    
+    download_png_btn.click(
+        fn=lambda x: x,
+        inputs=[output_png],
+        outputs=[gr.File(label="Download PNG", file_count="single", file_types=[".png"])]
+    )
 
 if __name__ == "__main__":
     # Set port based on environment variable or default to 7860
     port = int(os.environ.get("PORT", 7860))
+    
+    # Check if CairoSVG is installed
+    try:
+        import cairosvg
+        print("CairoSVG is installed and ready for PNG export")
+    except ImportError:
+        print("WARNING: CairoSVG is not installed. PNG export will not work.")
+        print("Please install it using: pip install cairosvg")
+    
     demo.launch(server_name="0.0.0.0", server_port=port)