Update streamlit_hub.py

streamlit_hub.py

@@ -1,22 +1,30 @@
 import streamlit as st
+from pathlib import Path
 import yaml
+
+# Import demo modules
 from financegpt.app import main as finance_app
 from legaldoc_summarizer.app import main as legal_app
 from retailgpt_evaluator.app import main as retail_app
-from app import main as gradio_app
-from pathlib import Path
+from app import main as gradio_app  # Base Gradio-style demo
 
+# -------------------------------
+# Global Page Config
+# -------------------------------
 st.set_page_config(page_title="AxionX Digital Hub", page_icon="🚀", layout="wide")
 
 # -------------------------------
-# Optional: Show Logo
+# Sidebar Logo
 # -------------------------------
 logo_path = Path(__file__).parent / "assets" / "axionx_logo.png"
 if logo_path.exists():
-    st.sidebar.image(str(logo_path.resolve()), width=180)  # 👈 Use absolute path
+    st.sidebar.image(str(logo_path.resolve()), width=180)
 else:
-    st.sidebar.warning("Logo not found.")
+    st.sidebar.warning("⚠️ Logo not found at /assets/axionx_logo.png")
 
+# -------------------------------
+# Sidebar Navigation
+# -------------------------------
 st.sidebar.title("🧠 Select Demo")
 choice = st.sidebar.radio(
     "Choose a demo:",
@@ -34,16 +42,25 @@ st.sidebar.info(
     "and Retail use cases."
 )
 
+# -------------------------------
+# Main Title
+# -------------------------------
 st.title("🚀 AxionX Digital — Model Training & Evaluation Suite")
 
-if "FinanceGPT" in choice:
+# -------------------------------
+# Load Selected Demo
+# -------------------------------
+if choice.startswith("💰"):
     finance_app()
-elif "LegalDoc" in choice:
+elif choice.startswith("⚖️"):
     legal_app()
-elif "RetailGPT" in choice:
+elif choice.startswith("🛍️"):
     retail_app()
-elif "Base Demo" in choice:
+elif choice.startswith("🤖"):
     gradio_app()
 
+# -------------------------------
+# Footer
+# -------------------------------
 st.markdown("---")
 st.caption("© 2025 AxionX Digital — Innovating Tomorrow")