Update app.py

app.py

@@ -235,43 +235,51 @@ def generate_answer(query, history):
             for chunk, meta in zip(relevant_chunks, metadata)
         ])
         
-        # Build conversation history for context
-        history_context = ""
-        if history and len(history) > 0:
-            recent_history = history[-3:] if len(history) > 3 else history
-            history_parts = []
-            for h in recent_history:
-                if isinstance(h, (list, tuple)) and len(h) >= 2:
-                    history_parts.append(f"User: {h[0]}\nAssistant: {h[1]}")
-            history_context = "\n".join(history_parts)
+        # Build messages array for Groq API
+        messages = []
         
-        # Create prompt
-        prompt = f"""You are a helpful assistant that answers questions based on the provided document context.
-
-Previous conversation:
-{history_context}
-
-Context from documents:
-{context}
-
-Current question: {query}
+        # Add system message
+        system_prompt = """You are a helpful assistant that answers questions based on the provided document context.
 
 Instructions:
 - Answer based strictly on the provided context
-- If the answer isn't in the context, say so
+- If the answer isn't in the context, say so clearly
 - Be concise and accurate
-- Reference specific sources when relevant
+- Reference specific sources when relevant"""
+        
+        messages.append({
+            "role": "system",
+            "content": system_prompt
+        })
+        
+        # Add conversation history (last 3 exchanges)
+        if history and len(history) > 0:
+            recent_history = history[-3:] if len(history) > 3 else history
+            for h in recent_history:
+                if isinstance(h, (list, tuple)) and len(h) >= 2:
+                    messages.append({
+                        "role": "user",
+                        "content": h[0]
+                    })
+                    messages.append({
+                        "role": "assistant",
+                        "content": h[1]
+                    })
+        
+        # Add current query with context
+        user_message = f"""Context from documents:
+{context}
 
-Answer:"""
+Question: {query}"""
+        
+        messages.append({
+            "role": "user",
+            "content": user_message
+        })
         
         # Call Groq API
         chat_completion = client.chat.completions.create(
-            messages=[
-                {
-                    "role": "user",
-                    "content": prompt,
-                }
-            ],
+            messages=messages,
             model="llama3-8b-8192",
             temperature=0.3,
             max_tokens=1024,