update S3 bucket

app.py

@@ -7,6 +7,7 @@ import requests
 from datetime import datetime
 import time
 import os
+import boto3
 
 
 ### Config
@@ -16,30 +17,54 @@ st.set_page_config(
     layout="wide"
 )
 
+
+
+# Initialize AWS session with credentials from Hugging Face secrets
+session = boto3.Session(
+    aws_access_key_id=st.secrets["AWS_ACCESS_KEY_ID"],
+    aws_secret_access_key=st.secrets["AWS_SECRET_ACCESS_KEY"],
+)
+
+# Initialize S3 resource
+s3 = session.resource("s3")
+bucket_name = 'llepogam-app-history'
+bucket = s3.Bucket(bucket_name)
+
+
 # File path for history
 HISTORY_FILE = "https://llepogam-app-history.s3.eu-north-1.amazonaws.com/history.csv"
 
-# Initialize session state and load history
-if 'history' not in st.session_state:
-    if os.path.exists(HISTORY_FILE):
-        try:
-            history_df = pd.read_csv(HISTORY_FILE)
-            st.session_state.history = history_df.to_dict('records')
-        except Exception as e:
-            st.error(f"Error loading history: {str(e)}")
-            st.session_state.history = []
-    else:
-        st.session_state.history = []
-
-    
+   
 
 def save_history():
-    """Save history to CSV file"""
+    """Save history to S3"""
     try:
         history_df = pd.DataFrame(st.session_state.history)
-        history_df.to_csv(HISTORY_FILE, index=False)
+        # Save to temporary file first
+        history_df.to_csv("/tmp/temp_history.csv", index=False)
+        # Upload to S3
+        bucket.upload_file("/tmp/temp_history.csv", "history.csv")
+        # Clean up temp file
+        os.remove("/tmp/temp_history.csv")
     except Exception as e:
-        st.error(f"Error saving history: {str(e)}")
+        st.error(f"Error saving history to S3: {str(e)}")
+
+def load_history():
+    """Load history from S3"""
+    try:
+        # Download from S3 to temporary file
+        bucket.download_file("history.csv", "/tmp/temp_history.csv")
+        # Read the CSV
+        history_df = pd.read_csv("/tmp/temp_history.csv")
+        # Clean up temp file
+        os.remove("/tmp/temp_history.csv")
+        return history_df.to_dict('records')
+    except Exception as e:
+        st.error(f"Error loading history from S3: {str(e)}")
+        return []
+    
+if 'history' not in st.session_state:
+    st.session_state.history = load_history()
 
 # Custom CSS
 st.markdown("""