done some error correction

pages/Continuous.py

@@ -2,7 +2,7 @@ import streamlit as st
 import torch
 import plotly.graph_objects as go
 from config.config_continuous import CONTINUOUS_DISTRIBUTIONS
-from utils import compute_pdf
+from utils import compute_pdf, plot_pdf
 
 st.title("Continuous Distributions")
 
@@ -14,7 +14,5 @@ support = support(params) if callable(support) else None
 
 x_range = torch.linspace(-10, 10, 1000)
 pdf = compute_pdf(dist, x_range, support)
+plot_pdf(pdf, x_range, selected_dist)
 
-fig = go.Figure()
-fig.add_trace(go.Scatter(x=x_range.numpy(), y=pdf.numpy(), mode="lines", name=selected_dist))
-st.plotly_chart(fig)

pages/Discrete.py

@@ -2,6 +2,7 @@ import streamlit as st
 import torch
 import plotly.graph_objects as go
 from config.config_discrete import DISCRETE_DISTRIBUTIONS
+from utils import compute_pdf, plot_pmf
 
 st.title("Discrete Distributions")
 
@@ -18,7 +19,4 @@ else:
     x_range = torch.arange(0, 20)
 
 pmf = dist.log_prob(x_range).exp()
-
-fig = go.Figure()
-fig.add_trace(go.Bar(x=x_range.numpy(), y=pmf.numpy(), name=selected_dist))
-st.plotly_chart(fig)
+plot_pmf(pmf, x_range, selected_dist)

utils.py

@@ -1,4 +1,6 @@
 import torch
+import streamlit as st
+import plotly.graph_objects as go
 
 def compute_pdf(dist, x_range, support):
     pdf = torch.zeros_like(x_range, dtype=torch.float)
@@ -12,4 +14,23 @@ def compute_pdf(dist, x_range, support):
         mask = torch.ones_like(x_range, dtype=torch.bool)
 
     pdf[mask] = dist.log_prob(x_range[mask]).exp()
-    return pdf
+    return pdf
+
+def plot_pdf(pdf, x_range, dist_name):
+    # Main plot area
+    st.markdown(f'#### Probability Density Function for {dist_name} Distribution')
+    fig = go.Figure()
+    fig.add_trace(go.Scatter(x=x_range, y=pdf, mode='lines',
+                            name='', hovertemplate='x: %{x:.2f}<br>f(x): %{y:.2f}'))
+    fig.update_layout(xaxis_title='x', yaxis_title='f(x)', showlegend=False)
+
+    st.plotly_chart(fig)
+    
+def plot_pmf(pmf, x_range, dist_name):
+    # Main plot area
+    st.markdown(f'#### Probability Mass Function for {dist_name} Distribution')
+    fig = go.Figure()
+    fig.add_trace(go.Bar(x=x_range.tolist(), y=pmf.tolist()))
+    fig.update_layout(xaxis_title='x', yaxis_title='P(x)', showlegend=False)
+
+    st.plotly_chart(fig)