Update streamlit_hub.py

streamlit_hub.py

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