Update app.py

app.py

@@ -3,11 +3,12 @@ import speedtest
 import time
 import pandas as pd
 from datetime import datetime
+import requests
 
-# UI setup
+# ========== PAGE SETUP ==========
 st.set_page_config(page_title="Internet Speed Test", layout="centered")
 
-# Custom dark mode styling
+# Custom dark theme
 st.markdown("""
     <style>
     body { background-color: #111827; color: #f9fafb; }
@@ -16,19 +17,32 @@ st.markdown("""
     """, unsafe_allow_html=True)
 
 st.title("📶 Internet Speed Test")
-st.caption("Test your download/upload speed with history and auto-retest option.")
 
-# Session state to store history & timing
+# ========== GET PUBLIC IP ==========
+def get_public_ip():
+    try:
+        response = requests.get("https://api.ipify.org?format=json", timeout=5)
+        if response.status_code == 200:
+            return response.json()["ip"]
+        else:
+            return "Unavailable"
+    except Exception:
+        return "Unavailable"
+
+public_ip = get_public_ip()
+st.markdown(f"🌍 **Your Public IP:** `{public_ip}`")
+
+# ========== SESSION SETUP ==========
 if "history" not in st.session_state:
     st.session_state.history = []
 if "last_test" not in st.session_state:
     st.session_state.last_test = 0
 
-# Set retest interval (in minutes)
+# ========== RETEST INTERVAL ==========
 interval = st.selectbox("⏱️ Retest every (minutes)", [0, 1, 5, 10, 15, 30, 60], index=0)
 auto_retest = interval > 0
 
-# Perform test
+# ========== SPEED TEST FUNCTION ==========
 def run_speed_test():
     stt = speedtest.Speedtest()
     stt.get_best_server()
@@ -57,7 +71,7 @@ def run_speed_test():
     st.markdown(f"**📤 Upload  :** `{upload_mbps} Mbps`")
     st.markdown(f"**⏱️ Time    :** `{test_time_sec} sec`")
 
-# Manual test
+# ========== MANUAL TEST BUTTON ==========
 if st.button("▶️ Start Speed Test"):
     with st.spinner("Running test... please wait..."):
         try:
@@ -65,7 +79,7 @@ if st.button("▶️ Start Speed Test"):
         except Exception as e:
             st.error(f"❌ Speed test failed: {e}")
 
-# Auto retest logic (based on interval)
+# ========== AUTO RETEST ==========
 if auto_retest:
     now = time.time()
     if now - st.session_state.last_test > interval * 60:
@@ -75,7 +89,7 @@ if auto_retest:
             except Exception as e:
                 st.error(f"❌ Auto test failed: {e}")
 
-# Display history if available
+# ========== SHOW HISTORY ==========
 if st.session_state.history:
     st.markdown("---")
     st.subheader("📊 Test History")
@@ -84,4 +98,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)