Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import time | |
| from business_ai import BusinessAIAssistant | |
| from business_config import BUSINESS_CONFIG, PRODUCTS, SERVICES | |
| # Initialize the AI assistant | |
| ai_assistant = BusinessAIAssistant(BUSINESS_CONFIG, PRODUCTS, SERVICES) | |
| # Predefined responses for common queries | |
| def get_business_response(intent): | |
| responses = { | |
| "hours": f"Our store hours are {BUSINESS_CONFIG['hours']}.", | |
| "location": f"We're located at {BUSINESS_CONFIG['address']}. Come visit us!", | |
| "contact": f"You can reach us at {BUSINESS_CONFIG['phone']} or email {BUSINESS_CONFIG['email']}.", | |
| "products": f"We offer a wide range of {BUSINESS_CONFIG['products']}. What specifically are you interested in?", | |
| "services": f"We provide these services: {BUSINESS_CONFIG['services']}.", | |
| "returns": f"Our return policy: {BUSINESS_CONFIG['return_policy']}.", | |
| "shipping": f"Yes! We offer {BUSINESS_CONFIG['shipping'].lower()}", | |
| "greeting": f"Hello! Welcome to {BUSINESS_CONFIG['name']}. How can I help you today?", | |
| "thanks": "You're welcome! Is there anything else I can help you with?", | |
| "goodbye": f"Thank you for contacting {BUSINESS_CONFIG['name']}. Have a great day!", | |
| "help": "I can help you with: \n• Store hours and location \n• Product information and pricing \n• Return and shipping policies \n• Technical comparisons between products \n• And much more! What do you need help with today?", | |
| "capabilities": f"I'm your virtual assistant for {BUSINESS_CONFIG['name']}! I can answer questions about our products, services, policies, and more. What would you like to know?", | |
| "values": f"Our company values: {BUSINESS_CONFIG['values']}", | |
| "people": f"Key people at our company: {BUSINESS_CONFIG['key_people']}", | |
| "payment": f"We accept these payment methods: {BUSINESS_CONFIG['payment_methods']}." | |
| } | |
| return responses.get(intent, "") | |
| # Function to handle product inquiries | |
| def handle_product_inquiry(product_type, model=None): | |
| if product_type not in PRODUCTS: | |
| available_products = ", ".join(PRODUCTS.keys()) | |
| return f"We don't carry that product. We mainly offer {available_products}." | |
| if model: | |
| if model in PRODUCTS[product_type]: | |
| product = PRODUCTS[product_type][model] | |
| return f"Our {model} {product_type} is the {product['name']}, priced at {product['price']}. Features: {product['features']}" | |
| else: | |
| available_models = ", ".join(PRODUCTS[product_type].keys()) | |
| return f"We don't have a {model} model for {product_type}. We have {available_models}." | |
| else: | |
| models = PRODUCTS[product_type] | |
| response = f"For {product_type}, we have these models:" | |
| for model_name, details in models.items(): | |
| response += f"\n- {model_name}: {details['name']} ({details['price']})" | |
| return response | |
| # Main chat function | |
| def chat_with_bot(message, chat_history): | |
| user_message = message.lower() | |
| bot_response = "" | |
| # Format conversation history for context | |
| conversation_history = "" | |
| for entry in chat_history: | |
| if entry['role'] == 'user': | |
| conversation_history += f"Customer: {entry['content']}\n" | |
| else: | |
| conversation_history += f"Assistant: {entry['content']}\n" | |
| # Check for shipping first (before greetings to avoid "hi" in "shipping" triggering greeting) | |
| if any(word in user_message for word in ["ship", "delivery", "deliver", "shipping", "free shipping"]): | |
| bot_response = get_business_response("shipping") | |
| # Check for returns | |
| elif any(word in user_message for word in ["return", "exchange", "refund"]): | |
| bot_response = get_business_response("returns") | |
| # Check for products | |
| elif any(word in user_message for word in ["product", "sell", "offer", "item", "inventory", "have", "stock"]): | |
| bot_response = get_business_response("products") | |
| # Check for services | |
| elif any(word in user_message for word in ["service", "support", "repair", "maintenance", "warranty"]): | |
| bot_response = get_business_response("services") | |
| # Check for contact info | |
| elif any(word in user_message for word in ["contact", "phone", "call", "email", "reach"]): | |
| bot_response = get_business_response("contact") | |
| # Check for location | |
| elif any(word in user_message for word in ["where", "location", "address", "find", "directions"]): | |
| bot_response = get_business_response("location") | |
| # Check for business hours | |
| elif any(word in user_message for word in ["hour", "time", "open", "close", "when"]): | |
| bot_response = get_business_response("hours") | |
| # Check for company values | |
| elif any(word in user_message for word in ["value", "mission", "believe", "principle"]): | |
| bot_response = get_business_response("values") | |
| # Check for people | |
| elif any(word in user_message for word in ["who", "people", "team", "owner", "ceo", "manager"]): | |
| bot_response = get_business_response("people") | |
| # Check for payment methods | |
| elif any(word in user_message for word in ["pay", "payment", "credit", "debit", "cash", "finance"]): | |
| bot_response = get_business_response("payment") | |
| # Check for help or capabilities | |
| elif any(word in user_message for word in ["help", "what can you do", "how can you help", "capabilities", "what do you know"]): | |
| bot_response = get_business_response("capabilities") | |
| # Check for thanks | |
| elif any(word in user_message for word in ["thank", "appreciate", "thanks"]): | |
| bot_response = get_business_response("thanks") | |
| # Check for goodbye | |
| elif any(word in user_message for word in ["bye", "goodbye", "see you", "farewell"]): | |
| bot_response = get_business_response("goodbye") | |
| # Check for greetings (moved to end to avoid conflicts) | |
| elif any(word in user_message for word in ["hi", "hello", "hey", "greetings", "howdy"]): | |
| bot_response = get_business_response("greeting") | |
| # Handle product inquiries | |
| elif any(word in user_message for word in ["smartphone", "phone", "iphone", "android"]): | |
| if "pro" in user_message or "premium" in user_message or "high-end" in user_message: | |
| bot_response = handle_product_inquiry("smartphone", "pro") | |
| elif "basic" in user_message or "cheap" in user_message or "budget" in user_message: | |
| bot_response = handle_product_inquiry("smartphone", "basic") | |
| else: | |
| bot_response = handle_product_inquiry("smartphone") | |
| elif any(word in user_message for word in ["laptop", "computer", "notebook", "macbook", "pc"]): | |
| if "pro" in user_message or "premium" in user_message or "high-end" in user_message: | |
| bot_response = handle_product_inquiry("laptop", "premium") | |
| elif "standard" in user_message or "basic" in user_message or "budget" in user_message: | |
| bot_response = handle_product_inquiry("laptop", "standard") | |
| else: | |
| bot_response = handle_product_inquiry("laptop") | |
| elif any(word in user_message for word in ["tablet", "ipad", "android tablet"]): | |
| bot_response = "We offer a variety of tablets from leading brands. Would you like information about Apple iPads or Android tablets?" | |
| elif any(word in user_message for word in ["accessory", "accessories", "case", "charger", "headphone"]): | |
| bot_response = "We carry a wide range of accessories including cases, chargers, headphones, and more. Is there a specific accessory you're interested in?" | |
| # For all other questions, use the AI module | |
| else: | |
| bot_response = ai_assistant.generate_response(user_message, conversation_history) | |
| # If the AI module returns a generic response, try to provide something more specific | |
| if "thank you for your question" in bot_response.lower(): | |
| if "?" in user_message or "what" in user_message or "how" in user_message: | |
| bot_response = f"I'd be happy to help! At {BUSINESS_CONFIG['name']}, we offer {BUSINESS_CONFIG['products']}. Is there a specific product or service you're interested in?" | |
| else: | |
| bot_response = "How can I assist you today? I can help with product information, store details, or technical support." | |
| # Add a small delay to make it feel more natural | |
| time.sleep(0.3) | |
| # Append to chat history using the new Messages format | |
| chat_history.append({"role": "user", "content": message}) | |
| chat_history.append({"role": "assistant", "content": bot_response}) | |
| return "", chat_history | |
| # Simple CSS for basic styling | |
| custom_css = """ | |
| #chatbot { | |
| height: 400px; | |
| overflow-y: auto; | |
| border: 1px solid #e0e0e0; | |
| border-radius: 8px; | |
| padding: 15px; | |
| } | |
| .gr-button { | |
| background-color: #4CAF50 !important; | |
| color: white !important; | |
| } | |
| .send-button { | |
| background-color: #2196F3 !important; | |
| } | |
| .clear-button { | |
| background-color: #f44336 !important; | |
| } | |
| """ | |
| with gr.Blocks(css=custom_css, title=f"{BUSINESS_CONFIG['name']} Support") as demo: | |
| with gr.Column(): | |
| gr.Markdown(f"# 🤖 {BUSINESS_CONFIG['name']} Support Chat") | |
| # Plain text instructions - no boxes | |
| gr.Markdown("### 💡 How to Use This Chatbot") | |
| gr.Markdown("This AI assistant can help you with product information, store policies, technical support, and more. Just type your question in the chat box or click on any of the example questions to get started.") | |
| gr.Markdown("### ✅ What I Can Help With") | |
| gr.Markdown(""" | |
| • Product information, specifications, and pricing | |
| • Store hours, location, and contact information | |
| • Shipping options and return policies | |
| • Technical support and troubleshooting | |
| • Product comparisons and recommendations | |
| • Service information and repair options | |
| """) | |
| gr.Markdown("### ⚠️ Limitations") | |
| gr.Markdown(""" | |
| • I can't process orders or payments directly | |
| • I don't have access to your personal account information | |
| • For complex issues, I may connect you with a human specialist | |
| • My knowledge is based on information available up to my last training | |
| """) | |
| with gr.Row(): | |
| with gr.Column(scale=2): | |
| gr.Markdown("### 🏪 Store Information") | |
| gr.Markdown(f""" | |
| **Hours**: {BUSINESS_CONFIG['hours']} | |
| **Address**: {BUSINESS_CONFIG['address']} | |
| **Phone**: {BUSINESS_CONFIG['phone']} | |
| **Email**: {BUSINESS_CONFIG['email']} | |
| **Shipping**: {BUSINESS_CONFIG['shipping']} | |
| """) | |
| gr.Markdown("---") | |
| with gr.Row(): | |
| with gr.Column(scale=2): | |
| chatbot = gr.Chatbot( | |
| label="Conversation", | |
| elem_id="chatbot", | |
| type="messages", | |
| show_copy_button=True | |
| ) | |
| with gr.Row(): | |
| msg = gr.Textbox( | |
| label="Type your message here...", | |
| placeholder="Ask me about products, services, or anything else...", | |
| lines=1, | |
| scale=4 | |
| ) | |
| send_btn = gr.Button("Send", variant="primary", elem_classes="send-button", scale=1) | |
| clear = gr.Button("Clear Conversation", variant="stop", elem_classes="clear-button") | |
| with gr.Column(scale=1): | |
| gr.Markdown("### 💡 Common Questions") | |
| examples = gr.Examples( | |
| examples=[ | |
| ["What products do you have?"], | |
| ["Do you offer free shipping?"], | |
| ["What are your store hours?"], | |
| ["What's your return policy?"], | |
| ["Can you compare your laptop models?"], | |
| ["What services do you provide?"], | |
| ["What payment methods do you accept?"] | |
| ], | |
| inputs=msg, | |
| label="Click any question to try it!", | |
| examples_per_page=7 | |
| ) | |
| # Event handlers | |
| msg.submit(chat_with_bot, [msg, chatbot], [msg, chatbot]) | |
| send_btn.click(chat_with_bot, [msg, chatbot], [msg, chatbot]) | |
| clear.click(lambda: None, None, chatbot, queue=False) | |
| if __name__ == "__main__": | |
| demo.launch(share=True) |