Update app.py

app.py

@@ -110,4 +110,103 @@ interface = gr.Interface(
 
 interface.launch()
 
+import gradio as gr
+from transformers import pipeline
+
+# Use a pre-trained chat model (no coding or training needed)
+chatbot = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3")
+
+# Simple customer knowledge base
+knowledge = {
+    "reset password": "To reset your password, go to Settings > Account > Reset Password.",
+    "refund": "Refunds take 3–5 business days after approval.",
+    "support": "You can contact support at support@example.com."
+}
+
+# Simple rule-based guardrails
+def apply_guardrails(answer):
+    if any(bad in answer.lower() for bad in ["credit card", "ssn", "password"]):
+        return "For your safety, I can’t discuss personal account details here."
+    return answer
+
+# Main chatbot function
+def chat_fn(message):
+    # Step 1: Look for matching info in knowledge base
+    for key, info in knowledge.items():
+        if key in message.lower():
+            answer = info
+            break
+    else:
+        # Step 2: Use AI model if we don’t know the answer
+        prompt = f"Customer question: {message}\nAnswer politely and professionally:"
+        answer = chatbot(prompt, max_new_tokens=100)[0]["generated_text"]
+
+    # Step 3: Guardrails check
+    safe_answer = apply_guardrails(answer)
+    return safe_answer
+
+# ✨ Custom colorful CSS style ✨
+custom_css = """
+body {
+    background: linear-gradient(135deg, #6EE7B7, #3B82F6, #9333EA);
+    background-size: 400% 400%;
+    animation: gradientShift 10s ease infinite;
+    font-family: 'Poppins', sans-serif;
+    color: white;
+}
+@keyframes gradientShift {
+    0% { background-position: 0% 50%; }
+    50% { background-position: 100% 50%; }
+    100% { background-position: 0% 50%; }
+}
+.gradio-container {
+    background-color: rgba(255, 255, 255, 0.15) !important;
+    border-radius: 18px;
+    padding: 25px;
+    box-shadow: 0px 0px 30px rgba(255,255,255,0.4);
+}
+h1 {
+    background: linear-gradient(to right, #FF0080, #FF8C00, #40E0D0);
+    -webkit-background-clip: text;
+    -webkit-text-fill-color: transparent;
+    text-align: center;
+    font-weight: 800;
+    font-size: 2.4em;
+    letter-spacing: 1px;
+}
+textarea, input {
+    border-radius: 12px !important;
+    border: 2px solid #FFD700 !important;
+    background-color: rgba(255, 255, 255, 0.2) !important;
+    color: white !important;
+    font-size: 1em !important;
+}
+button {
+    background: linear-gradient(90deg, #FF0080, #FF8C00);
+    border: none !important;
+    border-radius: 10px !important;
+    color: white !important;
+    font-weight: bold;
+    font-size: 1em;
+    transition: all 0.3s ease;
+}
+button:hover {
+    transform: scale(1.05);
+    background: linear-gradient(90deg, #40E0D0, #FF8C00);
+}
+"""
+
+# 🎨 Gradio interface (colorful + modern)
+interface = gr.Interface(
+    fn=chat_fn,
+    inputs="text",
+    outputs=gr.Textbox(lines=8, label="💬 Chatbot Response"),
+    title="💎 Account & Fund Support Chatbot 💬",
+    description="Ask questions about your account, refunds, or support — your AI assistant is here to help!",
+    css=custom_css
+)
+
+interface.launch()
+
+