Update app.py

app.py

@@ -42,9 +42,16 @@ def process_query(query, dataset_choice="all"):
         
         for dataset_name in search_datasets:
             if dataset_name in datasets:
-                for doc in datasets[dataset_name]['documents']:
-                    if any(keyword.lower() in doc.lower() for keyword in query.split()):
-                        relevant_contexts.append((doc, dataset_name))
+                documents = datasets[dataset_name]['documents']
+                for doc in documents:
+                    # Handle both string and list document types
+                    if isinstance(doc, list):
+                        doc_text = ' '.join(doc)
+                    else:
+                        doc_text = str(doc)
+                    
+                    if any(keyword.lower() in doc_text.lower() for keyword in query.split()):
+                        relevant_contexts.append((doc_text, dataset_name))
         
         context_info = f"From {relevant_contexts[0][1]}: {relevant_contexts[0][0]}" if relevant_contexts else "Searching across datasets..."