Bugfix DeepSvg

animationPipeline.py

@@ -35,8 +35,10 @@ def animateLogo(path : str, targetPath : str):
     ).to(device)
 
     model.load_state_dict(torch.load("models/animation_transformer2.pth", map_location=torch.device('cpu')), strict=False)
-
-    df = compute_embedding(path, "models/deepSVG_hierarchical_ordered.pth.tar")
+    try:
+        df = compute_embedding(path, "models/deepSVG_hierarchical_ordered.pth.tar")
+    except:
+        return False
     df = df.drop("animation_id", axis=1)
 
     df = pd.concat([df, pd.DataFrame(0, index=df.index, columns=range(df.shape[1], df.shape[1] + 26))], axis=1, ignore_index=True).astype(float)
@@ -52,6 +54,7 @@ def animateLogo(path : str, targetPath : str):
     result["animation_id"] = range(len(result))
     print(result, path)
     animate_logo(result, targetPath)
+    return True
 
 #logo = "data/examples/logo_181.svg"
 #animateLogo(logo)

app.py

@@ -20,7 +20,10 @@ if uploaded_file is not None:
         f.write(uploaded_file.getbuffer())         
       st.success("Saved File")
       sys.setrecursionlimit(1500)
-      animateLogo(path, targetPath)
-      with open(targetPath, "rb") as file:
-        st.write(file)
-        st.download_button('Download animated SVG', data=file, file_name=uploaded_file.name[:-4]+"_animated.svg")
+      success = animateLogo(path, targetPath)
+      if success:
+        with open(targetPath, "rb") as file:
+          st.write(file)
+          st.download_button('Download animated SVG', data=file, file_name=uploaded_file.name[:-4]+"_animated.svg")
+      else:
+        st.write('This SVG cannot be animated due to limitations of the used DeepSVG library. Please use another file.')