Update src/frontend/app.py

src/frontend/app.py

@@ -13,6 +13,14 @@ import plotly.graph_objects as go
 from datetime import datetime
 import joblib
 
+st.set_page_config(
+    page_title="AI Prediction Dashboard",
+    page_icon="🎯",
+    layout="wide",
+    initial_sidebar_state="expanded"
+)
+
+# Function to start the API server
 def run_api():
     subprocess.Popen(["python", "-m", "uvicorn", "src.api.main:app", "--host", "0.0.0.0", "--port", "8000"])
 
@@ -21,9 +29,24 @@ if "api_started" not in st.session_state:
     threading.Thread(target=run_api).start()
     st.session_state.api_started = True
     st.success("Starting API... please wait a few seconds.")
+    
+    # Wait a few seconds for server to start
+    time.sleep(15)
+
+# Function to check if API is running
+def is_api_running():
+    try:
+        response = requests.get("http://localhost:8000/health")
+        return response.status_code == 200
+    except:
+        return False
 
-# Wait a few seconds for server to start
-time.sleep(15)
+# Check API status
+api_status = is_api_running()
+if api_status:
+    st.sidebar.success("✅ API is running")
+else:
+    st.sidebar.error("❌ API is not running. Please refresh the page.")
 
 # Add the project root to the Python path
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../..')))