Update retailgpt_evaluator/app.py

retailgpt_evaluator/app.py

@@ -3,12 +3,27 @@ from shared.hf_helpers import build_pipeline
 from retailgpt_evaluator.leaderboard import build_leaderboard
 import yaml, pandas as pd, os
 
-st.set_page_config(page_title="RetailGPT Evaluator", page_icon="🛍️", layout="wide")
+# ✅ 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("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")
@@ -16,11 +31,27 @@ if os.path.exists("models/retail_eval_results.json"):
 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)
 
 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"])
+    if query.strip():
+        with st.spinner("Generating response..."):
+            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")