deathprocess.py

import streamlit as st
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import networkx as nx

# Models
def deathProcess(death_rate,total_time,population):
    time=0
    times=[time]
    populations=[population]
    while time<total_time and population>0:
        time_to_next_birth = np.random.exponential(1 / death_rate)
        time += time_to_next_birth
        if time<total_time:
            population-=1
            times.append(time)
            populations.append(population)
    return times,populations

# Visualization
def visualize_death(times, populations):
    plt.figure(figsize=(10, 5))
    plt.step(times, populations, where='post')
    plt.title('Pure Death Process Simulation')
    plt.xlabel('Time')
    plt.ylabel('Population Size')
    plt.grid()
    st.pyplot(plt)

# Diagram
def plot_state_transition_diagram_death(max_population, death_rate):
    G = nx.DiGraph()
    
    for i in range(max_population + 1):
        G.add_node(i)
        if i > 0:  
            G.add_edge(i, i - 1) 

    pos = nx.spring_layout(G)  
    plt.figure(figsize=(10, 8))
    nx.draw(G, pos, with_labels=True, node_size=1800, node_color='lightblue', font_size=12)
    
    edge_labels = {(i, i - 1): f' µ={death_rate}' for i in range(1, max_population + 1)}
    nx.draw_networkx_edge_labels(G, pos, edge_labels=edge_labels, font_color='red', font_size=10)
    
    plt.title("State Transition Diagram for Pure Death Process")
    plt.xlabel("Population Size")
    plt.ylabel("Next Population Size")
    plt.grid()
    st.pyplot(plt)

# Statistics
def display_statistics(populations):
    # Calculate statistics
    final_population = populations[-1]
    mean_population = np.mean(populations)
    std_population = np.std(populations)
    
    # Create a DataFrame for more statistics if needed
    df = pd.DataFrame({'Time': range(len(populations)), 'Population': populations})
    
    # Display statistics in Streamlit
    st.write("### Statistics")
    st.write(f"Final Population: {final_population}")
    st.write(f"Mean Population: {mean_population:.2f}")
    st.write(f"Standard Deviation of Population: {std_population:.2f}")

    # Plot population over time
    plt.figure(figsize=(10, 5))
    plt.plot(df['Time'], df['Population'], marker='o', linestyle='-', color='blue')
    plt.title("Population Over Time")
    plt.xlabel("Time")
    plt.ylabel("Population")
    plt.grid()
    st.pyplot(plt)