added history

app.py

@@ -83,7 +83,7 @@ qa_answers = list(qa_data.values())
 qa_embeddings = semantic_model.encode(qa_questions, convert_to_tensor=True)
 
 # ---------------------- History-Aware CAG ----------------------
-def retrieve_from_cag(user_query):
+def retrieve_from_cag(user_query, chat_history):
     query_embedding = semantic_model.encode(user_query, convert_to_tensor=True)
     cosine_scores = util.cos_sim(query_embedding, qa_embeddings)[0]
     best_idx = int(np.argmax(cosine_scores))
@@ -120,10 +120,14 @@ def retrieve_from_rag(user_query, chat_history):
     return documents
 
 # ---------------------- Generation function (OpenRouter) ----------------------
-def generate_via_openrouter(context, query):
+def generate_via_openrouter(context, query, chat_history=None):
     print("\n--- Generating via OpenRouter ---")
     print("Context received:", context)
 
+    history_text = ""
+    if chat_history:
+        history_text = "\n".join([f"User: {q}\nBot: {a}" for q, a in chat_history[-2:]])  # Last 2 exchanges only
+
     prompt = f"""<s>[INST]
 You are a Moodle expert assistant.
 Instructions:
@@ -131,16 +135,20 @@ Instructions:
 - Use only the provided documents below to answer.
 - If the answer is not in the documents, simply say: "I don't know." / "Je ne sais pas."
 - Cite only the sources you use, indicated at the end of each document like (Source: https://example.com).
-Documents :
+
+Chat History:
+{history_text}
+
+Documents:
 {context}
-Question : {query}
-Answer :
+
+Question: {query}
+Answer:
 [/INST]
 """
-
     try:
         response = client1.chat.completions.create(
-            model="mistralai/mistral-small-3.1-24b-instruct:free",
+            model="mistralai/mistral-7b-instruct:free",
             messages=[{"role": "user", "content": prompt}]
         )
         return response.choices[0].message.content.strip()
@@ -148,6 +156,7 @@ Answer :
         print(f"Erreur lors de la génération : {e}")
         return "Erreur lors de la génération."
 
+
 # ---------------------- Main Chatbot ----------------------
 def chatbot(query, chat_history):
     print("\n==== New Query ====")