Create app.py

app.py

@@ -0,0 +1,137 @@
+import streamlit as st
+import time
+import plotly.graph_objects as go
+
+# Set page layout to wide mode
+st.set_page_config(layout="wide")
+
+# Treatment outcomes and possibilities
+outcomes = [
+    ("Doctor prescribes Treatment 1 (Personalized Drug A)", "info"),
+    ("Nurse monitors Patient - Slight improvement", "success"),
+    ("Clinician evaluates lab results - Treatment seems effective", "info"),
+    ("Doctor prescribes Treatment 2 (Personalized Drug B)", "warning"),
+    ("Nurse monitors Patient - Side effects observed", "error"),
+    ("Clinician suggests modifying dosage", "warning"),
+    ("Doctor prescribes modified Treatment 2", "info"),
+    ("Patient shows significant improvement", "success")
+]
+
+# Different possible outcomes of each treatment
+possibilities = {
+    1: [
+        ("Possibility 1: Patient responds well to Treatment 1", "success"),
+        ("Possibility 2: Slight improvement, but inconclusive results", "info"),
+        ("Possibility 3: No response, reevaluation needed", "error")
+    ],
+    2: [
+        ("Possibility 1: Significant improvement", "success"),
+        ("Possibility 2: Mild side effects", "warning")
+    ],
+    3: [
+        ("Possibility 1: Treatment is effective", "success"),
+        ("Possibility 2: Inconclusive lab results", "info")
+    ],
+    4: [
+        ("Possibility 1: Improvement with Drug B", "success"),
+        ("Possibility 2: Significant side effects", "error")
+    ],
+    5: [
+        ("Possibility 1: Side effects worsen, modify dosage", "warning"),
+        ("Possibility 2: Manageable side effects", "info")
+    ],
+    6: [
+        ("Possibility 1: Dosage adjustment successful", "success"),
+        ("Possibility 2: Further modification needed", "warning")
+    ],
+    7: [
+        ("Possibility 1: Patient responds well to modified treatment", "success"),
+        ("Possibility 2: Limited response, consider alternatives", "warning")
+    ],
+    8: [
+        ("Possibility 1: Complete recovery", "success"),
+        ("Possibility 2: Partial improvement, continue monitoring", "info")
+    ]
+}
+
+# Create a tree diagram using Plotly with animations
+def create_tree_diagram(stage):
+    labels = [f"Stage {stage}: {outcomes[stage - 1][0]}"]
+    parents = [""]  # Root node
+    values = [1]  # Root node value
+    colors = ['lightgrey']  # Root node color
+    
+    stage_possibilities = possibilities.get(stage, [("No specific possibilities defined", "info")])
+    
+    for possibility, outcome_type in stage_possibilities:
+        labels.append(possibility)
+        parents.append(f"Stage {stage}: {outcomes[stage - 1][0]}")
+        values.append(1)  # Equal weight for all possibilities
+        if outcome_type == "success":
+            colors.append('#d4edda')
+        elif outcome_type == "info":
+            colors.append('#cce5ff')
+        elif outcome_type == "warning":
+            colors.append('#fff3cd')
+        elif outcome_type == "error":
+            colors.append('#f8d7da')
+        else:
+            colors.append('lightgrey')
+    
+    fig = go.Figure(go.Treemap(
+        labels=labels,
+        parents=parents,
+        values=values,
+        marker=dict(colors=colors),
+        textinfo="label",
+        textfont=dict(size=14),
+        hoverinfo="label+value+percent entry"
+    ))
+
+    fig.update_layout(
+        margin=dict(t=10, l=10, r=10, b=10),
+        width=600, height=400,
+        uniformtext=dict(minsize=12, mode='show'),
+        transition_duration=500  # Animation speed
+    )
+    
+    return fig
+
+# Streamlit app layout
+st.title("Precision Medicine AI Agents - Treatment Decision Tree with Animations")
+
+# Add a start animation button
+start_button = st.button("Start Animation")
+
+if start_button:
+    for i, (outcome, outcome_type) in enumerate(outcomes, 1):
+        with st.container():
+            col1, col2 = st.columns([1, 2])
+
+            with col1:
+                if outcome_type == "success":
+                    st.markdown(f"<div style='padding:10px; border-radius:5px; background-color:#d4edda; color:#155724;'><strong>Stage {i}:</strong> {outcome}</div>", unsafe_allow_html=True)
+                elif outcome_type == "info":
+                    st.markdown(f"<div style='padding:10px; border-radius:5px; background-color:#cce5ff; color:#004085;'><strong>Stage {i}:</strong> {outcome}</div>", unsafe_allow_html=True)
+                elif outcome_type == "warning":
+                    st.markdown(f"<div style='padding:10px; border-radius:5px; background-color:#fff3cd; color:#856404;'><strong>Stage {i}:</strong> {outcome}</div>", unsafe_allow_html=True)
+                elif outcome_type == "error":
+                    st.markdown(f"<div style='padding:10px; border-radius:5px; background-color:#f8d7da; color:#721c24;'><strong>Stage {i}:</strong> {outcome}</div>", unsafe_allow_html=True)
+
+                st.markdown("### Possibilities:")
+                stage_possibilities = possibilities.get(i, [("No specific possibilities defined", "info")])
+                for possibility, possibility_type in stage_possibilities:
+                    if possibility_type == "success":
+                        st.markdown(f"<div style='padding:5px; background-color:#d4edda; color:#155724;'><strong>{possibility}</strong></div>", unsafe_allow_html=True)
+                    elif possibility_type == "info":
+                        st.markdown(f"<div style='padding:5px; background-color:#cce5ff; color:#004085;'><strong>{possibility}</strong></div>", unsafe_allow_html=True)
+                    elif possibility_type == "warning":
+                        st.markdown(f"<div style='padding:5px; background-color:#fff3cd; color:#856404;'><strong>{possibility}</strong></div>", unsafe_allow_html=True)
+                    elif possibility_type == "error":
+                        st.markdown(f"<div style='padding:5px; background-color:#f8d7da; color:#721c24;'><strong>{possibility}</strong></div>", unsafe_allow_html=True)
+
+            with col2:
+                fig = create_tree_diagram(i)
+                st.plotly_chart(fig, use_container_width=True)
+
+        time.sleep(3)  # Shorter delay for smoother transitions