Update app.py

app.py

@@ -9,27 +9,6 @@ from streamlit_flow import streamlit_flow
 from streamlit_flow.elements import StreamlitFlowNode, StreamlitFlowEdge
 from streamlit_flow.layouts import TreeLayout
 
-# 1. Configuration
-Site_Name = '🦁CatRider🐈'
-title="🦁CatRider🐈by👤Aaron Wacker"
-helpURL='https://huggingface.co/awacke1'
-bugURL='https://huggingface.co/spaces/awacke1'
-icons='🦁'
-
-useConfig=True
-if useConfig: # Component code - useConfig=False should allow it to work if re-evaluating UI elements due to code modify
-    st.set_page_config(
-        page_title=title,
-        page_icon=icons,
-        layout="wide",
-        #initial_sidebar_state="expanded",
-        initial_sidebar_state="auto",
-        menu_items={
-            'Get Help': helpURL,
-            'Report a bug': bugURL,
-            'About': title
-        }
-    )
 # 🐱 Cat Rider and Gear Data
 CAT_RIDERS = [
     {"name": "Whiskers", "type": "Speed", "emoji": "🐾", "strength": 3, "skill": 7},
@@ -110,7 +89,7 @@ FAILURE_CONCLUSIONS = [
 def save_history_to_file(history_df, user_id):
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     filename = f"history_{user_id}_{timestamp}.md"
-    markdown = create_markdown_preview(history_df)
+    markdown, _, _ = process_journey_history(history_df)
     with open(filename, 'w', encoding='utf-8') as f:
         f.write(markdown)
     return filename
@@ -167,75 +146,74 @@ def update_character_stats(game_state, outcome):
         game_state['gear_strength'] = max(1, game_state['gear_strength'] - 0.1)
     return game_state
 
-# 🌳 Journey Visualization
-def create_knowledge_graph(history_df):
+# 🌳 Process Journey History to Create Markdown and Graph Data
+def process_journey_history(history_df):
+    markdown = "## 🌳 Journey Preview\n\n"
     nodes = []
     edges = []
     node_ids = {}
     score = 0  # To keep track of the score for success nodes
 
-    for index, row in history_df.iterrows():
-        situation_node_id = f"situation_{row['situation_id']}"
-        attempt_node_id = f"attempt_{index}"
+    grouped = history_df.groupby(['situation_name'], sort=False)
+
+    for situation_name, group in grouped:
+        situation_emoji = group.iloc[0]['situation_emoji']
+        situation_id = group.iloc[0]['situation_id']
+        situation_node_id = f"situation_{situation_id}"
 
-        # Situation node
+        markdown += f"### {situation_emoji} **{situation_name}**\n"
+
+        # Create situation node if not already created
         if situation_node_id not in node_ids:
             situation_node = StreamlitFlowNode(
                 situation_node_id,
                 pos=(0, 0),
-                data={'content': f"{row['situation_emoji']} {row['situation_name']}"},
+                data={'content': f"{situation_emoji} {situation_name}"},
                 type='default',
                 shape='ellipse'
             )
             nodes.append(situation_node)
             node_ids[situation_node_id] = situation_node_id
 
-        # Update score if outcome is success
-        if row['outcome']:
-            score = row['score']
-
-        # Attempt node
-        outcome_content = '✅ Success' if row['outcome'] else '❌ Failure'
-        stars = '⭐' * int(row['score'])
-        if row['outcome']:
-            # For success nodes, show current score and journey icons
-            attempt_content = f"Attempt {row['attempt']}: {row['action_emoji']} {row['action_name']} - {outcome_content}\n{stars}"
-        else:
-            attempt_content = f"Attempt {row['attempt']}: {row['action_emoji']} {row['action_name']} - {outcome_content}"
-
-        attempt_node = StreamlitFlowNode(
-            attempt_node_id,
-            pos=(0, 0),
-            data={'content': attempt_content},
-            type='default',
-            shape='rectangle'
-        )
-        nodes.append(attempt_node)
-
-        # Edge from situation to attempt
-        edges.append(StreamlitFlowEdge(
-            id=f"edge_{situation_node_id}_{attempt_node_id}",
-            source=situation_node_id,
-            target=attempt_node_id,
-            animated=True
-        ))
-
-    return nodes, edges
-
-# 📝 Markdown Preview with Subpoints for Each Action
-def create_markdown_preview(history_df):
-    markdown = "## 🌳 Journey Preview\n\n"
-    grouped = history_df.groupby(['situation_name'], sort=False)
-
-    for situation_name, group in grouped:
-        markdown += f"### {group.iloc[0]['situation_emoji']} **{situation_name}**\n"
         for _, row in group.iterrows():
-            outcome_str = '✅ Success' if row['outcome'] else '❌ Failure'
-            stars = '⭐' * int(row['score']) if row['outcome'] else ''
-            markdown += f"Attempt {row['attempt']}: {row['action_emoji']} **{row['action_name']}**: {outcome_str} {stars}\n"
-            markdown += f"{row['conclusion']}\n"
+            attempt = row['attempt']
+            action_emoji = row['action_emoji']
+            action_name = row['action_name']
+            outcome = row['outcome']
+            outcome_str = '✅ Success' if outcome else '❌ Failure'
+            stars = '⭐' * int(row['score']) if outcome else ''
+            conclusion = row['conclusion']
+
+            markdown += f"Attempt {attempt}: {action_emoji} **{action_name}**: {outcome_str} {stars}\n"
+            markdown += f"{conclusion}\n"
+
+            # Create attempt node
+            attempt_node_id = f"attempt_{situation_id}_{attempt}_{row['action_id']}"
+            if outcome:
+                attempt_content = f"Attempt {attempt}: {action_emoji} {action_name} - {outcome_str}\n{stars}"
+            else:
+                attempt_content = f"Attempt {attempt}: {action_emoji} {action_name} - {outcome_str}"
+
+            attempt_node = StreamlitFlowNode(
+                attempt_node_id,
+                pos=(0, 0),
+                data={'content': attempt_content},
+                type='default',
+                shape='rectangle'
+            )
+            nodes.append(attempt_node)
+
+            # Edge from situation to attempt
+            edges.append(StreamlitFlowEdge(
+                id=f"edge_{situation_node_id}_{attempt_node_id}",
+                source=situation_node_id,
+                target=attempt_node_id,
+                animated=True
+            ))
+
         markdown += "\n"
-    return markdown
+
+    return markdown, nodes, edges
 
 # 🔄 Update game state with the result of the action
 def update_game_state(game_state, situation, action, outcome, timestamp):
@@ -441,12 +419,14 @@ def main():
 
         # Integration point for both functions
         if not game_state['history_df'].empty:
+            # 📝 Process Journey History to get markdown and graph data
+            markdown_preview, nodes, edges = process_journey_history(game_state['history_df'])
+
             # 📝 Display Markdown Preview
-            st.markdown(create_markdown_preview(game_state['history_df']))
+            st.markdown(markdown_preview)
 
             # 🌳 Display Knowledge Journey Graph
             st.markdown("## 🌳 Your Journey (Knowledge Graph)")
-            nodes, edges = create_knowledge_graph(game_state['history_df'])
             try:
                 streamlit_flow('cat_rider_flow',
                                nodes,