Update app.py

app.py

@@ -411,8 +411,9 @@ def mm(graph,zoom):
         code=ea.strip().strip("\n")
         code_out+=code
     #out_html=f'''<div><iframe src="https://omnibus-mermaid-script.static.hf.space/index.html?mermaid={code_out}&rand={random.randint(1,1111111111)}" height="500" width="500"></iframe></div>'''
-    out_html=f'''<div id="wrap" style="width: 100%; height: 100%;max-height:600px; padding: 0; overflow: auto;"><iframe id="frame" src="https://omnibus-mermaid-script.static.hf.space/index.html?mermaid={code_out}" style="width:100%; height:600px; border: 1px solid black; zoom: {str(zoom)}; -moz-transform: scale({str(zoom)}); -moz-transform-origin: 0 0;"></iframe></div>'''
-    return out_html
+    url=f"https://omnibus-mermaid-script.static.hf.space/index.html?mermaid={code_out}"
+    out_html=f'''<div id="wrap" style="width: 100%; height: 100%;max-height:600px; padding: 0; overflow: auto;"><iframe id="frame" src="{url}" style="width:100%; height:600px; border: 1px solid black; zoom: {str(zoom)}; -moz-transform: scale({str(zoom)}); -moz-transform-origin: 0 0;"></iframe></div>'''
+    return out_html,url
     
 def summarize(inp,history,data=None,files=None,directory=None,url=None,pdf_url=None,pdf_batch=None):
     json_box=[]
@@ -538,8 +539,24 @@ def summarize(inp,history,data=None,files=None,directory=None,url=None,pdf_url=N
 def clear_fn():
     return "",[(None,None)]
 
-
-
+def create_image(url):
+    source = requests.get(url)
+    #source = urllib.request.urlopen(url).read()
+    soup = bs4.BeautifulSoup(source.content,'lxml')
+
+    #rawp=(f'RAW TEXT RETURNED: {soup.text}')
+    #cnt=0
+    #cnt+=len(rawp)
+    #out.append(rawp)
+    #out.append("HTML fragments: ")
+    #q=("a","p","span","content","article")
+    out=[]
+    for p in soup.find_all("svg"):
+        print(p)
+        out.append(p.string) 
+    with open("tmp.svg","w") as svg:
+        svg.write(out[0])
+    return "tmp.svg"
 def zoom_update(inp):
     this.clear()
     this.append(str(inp))
@@ -575,12 +592,18 @@ with gr.Blocks() as app:
     m_box=gr.HTML()
     zoom_btn=gr.Slider(label="Zoom",step=0.01,minimum=0.1,maximum=20,value=1,interactive=True)
     e_box=gr.Textbox(interactive=True)
-    upd_button=gr.Button("Update Chart")
+    with gr.Row():
+        upd_button=gr.Button("Update Chart")
+        create_im=gr.Button("Create Image")
+    svg_img=gr.Image()
+    url_box=gr.Textbox()
+
     json_out=gr.JSON()
     #text=gr.JSON()
 
+    create_im.click(create_image,url_box,svg_img)
     #zoom_btn.change(zoom_update,zoom_btn,None)
-    upd_button.click(mm,[e_box,zoom_btn],[m_box])
+    upd_button.click(mm,[e_box,zoom_btn],[m_box,url_box])
     #inp_query.change(search_models,inp_query,models_dd)
     clear_btn.click(clear_fn,None,[prompt,chatbot])