Update app.py

app.py

@@ -826,13 +826,13 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
     
     if selected_document is not None:
         # Remove the "/home/user/app/" part from the document name
-        modified_source = selected_document.metadata['source'].replace('/home/user/app/', '')
-        source_info = f"\n**Primary source document:** {modified_source}, **Page:** {selected_document.metadata['page']}"
+        modified_source = selected_document.metadata['source'].replace('/home/user/app/', '').replace('.pdf', '')
+        source_info = f"\n**Leading source:** {modified_source}, **Page:** {selected_document.metadata['page']}"
     else:
         source_info = "Primary source document: not determined"
     
     #chat_messages = [(prompt_msg['content'], completion['content'])]    
-    chat_messages = [(prompt, completion['content']['answer'] + " -- " + source_info )] 
+    chat_messages = [(prompt, completion['content']['answer'] + source_info )] 
     return '', chat_messages,  state    # total_tokens_used_msg,