Update app.py

app.py

@@ -4,6 +4,7 @@ import plotly.express as px
 import random
 import uuid
 import os
+import re
 from datetime import datetime
 from streamlit_flow import streamlit_flow
 from streamlit_flow.elements import StreamlitFlowNode, StreamlitFlowEdge
@@ -94,10 +95,47 @@ def save_history_to_file(history_df, user_id):
         f.write(markdown)
     return filename
 
-# Function to load saved histories
-def load_saved_histories():
-    history_files = [f for f in os.listdir('.') if f.startswith('history_') and f.endswith('.md')]
-    return history_files
+# Function to load saved histories and reconstruct game state
+def load_saved_histories(user_id):
+    history_files = [f for f in os.listdir('.') if f.startswith(f'history_{user_id}_') and f.endswith('.md')]
+    history_files.sort()  # Ensure files are sorted by timestamp
+    combined_history = ''
+    for history_file in history_files:
+        with open(history_file, 'r', encoding='utf-8') as f:
+            combined_history += f.read() + '\n'
+    return combined_history
+
+# Function to parse markdown history and reconstruct history DataFrame
+def parse_history_markdown(markdown_text):
+    lines = markdown_text.split('\n')
+    data = []
+    current_situation = None
+    for line in lines:
+        situation_match = re.match(r"### (\S+) \*\*(.+)\*\*", line)
+        if situation_match:
+            situation_emoji = situation_match.group(1)
+            situation_name = situation_match.group(2)
+            continue
+        attempt_match = re.match(r"Attempt (\d+): (\S+) \*\*(.+)\*\*: (✅ Success|❌ Failure) (\S*)", line)
+        if attempt_match:
+            attempt = int(attempt_match.group(1))
+            action_emoji = attempt_match.group(2)
+            action_name = attempt_match.group(3)
+            outcome_str = attempt_match.group(4)
+            stars = attempt_match.group(5)
+            outcome = True if outcome_str == '✅ Success' else False
+            score = len(stars)
+            data.append({
+                'situation_name': situation_name,
+                'situation_emoji': situation_emoji,
+                'attempt': attempt,
+                'action_name': action_name,
+                'action_emoji': action_emoji,
+                'outcome': outcome,
+                'score': score
+            })
+    history_df = pd.DataFrame(data)
+    return history_df
 
 # 🧠 Game Mechanics
 def generate_situation():
@@ -158,7 +196,7 @@ def process_journey_history(history_df):
 
     for situation_name, group in grouped:
         situation_emoji = group.iloc[0]['situation_emoji']
-        situation_id = group.iloc[0]['situation_id']
+        situation_id = group.iloc[0].get('situation_id', uuid.uuid4())
         situation_node_id = f"situation_{situation_id}"
 
         markdown += f"### {situation_emoji} **{situation_name}**\n"
@@ -181,14 +219,14 @@ def process_journey_history(history_df):
             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']
+            stars = '⭐' * int(row.get('score', 0)) if outcome else ''
+            conclusion = row.get('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']}"
+            attempt_node_id = f"attempt_{situation_id}_{attempt}_{uuid.uuid4()}"
             if outcome:
                 attempt_content = f"Attempt {attempt}: {action_emoji} {action_name} - {outcome_str}\n{stars}"
             else:
@@ -259,6 +297,9 @@ def update_game_state(game_state, situation, action, outcome, timestamp):
     else:
         game_state['history'][action['id']] = 1 if outcome else -1
 
+    # Automatically save history after each action
+    save_history_to_file(game_state['history_df'], game_state['user_id'])
+
     return game_state
 
 # 🏅 Display Scoreboard with Star Emojis and Buckyball Outline
@@ -303,7 +344,7 @@ def main():
     # 🏁 Initialize game state
     if 'game_state' not in st.session_state:
         st.session_state.game_state = {
-            'user_id': str(uuid.uuid4()),
+            'user_id': None,
             'score': 0,
             'history': {},
             'gear_strength': 0,
@@ -317,38 +358,38 @@ def main():
             'succeeded': False
         }
 
