Update app.py

app.py

@@ -3,74 +3,72 @@ import streamlit as st
 import requests
 import pandas as pd
 
-# ------------------ Page Setup ------------------
 st.set_page_config(page_title="🛒 SuperKart Sales Forecast", layout="centered")
 
 st.title("SuperKart Sales Forecast")
 st.caption("Frontend powered by Streamlit → calls Flask backend for predictions")
 
-# ------------------ Backend URL Resolution ------------------
-def resolve_backend_url() -> str:
-    # 1) HF Variables/Secrets are exposed as env vars
+# -----------------------------
+# Backend URL Resolver
+# -----------------------------
+def resolve_backend_url():
     env_url = os.getenv("BACKEND_URL")
     if env_url:
         return env_url.strip()
-
-    # 2) Optional: Streamlit secrets (if .streamlit/secrets.toml exists)
     try:
-        return st.secrets["BACKEND_URL"].strip()  # will raise if no secrets
+        return st.secrets["BACKEND_URL"].strip()
     except Exception:
         pass
-
-    # 3) Fallback placeholder (change this to your real backend if you prefer)
-    return "https://<your-backend-space>.hf.space"
+    return "https://rizwan9-backend.hf.space"  # fallback if not set
 
 BACKEND_URL = resolve_backend_url()
 
-# ------------------ Sidebar: Backend Controls ------------------
+# -----------------------------
+# Sidebar: Backend Health
+# -----------------------------
 st.sidebar.title("⚙️ Backend")
-# Clickable link to backend root
+
+default_url = BACKEND_URL
+backend_url_input = st.sidebar.text_input("Backend URL", value=default_url)
+BACKEND_URL = backend_url_input.strip() or default_url
 st.sidebar.markdown(f"**URL:** [{BACKEND_URL}]({BACKEND_URL})")
 
-# Cache health check result briefly to avoid spamming
-@st.cache_data(ttl=60)
-def check_backend_health(url: str):
+@st.cache_data(ttl=60, show_spinner=False)
+def check_backend_health(url: str, timeout: int = 45):
     try:
-        r = requests.get(f"{url}/health", timeout=20)
+        r = requests.get(f"{url}/health", timeout=timeout)
         return r.status_code, r.text
     except Exception as e:
         return None, str(e)
 
-# Auto-check toggle
 auto_check = st.sidebar.toggle("Auto check health on load", value=True)
-
 status_area = st.sidebar.empty()
+
 if auto_check:
-    code, msg = check_backend_health(BACKEND_URL)
+    with status_area, st.spinner("Checking backend health..."):
+        code, msg = check_backend_health(BACKEND_URL)
     if code == 200:
-        status_area.success(f"✅ Healthy (200)")
+        status_area.success("✅ Healthy (200)")
     elif code is None:
         status_area.error(f"❌ Unreachable\n{msg}")
     else:
         status_area.warning(f"⚠️ Status {code}\n{msg}")
 
-# Manual check button
 if st.sidebar.button("🔍 Check Health Now"):
-    code, msg = check_backend_health.clear()(BACKEND_URL)  # clear cache for fresh check
+    check_backend_health.clear()
+    with status_area, st.spinner("Re-checking..."):
+        code, msg = check_backend_health(BACKEND_URL)
     if code == 200:
-        st.sidebar.success(f"✅ Healthy (200)")
+        status_area.success("✅ Healthy (200)")
     elif code is None:
-        st.sidebar.error(f"❌ Unreachable\n{msg}")
+        status_area.error(f"❌ Unreachable\n{msg}")
     else:
-        st.sidebar.warning(f"⚠️ Status {code}\n{msg}")
-
-# Config guidance/warning
-if "<your-backend-space>" in BACKEND_URL:
-    st.sidebar.warning("Set BACKEND_URL in Space settings → Variables and secrets.")
+        status_area.warning(f"⚠️ Status {code}\n{msg}")
 
+# -----------------------------
+# Main Form
+# -----------------------------
 st.divider()
-
-# ------------------ Form: Inputs ------------------
 st.subheader("Enter Product and Store Details")
 
 with st.form("input_form"):
@@ -93,7 +91,9 @@ with st.form("input_form"):
 
     submitted = st.form_submit_button("🚀 Predict Sales")
 
-# ------------------ Prediction Call ------------------
+# -----------------------------
+# Prediction
+# -----------------------------
 if submitted:
     payload = {
         "Product_Weight": Product_Weight,
@@ -106,10 +106,13 @@ if submitted:
         "Store_Location_City_Type": Store_Location_City_Type,
         "Store_Type": Store_Type
     }
+
     try:
-        r = requests.post(f"{BACKEND_URL}/predict", json=payload, timeout=30)
-        r.raise_for_status()
-        prediction = r.json()["predictions"][0]
+        with st.spinner("Fetching prediction from backend..."):
+            r = requests.post(f"{BACKEND_URL}/predict", json=payload, timeout=60)
+            r.raise_for_status()
+            prediction = r.json()["predictions"][0]
         st.success(f"📈 Predicted Product Store Sales Total: **{prediction:,.2f}**")
     except Exception as e:
         st.error(f"❌ Prediction failed:\n\n{e}")
+