Simplify message rendering and update citation format in views

Author: sahilds1

frontend/src/components/Header/Chat.tsx

@@ -4,7 +4,6 @@ import "../../components/Header/chat.css";
 import { useState, useEffect, useRef } from "react";
 import TypingAnimation from "./components/TypingAnimation";
 import ErrorMessage from "../ErrorMessage";
-import ParseStringWithLinks from "../../services/parsing/ParseWithSource";
 import axios from "axios";
 import {
   FaPlus,
@@ -321,30 +320,15 @@ const Chat: React.FC<ChatDropDownProps> = ({ showChat, setShowChat }) => {
                           <FaComment className="chat_text_icon" />
                         </div>
                         <div className="chat_text_wrap">
-                          {message.is_user ? (
-                            <pre
-                              style={{
-                                fontFamily: "inherit",
-                                whiteSpace: "pre-wrap",
-                                wordWrap: "break-word",
-                              }}
-                            >
-                              {message.content}
-                            </pre>
-                          ) : (
-                            <div
-                              style={{
-                                fontFamily: "inherit",
-                                whiteSpace: "pre-wrap",
-                                wordWrap: "break-word",
-                              }}
-                            >
-                              <ParseStringWithLinks 
-                                text={message.content} 
-                                chunkData={[]} 
-                              />
-                            </div>
-                          )}
+                          <pre
+                            style={{
+                              fontFamily: "inherit",
+                              whiteSpace: "pre-wrap",
+                              wordWrap: "break-word",
+                            }}
+                          >
+                            {message.content}
+                          </pre>
                         </div>
                       </div>
                     ))

server/api/views/assistant/views.py

@@ -186,7 +186,7 @@ def search_documents(query: str, user=user) -> str:
 
                     # Format results with clear structure and metadata
                     prompt_texts = [
-                        f"[Document {i + 1} - File: {obj['file_id']}, Page: {obj['page_number']}, Chunk: {obj['chunk_number']}, Similarity: {1 - obj['distance']:.3f}]\n{obj['text']}\n[End Document {i + 1}]"
+                        f"[Document {i + 1} - File: {obj['file_id']}, Name: {obj['name']}, Page: {obj['page_number']}, Chunk: {obj['chunk_number']}, Similarity: {1 - obj['distance']:.3f}]\n{obj['text']}\n[End Document {i + 1}]"
                         for i, obj in enumerate(embeddings_results)
                     ]
 
@@ -214,7 +214,7 @@ def search_documents(query: str, user=user) -> str:
             After gathering information through semantic searches, provide responses that:
             1. Answer the user's question directly using only the found information
             2. Structure responses with clear sections and paragraphs
-            3. Include citations after EACH sentence using this exact format: ***[File ID {file_id}, Page {page_number}, Chunk {chunk_number}]***
+            3. Include citations using this exact format: ***[Name {name}, Page {page_number}]***
             4. Only cite information that directly supports your statements
 
             If no relevant information is found in the documents, clearly state that the information is not available in the uploaded documents.