Update app.py

app.py

@@ -2,8 +2,9 @@ import streamlit as st
 import asyncio
 import threading
 import random
-import matplotlib.pyplot as plt
-import numpy as np
+import time
+import psutil
+import plotly.graph_objects as go
 
 # Yoruba gods and their roles
 gods = {
@@ -19,23 +20,7 @@ gods = {
 
 # Story of Olorun and Yoruba gods
 story = """
-In the beginning, there was only Olorun, the supreme creator god, dwelling in the heavens above. 🌤️ Olorun, the almighty, decided to create the world and all that exists within it. He summoned Obatala, the god of creation, and tasked him with crafting the earth and all its inhabitants. 🌍
-
-Obatala descended from the heavens, carrying a snail shell filled with sacred sand, a chameleon, and a pigeon. 🐚🦎🕊️ As he reached the watery void below, Obatala released the sand, and the chameleon stepped upon it, spreading the grains far and wide. Thus, the earth was formed, and the pigeon flew across its expanse, bringing life and vitality to the land. 🌿
-
-With the earth now created, Olorun sent down the other gods to populate and govern various aspects of the world. Among them was Eshu, the trickster god of crossroads and fate. 🃏 Eshu, ever mischievous, would often play pranks on both mortals and gods alike, teaching valuable lessons through his cunning ways.
-
-Orunmila, the god of wisdom and divination, also descended to guide humanity with his prophetic insights. 🔮 He introduced the sacred divination system called Ifa, which allowed mortals to communicate with the gods and seek guidance in their lives.
-
-As the world flourished, Olorun watched from above, pleased with his creation. However, darkness crept in as Ikú, the god of death, began to claim the lives of mortals. 💀 Olorun, in his infinite wisdom, knew that life and death were part of the natural cycle, and he allowed Ikú to maintain the balance.
-
-Shango, the god of thunder and lightning, ruled the skies with his mighty axe. ⚡ His tempestuous nature mirrored the powerful storms that swept across the land, and mortals both feared and revered him.
-
-Oshun, the goddess of love, beauty, and fertility, graced the earth with her presence. 💛 She brought joy, passion, and abundance to the lives of mortals, and her followers celebrated her with vibrant festivities and offerings.
-
-Throughout the ages, the Yoruba gods continued to shape the world and guide humanity. They taught valuable lessons, bestowed blessings, and occasionally unleashed their wrath upon those who transgressed against them. 🌈
-
-The Yoruba mythology is a tapestry woven with tales of creation, destiny, and the eternal dance between the gods and mortals. These stories, passed down through generations, continue to inspire and teach the Yoruba people, reminding them of their rich heritage and the interconnectedness of all things in the cosmos. ✨
+...
 """
 
 # Streamlit app
@@ -43,51 +28,69 @@ def main():
     st.title("Yoruba Gods and Mythology")
     st.write(story)
 
-    st.sidebar.title("Agents")
-    agent_count = st.sidebar.slider("Number of Agents", min_value=1, max_value=8, value=8)
+    st.sidebar.title("Simulation")
+    timer_duration = st.sidebar.slider("Timer Duration (seconds)", min_value=1, max_value=60, value=60)
+    if st.sidebar.button("Start Simulation"):
+        simulate(timer_duration)
+
+def simulate(duration):
+    start_time = time.time()
+    end_time = start_time + duration
+
+    agents = []
+    for i in range(8):
+        agent = threading.Thread(target=agent_function, args=(i,))
+        agents.append(agent)
+        agent.start()
+
+    while time.time() < end_time:
+        remaining_time = int(end_time - time.time())
+        st.write(f"⏰ Time Remaining: {remaining_time} seconds")
 
-    if st.sidebar.button("Start Agents"):
-        with st.spinner("Spawning agents..."):
-            agents = []
-            for i in range(agent_count):
-                agent = threading.Thread(target=agent_function, args=(i,))
-                agents.append(agent)
-                agent.start()
+        for i, agent in enumerate(agents):
+            if agent.is_alive():
+                st.write(f"Agent {i+1}: {agent.message}")
+            else:
+                st.write(f"Agent {i+1}: Idle")
 
-            for agent in agents:
-                agent.join()
+        update_process_metrics()
+        time.sleep(1)
 
-        st.success("All agents have completed their tasks.")
+    for agent in agents:
+        agent.join()
 
-    st.sidebar.title("Heartbeat Graph")
-    if st.sidebar.button("Update Graph"):
-        update_heartbeat_graph()
+    st.success("Simulation completed.")
 
 def agent_function(agent_id):
+    thread = threading.current_thread()
     god_name = list(gods.keys())[agent_id]
     role = gods[god_name]
-    st.write(f"Agent {agent_id+1}: {god_name} - {role}")
+    thread.message = f"{god_name} - {role}"
     asyncio.run(simulate_task(agent_id))
 
 async def simulate_task(agent_id):
-    progress_bar = st.progress(0)
-    for i in range(100):
-        progress_bar.progress(i + 1)
-        await asyncio.sleep(random.uniform(0.01, 0.1))
-
-def update_heartbeat_graph():
-    active_threads = threading.active_count() - 1  # Subtract 1 for the main thread
-    ip_addresses = [f"192.168.0.{i+1}" for i in range(active_threads)]
-
-    fig, ax = plt.subplots()
-    ax.bar(range(active_threads), [1] * active_threads, color='blue')
-    ax.set_xlabel("User")
-    ax.set_ylabel("Heartbeat")
-    ax.set_title("Active Users")
-    ax.set_xticks(range(active_threads))
-    ax.set_xticklabels(ip_addresses)
-
-    st.pyplot(fig)
+    await asyncio.sleep(random.uniform(1, 5))
+
+def update_process_metrics():
+    thread_count = threading.active_count()
+    cpu_percent = psutil.cpu_percent()
+    mem_percent = psutil.virtual_memory().percent
+
+    fig = go.Figure(
+        data=[
+            go.Bar(name="Thread Count", x=["Metrics"], y=[thread_count]),
+            go.Bar(name="CPU Usage (%)", x=["Metrics"], y=[cpu_percent]),
+            go.Bar(name="Memory Usage (%)", x=["Metrics"], y=[mem_percent])
+        ],
+        layout=go.Layout(
+            title="Process Metrics",
+            xaxis=dict(title="Metrics"),
+            yaxis=dict(title="Value"),
+            barmode="group"
+        )
+    )
+
+    st.plotly_chart(fig)
 
 if __name__ == "__main__":
     main()