Update retailgpt_evaluator/app.py

retailgpt_evaluator/app.py

@@ -2,56 +2,41 @@ import streamlit as st
 from shared.hf_helpers import build_pipeline
 from retailgpt_evaluator.leaderboard import build_leaderboard
 import yaml, pandas as pd, os
+from pathlib import Path
+
+def main():
+    # Safely set page config (won’t error inside streamlit_hub.py)
+    try:
+        st.set_page_config(page_title="RetailGPT Evaluator", page_icon="🛍️", layout="wide")
+    except st.errors.StreamlitAPIException:
+        pass
+
+    st.title("🛍️ RetailGPT Evaluator — AxionX Digital")
+
+    # Load config safely
+    config_path = Path(__file__).resolve().parent / "config.yaml"
+    with open(config_path) as f:
+        cfg = yaml.safe_load(f)
+
+    # Show leaderboard if exists
+    if os.path.exists("models/retail_eval_results.json"):
+        df = build_leaderboard()
+        st.subheader("📊 Model Leaderboard")
+        st.dataframe(df, use_container_width=True)
+    else:
+        st.warning("Run `evaluate.py` first to generate metrics.")
 
-# ✅ Safe config — only runs if not already set by hub
-try:
-    st.set_page_config(page_title="RetailGPT Evaluator", page_icon="🛍️", layout="wide")
-except st.errors.StreamlitAPIException:
-    pass  # Already configured by main Streamlit Hub
-
-# -------------------------------
-# Title & Description
-# -------------------------------
-st.title("🛍️ RetailGPT Evaluator — AxionX Digital")
-st.write("Benchmark and chat with multiple retail QA models.")
-
-# -------------------------------
-# Load Configuration
-# -------------------------------
-with open("retailgpt_evaluator/config.yaml") as f:
-    cfg = yaml.safe_load(f)
-
-# -------------------------------
-# Leaderboard Section
-# -------------------------------
-if os.path.exists("models/retail_eval_results.json"):
-    df = build_leaderboard()
-    st.subheader("📊 Model Leaderboard")
-    st.dataframe(df, use_container_width=True)
-else:
-    st.warning("Run `evaluate.py` first to generate metrics.")
-
-# -------------------------------
-# Chat Interface
-# -------------------------------
-st.markdown("---")
-st.subheader("💬 Chat With a Model")
-
-model_name = st.selectbox("Choose a model to chat with:", cfg["models"])
-pipe = build_pipeline(model_name)
+    # Model chat interface
+    st.markdown("---")
+    model_name = st.selectbox("Choose a model to chat with:", cfg["models"])
+    pipe = build_pipeline(model_name)
 
-query = st.text_area("Customer query:", "I want to return a damaged product.")
-if st.button("Ask Model"):
-    if query.strip():
-        with st.spinner("Generating response..."):
-            result = pipe(query, max_new_tokens=cfg["demo"]["max_new_tokens"])
+    query = st.text_area("Customer query:", "I want to return a damaged product.")
+    if st.button("Ask Model"):
+        result = pipe(query, max_new_tokens=cfg["demo"]["max_new_tokens"])
         st.markdown("### 🧠 Model Response")
         st.write(result[0]["generated_text"])
-    else:
-        st.warning("Please enter a customer query above before submitting.")
 
-# -------------------------------
-# Footer
-# -------------------------------
-st.markdown("---")
-st.caption("© 2025 AxionX Digital — Innovating Tomorrow")
+# Ensure it's import-safe
+if __name__ == "__main__":
+    main()