.

app.py

@@ -1,5 +1,13 @@
 import streamlit as st
 
+from ase.visualize import view
+
+try:
+    from StringIO import StringIO
+except ImportError:
+    from io import StringIO
+
+import streamlit.components.v1 as components
 from scipy.spatial.distance import cdist
 
 import ase
@@ -136,37 +144,12 @@ mono = dcm1results["mono"]
 atoms = dcm1results["atoms"]
 dcmol = dcm1results["dcmol"]
 
-#st.write(dipo)
-#st.write(mono)
-#st.write(MessagePassingModel)
-#st.write(test_weights)
-#
-st.write(atoms)
-st.write(dcmol)
 
-from ase.visualize import view
 
-try:
-    from StringIO import StringIO
-except ImportError:
-    from io import StringIO
 
 output = StringIO()
-
 (atoms+dcmol).write(output, format="html")
-
 data = output.getvalue()
 
-print(data)
-
-#st.write(data)
-
-st.html(data)
-st.html(data.strip("\n"))
-
-import streamlit.components.v1 as components
-
-components.html("example")
-
-components.html(data)
+components.html(data, width=1000, height=1000)