Update app.py

app.py

@@ -62,33 +62,26 @@ def stream_chat_with_rag(
 
 def format_answer_string(answer: str):
     """
-    This function takes the full string returned by the API (which includes both the assistant's answer and documents)
-    and splits it into separate sections: the assistant's answer and the relevant documents.
+    This function extracts and formats the assistant's response before document metadata.
+    Anything after the marker `[(` (where documents are listed) is ignored.
     """
-    # Step 1: Split the answer by the point where document context starts
-    split_marker = "Extracted Documents with Descriptive Context:"
+    # Step 1: Split the response at the start of the document metadata
+    split_marker = "[("
     if split_marker in answer:
-        assistant_answer, documents_section = answer.split(split_marker, 1)
+        # Everything before the marker is the relevant answer
+        answer_before_docs = answer.split(split_marker)[0]
     else:
-        # If no documents found, return the whole answer as the assistant's response
-        return f"Assistant: {answer.strip()}"
+        # If no documents metadata, return the entire answer
+        answer_before_docs = answer
     
-    # Step 2: Clean and format the assistant's answer
-    formatted_answer = f"Assistant: {assistant_answer.strip()}\n\n"
-
-    # Step 3: Format the documents section
-    formatted_answer += "Relevant Documents:\n"
-    document_entries = documents_section.split("Document ")
+    # Step 2: Clean up formatting by replacing escaped newline characters
+    formatted_answer = answer_before_docs.replace("\\n", "\n").strip()
     
-    # Step 4: Reformat each document entry
-    for entry in document_entries[1:]:  # Skip the first empty split
-        doc_number, rest = entry.split(":", 1)
-        formatted_answer += f"{doc_number}. {rest.strip()}\n\n"
-
     return formatted_answer
 
 
 
+
 # this version works just issue with formatting 
 # def stream_chat_with_rag(
 #     message: str,