Update app.py

app.py

@@ -168,9 +168,20 @@ def save_qa_history(history_entry):
             st.error("Invalid history data format, must be a list")
             history_data = []
             
-        # Remove any None or invalid entries
-        history_data = [entry for entry in history_data if isinstance(entry, dict) and 
-                       all(key in entry for key in ["timestamp", "query", "answer"])]
+        # Process and validate each entry
+        processed_history = []
+        for entry in history_data:
+            if isinstance(entry, dict) and all(key in entry for key in ["timestamp", "query", "answer"]):
+                # Process answer if it's a Document or dict
+                if isinstance(entry["answer"], dict):
+                    entry["answer"] = entry["answer"].get('answer', str(entry["answer"]))
+                elif hasattr(entry["answer"], 'content'):
+                    entry["answer"] = entry["answer"].content
+                else:
+                    entry["answer"] = str(entry["answer"])
+                processed_history.append(entry)
+                
+        history_data = processed_history
             
         # Save updated history locally
         try:
@@ -244,11 +255,22 @@ def add_to_qa_history(query: str, answer: str):
             st.warning("Empty query or answer detected, skipping history update")
             return None
             
+        # Handle different answer types
+        if isinstance(answer, dict):
+            # If answer is a dict with 'answer' key, extract it
+            processed_answer = answer.get('answer', str(answer))
+        elif hasattr(answer, 'content'):
+            # If answer is a Document-like object with content attribute
+            processed_answer = answer.content
+        else:
+            # Convert answer to string for any other type
+            processed_answer = str(answer)
+            
         # Create history entry with proper timestamp
         history_entry = {
             "timestamp": (datetime.now() + timedelta(hours=5)).strftime("%Y-%m-%dT%H:%M:%S"),
             "query": query,
-            "answer": answer
+            "answer": processed_answer
         }
         
         # Save entry