Update app.py

app.py

@@ -1,52 +1,52 @@
 import streamlit as st
 from datasets import load_dataset
 import streamlit.components.v1 as components
+import requests
 
 # Load the dataset
 dataset = load_dataset("awacke1/DatasetOfDatasetsUSA")
 
-# Initialize session state for record navigation
+# Initialize session state for record navigation and autoplay
 if 'index' not in st.session_state:
     st.session_state.index = 0
+if 'autoplay' not in st.session_state:
+    st.session_state.autoplay = False
 
 # Define the maximum index as the length of the dataset - 1
 max_index = len(dataset['train']) - 1
 
-# Navigation buttons
-col1, col2, col3, col4, col5 = st.columns(5)
-with col1:
-    if st.button('⏮️'):
-        st.session_state.index = 0
-with col2:
-    if st.button('◀️') and st.session_state.index > 0:
-        st.session_state.index -= 1
-with col3:
-    st.write(f"Record {st.session_state.index + 1} of {max_index + 1}")
-with col4:
-    if st.button('▶️') and st.session_state.index < max_index:
+def advance_record():
+    if st.session_state.index < max_index:
         st.session_state.index += 1
-with col5:
-    if st.button('⏭️'):
-        st.session_state.index = max_index
-
-# Generate the A-Frame scene HTML with links from the dataset
-aframe_html = """
-<a-scene embedded style="width: 800px; height: 600px;">
-  <a-sky color="#ECECEC"></a-sky>
-"""
-
-# Loop through the dataset and create a-link entities for each record
-for index, item in enumerate(dataset['train']):
-    cityOrState = item['cityOrState']
-    link = item['link']
-    linkType = item['linkType']
-    position = f"{index * 1.5} 1.25 -3"  # Simple positioning logic
-    aframe_html += f"""
-    <a-link href="{link}" position="{position}" title="{cityOrState} - {linkType}" image="#homeThumbnail"></a-link>
-    """
-
-# Close the A-Scene tag
-aframe_html += "</a-scene>"
-
-# Use Streamlit components to render the A-Frame HTML
-components.html(aframe_html, height=600)
+    else:
+        st.session_state.autoplay = False  # Stop when reaching the end of the dataset
+
+# Autoplay control
+col1, col2 = st.columns([1, 10])
+with col1:
+    if st.button('▶️ Play'):
+        st.session_state.autoplay = not st.session_state.autoplay
+        if st.session_state.autoplay:
+            st.experimental_rerun()
+
+# If autoplay is enabled, advance the record every second
+if st.session_state.autoplay:
+    st.session_state.time = st.session_state.get('time', 0) + 1
+    st.write(f"Autoplaying... Record {st.session_state.index + 1} of {max_index + 1}")
+    st.experimental_rerun()
+    advance_record()
+
+item = dataset['train'][st.session_state.index]
+link = item['link']
+
+# Fetch and display the HTML content
+try:
+    response = requests.get(link)
+    if response.status_code == 200:
+        components.html(response.text, height=600, scrolling=True)
+    else:
+        st.error(f"Failed to load content from {link}")
+except Exception as e:
+    st.error(f"An error occurred: {e}")
+
+# Note: Adjust the delay and autoplay logic as needed