Update app.py

app.py

@@ -2,11 +2,12 @@ import streamlit as st
 import speedtest
 import time
 import pandas as pd
+from datetime import datetime
 
-# Configure page with dark theme styling
-st.set_page_config(page_title="Internet Speed Test", layout="centered", initial_sidebar_state="collapsed")
+# UI setup
+st.set_page_config(page_title="Internet Speed Test", layout="centered")
 
-# Custom CSS for dark mode (works even on Hugging Face)
+# Custom dark mode styling
 st.markdown("""
     <style>
     body { background-color: #111827; color: #f9fafb; }
@@ -15,50 +16,66 @@ st.markdown("""
     """, unsafe_allow_html=True)
 
 st.title("📶 Internet Speed Test")
-st.caption("Check your current internet speed and connection quality.")
+st.caption("Test your download/upload speed with history and auto-retest option.")
 
-# Store test history in session
+# Session state to store history & timing
 if "history" not in st.session_state:
-    st.session_state["history"] = []
+    st.session_state.history = []
+if "last_test" not in st.session_state:
+    st.session_state.last_test = 0
 
-# Start test button
+# Set retest interval (in minutes)
+interval = st.selectbox("⏱️ Retest every (minutes)", [0, 1, 5, 10, 15, 30, 60], index=0)
+auto_retest = interval > 0
+
+# Perform test
+def run_speed_test():
+    stt = speedtest.Speedtest()
+    stt.get_best_server()
+
+    start = time.time()
+    download = stt.download()
+    upload = stt.upload()
+    end = time.time()
+
+    download_mbps = round(download / 1_000_000, 2)
+    upload_mbps = round(upload / 1_000_000, 2)
+    test_time_sec = round(end - start, 2)
+
+    result = {
+        "Timestamp": datetime.now().strftime("%H:%M:%S"),
+        "Download (Mbps)": download_mbps,
+        "Upload (Mbps)": upload_mbps,
+        "Time (s)": test_time_sec,
+    }
+
+    st.session_state.history.append(result)
+    st.session_state.last_test = time.time()
+
+    st.success("✅ Test complete!")
+    st.markdown(f"**📥 Download:** `{download_mbps} Mbps`")
+    st.markdown(f"**📤 Upload  :** `{upload_mbps} Mbps`")
+    st.markdown(f"**⏱️ Time    :** `{test_time_sec} sec`")
+
+# Manual test
 if st.button("▶️ Start Speed Test"):
     with st.spinner("Running test... please wait..."):
         try:
-            stt = speedtest.Speedtest()
-            best = stt.get_best_server()
-
-            start = time.time()
-            download = stt.download()
-            upload = stt.upload()
-            end = time.time()
-
-            download_mbps = round(download / 1_000_000, 2)
-            upload_mbps = round(upload / 1_000_000, 2)
-            test_time_sec = round(end - start, 2)
-
-            # Add to history
-            result = {
-                "Download (Mbps)": download_mbps,
-                "Upload (Mbps)": upload_mbps,
-                "Time (s)": test_time_sec,
-                "ISP": best['sponsor'],
-                "Server": f"{best['name']}, {best['country']}"
-            }
-            st.session_state.history.append(result)
-
-            # Show results
-            st.success("✅ Test complete!")
-            st.markdown(f"**📥 Download:** `{download_mbps} Mbps`")
-            st.markdown(f"**📤 Upload  :** `{upload_mbps} Mbps`")
-            st.markdown(f"**⏱️ Time    :** `{test_time_sec} sec`")
-            st.markdown(f"**🏢 ISP     :** `{best['sponsor']}`")
-            st.markdown(f"**🌍 Server  :** `{best['name']}, {best['country']}`")
-
+            run_speed_test()
         except Exception as e:
             st.error(f"❌ Speed test failed: {e}")
 
-# Show speed test history as chart/table
+# Auto retest logic (based on interval)
+if auto_retest:
+    now = time.time()
+    if now - st.session_state.last_test > interval * 60:
+        with st.spinner(f"Auto-retesting every {interval} minutes..."):
+            try:
+                run_speed_test()
+            except Exception as e:
+                st.error(f"❌ Auto test failed: {e}")
+
+# Display history if available
 if st.session_state.history:
     st.markdown("---")
     st.subheader("📊 Test History")
@@ -67,4 +84,4 @@ if st.session_state.history:
     st.line_chart(df[["Download (Mbps)", "Upload (Mbps)"]])
 
     with st.expander("📋 See all results"):
-        st.dataframe(df, use_container_width=True)
+        st.dataframe(df, use_container_width=True)