fix the process_question_with_citations

rag_pipeline.py

@@ -279,10 +279,11 @@ def process_question_with_citations(agent, question: str, chunks_directory="./da
         # Get only the cited elements
         cited_elements = []
         for element in all_chunks_data:
-            if element.get("element_id") in unique_ids:
-                cited_elements.append(element)
-            # Also check nested elements
-            if "elements" in element:
+            if element.get("type") =='TableElement':
+               if element.get("element_id") in unique_ids:
+                 cited_elements.append(element['elements'])
+            else :
+             if "elements" in element:
                 for nested_element in element["elements"]:
                     if nested_element.get("element_id") in unique_ids:
                         cited_elements.append(nested_element)
@@ -299,7 +300,7 @@ def process_question_with_citations(agent, question: str, chunks_directory="./da
     except Exception as e:
         print(f"Error processing question: {e}")
         return {
-            "response": f"Error processing your question: {str(e)}",
+            "response": response_text,
             "cited_elements_json": "[]",
             "unique_ids": []
         }