Update app.py

app.py

@@ -18,6 +18,8 @@ import plotly.graph_objs as go
 import pandas as pd
 import plotly.offline as pyo
 
+from io import StringIO
+
 #@st.cache_resource()
 @st.cache()
 def get_nl():
@@ -278,17 +280,29 @@ if len(uploaded_file)>0:
 
             fig.update_layout(height=700, showlegend=False, title={'text': f"<b>{wrapped_title} - Sentiment Analysis Report</b>", 'x': 0.5, 'xanchor': 'center','font': {'size': 32}})
 
-            pyo.plot(fig, filename='report.html')
+            #pyo.plot(fig, filename='report.html')
+
+            buffer = io.StringIO()
+            fig.write_html(buffer, include_plotlyjs='cdn')
+            html_bytes = buffer.getvalue().encode()
+    
+            st.download_button(
+                label='Download HTML',
+                data=html_bytes,
+                file_name='report.html',
+                mime='text/html'
+            )
+
 
-            import base64
+            # import base64
 
-            # Convert the figure to HTML format
-            fig_html = pio.to_html(fig, full_html=False)
-            b64 = base64.b64encode(fig_html.encode()).decode()
+            # # Convert the figure to HTML format
+            # fig_html = pio.to_html(fig, full_html=False)
+            # b64 = base64.b64encode(fig_html.encode()).decode()
 
-            # Generate a download link
-            filename = "figure.html"
-            href = f'<a href="data:file/html;base64,{b64}" download="{filename}">Download Report</a>'
+            # # Generate a download link
+            # filename = "figure.html"
+            # href = f'<a href="data:file/html;base64,{b64}" download="{filename}">Download Report</a>'
 
-            # Display the link
-            st.markdown(href, unsafe_allow_html=True)
+            # # Display the link
+            # st.markdown(href, unsafe_allow_html=True)