adding formatting response text function

app.py

@@ -241,6 +241,26 @@ class RAGQuestionAnswering:
 
         return "\n\n".join(doc.page_content for doc in docs)
 
+    def extract_assistant_content(self, data: Dict[str, Any]) ->:
+        """
+        Extracts and returns the assistant content text from the provided data structure.
+        
+        :param data: A list containing a dictionary with 'generated_text'.
+        :return: The 'content' text of the 'assistant' role, or None if not found.
+        """
+        try:
+            # Navigate to the assistant content text
+            generated_text = data[0].get('generated_text', [])
+            for item in generated_text:
+                if item.get('role') == 'assistant':
+                    return item.get('content')
+        except (IndexError, AttributeError):
+            return None  # Return None if the structure is not as expected
+        
+        return None  # Return None if no assistant content is found
+
+
+
     def generate_response(self, input_dict: Dict[str, Any]) -> str:
         """
         Parameters
@@ -273,10 +293,11 @@ class RAGQuestionAnswering:
         ]
         
         response = self.pipe(messages)
-        print("#"*88)
-        print(response, type(response))
+        output_text = self.extract_assistant_content(response)
         
-        return str(response)
+        if not output_text:
+            return 'Unable to generate output'
+        return output_text
 
     def setup_rag_chain(self) -> None:
         """