Spaces:
Build error
Build error
| import streamlit as st | |
| import json | |
| import difflib | |
| import random | |
| from datetime import datetime | |
| # ----------------------------- | |
| # Load secrets from Hugging Face Secrets | |
| # ----------------------------- | |
| USER_CREDENTIALS = st.secrets["general"]["user_credentials"] | |
| PRODUCT_DATA = json.loads(st.secrets["general"]["product_data_json"]) | |
| QUIZ_DATA = json.loads(st.secrets["general"]["quiz_data_json"]) | |
| # ----------------------------- | |
| # Initialize session state | |
| # ----------------------------- | |
| if "chat_history" not in st.session_state: | |
| st.session_state.chat_history = [] | |
| if "quiz_score" not in st.session_state: | |
| st.session_state.quiz_score = 0 | |
| if "quiz_attempts" not in st.session_state: | |
| st.session_state.quiz_attempts = 0 | |
| if "logged_in" not in st.session_state: | |
| st.session_state.logged_in = False | |
| if "username" not in st.session_state: | |
| st.session_state.username = "" | |
| # ----------------------------- | |
| # Login function | |
| # ----------------------------- | |
| def login(): | |
| st.title("🔐 Sales Mentor Chatbot Login") | |
| username = st.text_input("Username") | |
| password = st.text_input("Password", type="password") | |
| if st.button("Login"): | |
| if username in USER_CREDENTIALS and USER_CREDENTIALS[username] == password: | |
| st.session_state.logged_in = True | |
| st.session_state.username = username | |
| st.success("Login successful!") | |
| else: | |
| st.error("Invalid credentials") | |
| # ----------------------------- | |
| # Fuzzy match for NLP queries | |
| # ----------------------------- | |
| def get_best_match(query, data_keys): | |
| return difflib.get_close_matches(query, data_keys, n=1, cutoff=0.6) | |
| # ----------------------------- | |
| # Respond to product query | |
| # ----------------------------- | |
| def respond_to_query(query): | |
| match = get_best_match(query.lower(), PRODUCT_DATA.keys()) | |
| if match: | |
| product = PRODUCT_DATA[match[0]] | |
| response = f"**{match[0]}**:\n- Composition: {product['composition']}\n- Indication: {product['indication']}\n- Dosage: {product['dosage']}\n- Warnings: {product['warnings']}" | |
| return response | |
| else: | |
| return "Sorry, I couldn't find that product. Try checking the spelling or ask about another." | |
| # ----------------------------- | |
| # Start quiz for a product | |
| # ----------------------------- | |
| def start_quiz(product_name): | |
| questions = QUIZ_DATA.get(product_name, []) | |
| if not questions: | |
| st.warning("No quiz available for this product.") | |
| return | |
| question = random.choice(questions) | |
| st.markdown(f"**Quiz:** {question['question']}") | |
| selected = st.radio("Choose your answer:", question["options"]) | |
| if st.button("Submit Answer"): | |
| st.session_state.quiz_attempts += 1 | |
| if selected == question["answer"]: | |
| st.success("Correct!") | |
| st.session_state.quiz_score += 1 | |
| else: | |
| st.error(f"Incorrect. Correct answer: {question['answer']}") | |
| # ----------------------------- | |
| # Show quiz summary | |
| # ----------------------------- | |
| def show_quiz_summary(): | |
| st.markdown(f"**Quiz Summary**") | |
| st.write(f"Attempts: {st.session_state.quiz_attempts}") | |
| st.write(f"Score: {st.session_state.quiz_score}") | |
| # ----------------------------- | |
| # Admin dashboard | |
| # ----------------------------- | |
| def admin_dashboard(): | |
| st.title("📊 Admin Dashboard") | |
| st.write(f"User: {st.session_state.username}") | |
| st.write(f"Login Time: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}") | |
| st.write(f"Questions Asked: {len(st.session_state.chat_history)}") | |
| st.write("📘 Quiz Performance") | |
| st.write(f"Total Attempts: {st.session_state.quiz_attempts}") | |
| st.write(f"Total Score: {st.session_state.quiz_score}") | |
| for q, a in st.session_state.chat_history: | |
| st.markdown(f"- **Q:** {q}\n **A:** {a}") | |
| # ----------------------------- | |
| # Chatbot UI | |
| # ----------------------------- | |
| def chat_ui(): | |
| st.title("💬 Sales Mentor Bot") | |
| st.markdown("Hello! I'm your sales mentor. Choose a topic or type your question:") | |
| tiles = ["1. Product Info", "2. FAQs", "3. Objections", "4. Quiz", "5. Recall History"] | |
| selected_tile = st.selectbox("Select an option:", tiles) | |
| if selected_tile.startswith("1"): | |
| user_input = st.text_input("Ask about a product:") | |
| if user_input: | |
| response = respond_to_query(user_input) | |
| st.session_state.chat_history.append((user_input, response)) | |
| st.markdown(f"**You:** {user_input}") | |
| st.markdown(f"**Bot:** {response}") | |
| elif selected_tile.startswith("2"): | |
| faqs = PRODUCT_DATA["Dytor"]["faq"] | |
| st.markdown("**FAQs for Dytor:**") | |
| for faq in faqs: | |
| st.markdown(f"- {faq}") | |
| elif selected_tile.startswith("3"): | |
| objections = PRODUCT_DATA["Dytor"]["objections"] | |
| st.markdown("**Common Objections for Dytor:**") | |
| for obj in objections: | |
| st.markdown(f"- {obj}") | |
| elif selected_tile.startswith("4"): | |
| start_quiz("Dytor") | |
| show_quiz_summary() | |
| elif selected_tile.startswith("5"): | |
| st.markdown("**Conversation History:**") | |
| for q, a in st.session_state.chat_history: | |
| st.markdown(f"- **You:** {q}\n **Bot:** {a}") | |
| # ----------------------------- | |
| # App flow | |
| # ----------------------------- | |
| if not st.session_state.logged_in: | |
| login() | |
| elif st.session_state.username == "admin": | |
| admin_dashboard() | |
| else: | |
| chat_ui() | |