Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import gradio as gr
 import requests
+import re
 
 # ----------------------------
 # Load Knowledge File
@@ -9,14 +10,14 @@ if not os.path.exists("knowledge.txt"):
     raise FileNotFoundError("Create a 'knowledge.txt' file in this folder.")
 
 with open("knowledge.txt", "r", encoding="utf-8") as f:
-    knowledge_base = f.read()
+    knowledge_lines = [line.strip() for line in f if line.strip()]
 
 # ----------------------------
 # Groq API Configuration
 # ----------------------------
 GROQ_API_KEY = os.getenv("GROQ_API_KEY") or "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"
-GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions"  # OpenAI-compatible endpoint
-MODEL_NAME = "llama3-8b-8192"  # Change to any Groq-supported model
+GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions"
+MODEL_NAME = "llama3-8b-8192"
 
 HEADERS = {
     "Authorization": f"Bearer {GROQ_API_KEY}",
@@ -24,35 +25,57 @@ HEADERS = {
 }
 
 # ----------------------------
-# System Prompt (Sales Assistant)
+# Helper: Retrieve relevant knowledge
 # ----------------------------
-SYSTEM_MESSAGE = (
-    "You are a helpful and friendly cosmetics sales assistant. "
-    "You ONLY know the products listed below:\n\n"
-    f"{knowledge_base}\n\n"
-    "Rules:\n"
-    "- Only answer using the provided product list.\n"
-    "- Do NOT invent products.\n"
-    "- If user writes in Swahili, reply in Swahili.\n"
-    "- If user writes in English, reply in English.\n"
-    "- Be friendly and professional."
-)
+def retrieve_relevant_kb(user_input, max_lines=10):
+    """
+    Returns up to max_lines from knowledge_lines that match words in user_input.
+    Simple keyword match.
+    """
+    user_words = set(re.findall(r"\w+", user_input.lower()))
+    relevant = []
+
+    for line in knowledge_lines:
+        line_words = set(re.findall(r"\w+", line.lower()))
+        if user_words & line_words:  # intersection
+            relevant.append(line)
+        if len(relevant) >= max_lines:
+            break
+
+    # If nothing matches, fallback to first few lines
+    if not relevant:
+        relevant = knowledge_lines[:max_lines]
+
+    return "\n".join(relevant)
 
 # ----------------------------
-# Chat Function (Using Groq API)
+# Chat Function
 # ----------------------------
 def chat(user_input, messages):
     # Ensure messages is a list of dicts
     if messages is None or not isinstance(messages, list):
         messages = []
 
-    # Build messages for Groq API
-    groq_messages = [{"role": "system", "content": SYSTEM_MESSAGE}]
+    # Retrieve relevant knowledge
+    kb_text = retrieve_relevant_kb(user_input, max_lines=10)
+
+    system_prompt = (
+        "You are a helpful and friendly cosmetics sales assistant. "
+        "You ONLY know the products listed below:\n\n"
+        f"{kb_text}\n\n"
+        "Rules:\n"
+        "- Only answer using the provided product list.\n"
+        "- Do NOT invent products.\n"
+        "- If user writes in Swahili, reply in Swahili.\n"
+        "- If user writes in English, reply in English.\n"
+        "- Be friendly and professional."
+    )
+
+    # Build Groq messages
+    groq_messages = [{"role": "system", "content": system_prompt}]
     for m in messages:
         if isinstance(m, dict) and "role" in m and "content" in m:
             groq_messages.append({"role": m["role"], "content": m["content"]})
-
-    # Add current user input
     groq_messages.append({"role": "user", "content": user_input})
 
     payload = {
@@ -70,7 +93,7 @@ def chat(user_input, messages):
     except Exception as e:
         reply = f"Sorry, I could not process your request at the moment. ({str(e)})"
 
-    # Update conversation state
+    # Update messages state
     messages.append({"role": "user", "content": user_input})
     messages.append({"role": "assistant", "content": reply})
 
@@ -84,13 +107,9 @@ with gr.Blocks() as demo:
 
     chatbot = gr.Chatbot()
     msg = gr.Textbox(label="Ask about products / Uliza kuhusu bidhaa", autofocus=True)
-    state = gr.State([])  # <-- ensure initial state is a list
+    state = gr.State([])
 
-    msg.submit(
-        chat,
-        inputs=[msg, state],
-        outputs=[chatbot, state]
-    )
+    msg.submit(chat, inputs=[msg, state], outputs=[chatbot, state])
 
 # ----------------------------
 # Launch App