Update app.py

app.py

@@ -5,7 +5,8 @@ from PIL import Image
 from io import BytesIO
 from tqdm import tqdm
 import time
-import cairosvg
+import numpy as np
+import potrace
 
 # Defining the repository information and the trigger word
 repo = "artificialguybr/LineAniRedmond-LinearMangaSDXL-V2"
@@ -52,9 +53,19 @@ def generate_image(prompt):
         print("API response status code:", response.status_code)
         if response.status_code == 200:
             print("Image generation successful!")
-            # Convert the response content to an SVG with only black values
-            image = Image.open(BytesIO(response.content)).convert("L")  # Convert to grayscale
-            svg_data = cairosvg.svg2svg(bytestring=image.tobytes(), output_width=image.width, output_height=image.height)
+            # Open the image and convert it to grayscale
+            image = Image.open(BytesIO(response.content)).convert("L")
+            # Convert the grayscale image to a binary image
+            image = image.point(lambda p: 255 if p > 128 else 0, mode='1')
+            # Convert the PIL image to a numpy array
+            array = np.array(image)
+            # Invert the array for potrace (black on white)
+            array = 255 - array
+            # Trace the bitmap to SVG
+            bitmap = potrace.Bitmap(array)
+            path = bitmap.trace()
+            # Generate SVG data
+            svg_data = path.to_svg()
             return svg_data
         elif response.status_code == 503:
             time.sleep(1)
@@ -69,7 +80,7 @@ def generate_image(prompt):
 iface = gr.Interface(
     fn=generate_image,
     inputs=gr.Textbox(lines=2, placeholder="Describe the subject here..."),
-    outputs="text",  # Changed output to text for SVG data
+    outputs="text",  # SVG data as text
     title="LineArt XL Image Generator",
     description=(
         "Powered by the generous GPU time from Redmond.AI, this LORA, fine-tuned on SD XL 1.0, excels at creating Lineart-themed images across a wide range of subjects. "
@@ -85,4 +96,4 @@ iface = gr.Interface(
 )
 
 print("Launching Gradio interface...")
-iface.launch()
+iface.launch()