-    # Sidebar functionality
-    st.sidebar.header("📂 Saved Histories")
-    history_files = load_saved_histories()
-    if history_files:
-        selected_history = st.sidebar.selectbox("Select a history to load:", history_files)
-        if st.sidebar.button("Load Selected History"):
-            with open(selected_history, 'r', encoding='utf-8') as f:
-                st.session_state.game_state['loaded_history'] = f.read()
-            st.sidebar.success(f"Loaded {selected_history}")
-
-    # 🐱 Cat Rider Selection
-    if st.session_state.game_state['cat_rider'] is None:
-        st.markdown("## Choose Your Cat Rider:")
+    game_state = st.session_state.game_state
+
+    # 🐱 Cat Rider Selection or Loading Previous State
+    if game_state['cat_rider'] is None:
+        st.markdown("## Choose Your Cat Rider or Load Previous Journey:")
+        # Check for existing histories
+        existing_riders = [f.split('_')[1] for f in os.listdir('.') if f.startswith('history_')]
+        existing_riders = list(set(existing_riders))
         cols = st.columns(len(CAT_RIDERS))
         for i, rider in enumerate(CAT_RIDERS):
             if cols[i].button(f"{rider['emoji']} {rider['name']} ({rider['type']})", key=f"rider_{i}"):
-                st.session_state.game_state['cat_rider'] = rider
-                st.session_state.game_state['rider_skill'] = rider['skill']
-                st.session_state.game_state['user_id'] = rider['name']  # Use rider name as user ID
+                game_state['cat_rider'] = rider
+                game_state['rider_skill'] = rider['skill']
+                game_state['user_id'] = rider['name']  # Use rider name as user ID
+
+                # Load existing history if available
+                existing_history = load_saved_histories(game_state['user_id'])
+                if existing_history:
+                    history_df = parse_history_markdown(existing_history)
+                    game_state['history_df'] = history_df
+                    game_state['score'] = history_df['score'].max() if not history_df.empty else 0
 
     # 🏇 Riding Gear Selection
-    if st.session_state.game_state['riding_gear'] is None and st.session_state.game_state['cat_rider'] is not None:
+    if game_state['riding_gear'] is None and game_state['cat_rider'] is not None:
         st.markdown("## Select Your Riding Gear:")
         cols = st.columns(len(RIDING_GEAR))
         for i, gear in enumerate(RIDING_GEAR):
             if cols[i].button(f"{gear['name']} ({gear['type']})", key=f"gear_{i}"):
-                st.session_state.game_state['riding_gear'] = gear
-                st.session_state.game_state['gear_strength'] = gear['strength']
+                game_state['riding_gear'] = gear
+                game_state['gear_strength'] = gear['strength']
 
     # 🎭 Game Loop
-    game_state = st.session_state.game_state
-
     if game_state['cat_rider'] is not None and game_state['riding_gear'] is not None:
         # Check if current_situation is None or if the player succeeded in the previous situation
         if game_state['current_situation'] is None or game_state['succeeded']:
@@ -445,23 +486,12 @@ def main():
             fig = px.bar(df, x='Stat', y='Value', title="Cat Rider Stats 📊")
             st.plotly_chart(fig)
 
-            # Save history to file
-            if st.button("💾 Save Game History"):
-                filename = save_history_to_file(game_state['history_df'], game_state['user_id'])
-                st.success(f"History saved to {filename}")
-                # Update history files list
-                history_files = load_saved_histories()
-
-            # Provide download buttons for saved histories
-            st.sidebar.markdown("### 📥 Download Histories")
-            for history_file in history_files:
-                with open(history_file, 'r', encoding='utf-8') as f:
-                    st.sidebar.download_button(
-                        label=f"Download {history_file}",
-                        data=f.read(),
-                        file_name=history_file,
-                        mime='text/markdown'
-                    )
+    # Automatically load saved history if available
+    if game_state['user_id'] and not game_state['history_df'].empty:
+        existing_history = load_saved_histories(game_state['user_id'])
+        if existing_history:
+            st.markdown("## 📂 Loaded Journey History")
+            st.markdown(existing_history)
 
 if __name__ == "__main__":
     main()