import streamlit as st
import yaml
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

st.set_page_config(page_title="AxionX Digital Hub", page_icon="🚀", layout="wide")

# -------------------------------
# Optional: Show Logo
# -------------------------------
logo_path = Path("assets/axionx_logo.png")
if logo_path.exists():
    st.sidebar.image(str(logo_path), width=180)  # 👈 Adjust size as needed

st.sidebar.title("🧠 Select Demo")
choice = st.sidebar.radio(
    "Choose a demo:",
    (
        "🤖 AxionX Base Demo",
        "💰 FinanceGPT",
        "⚖️ LegalDoc Summarizer",
        "🛍️ RetailGPT Evaluator",
    ),
)

st.sidebar.markdown("---")
st.sidebar.info(
    "AxionX Digital fine-tunes and evaluates domain-specific LLMs for Finance, Legal, "
    "and Retail use cases."
)

st.title("🚀 AxionX Digital — Model Training & Evaluation Suite")

if "FinanceGPT" in choice:
    finance_app()
elif "LegalDoc" in choice:
    legal_app()
elif "RetailGPT" in choice:
    retail_app()
elif "Base Demo" in choice:
    gradio_app()

st.markdown("---")
st.caption("© 2025 AxionX Digital — Innovating Tomorrow")