Update app.py

app.py

@@ -551,8 +551,9 @@ def create_image(url):
     #out.append("HTML fragments: ")
     #q=("a","p","span","content","article")
     out=[]
-    for p in soup.find_all("pre, {'class': 'mermaid'}"):
-        out.append(p.find('svg'))
+    for b in soup.find_all("div"):
+        for p in b.find_all("pre, {'class': 'mermaid'}"):
+            out.append(p.find('svg'))
     print(out)
     #out.append(p.string) 
     with open("tmp.svg","w") as svg: