Update app.py

app.py

@@ -112,7 +112,7 @@ history_aware_retriever = create_history_aware_retriever(llm, compression_retrie
 
 # QA system prompt and chain
 qa_system_prompt = """ You are a highly skilled information retrieval assistant. Use the following context to answer questions effectively. 
-If you don't know the answer, simply state that you don't know. 
+If you don't know the answer, state that you don't know. 
 Your answer should be in {language} language. 
 
 When responding to queries, follow these guidelines:
@@ -132,7 +132,7 @@ When responding to queries, follow these guidelines:
    - The inline citations should be in the format [[1]], [[2]], etc., in the response with links to reference sources.
    - AT THE END OF THE RESPONSE, LIST OUT THE CITATIONS WITH THEIR SOURCES. If there are multiple citations with same source url then only mention that single source url for all of those.
 
-FOLLOW ALL THE GIVEN INSTRUCTIONS, FAILURE TO DO SO WILL RESULT IN TERMINATION OF THE CHAT.
+FOLLOW ALL THE GIVEN INSTRUCTIONS, FAILURE TO DO SO WILL RESULT IN THE TERMINATION OF THE CHAT.
 {context}
 """
 qa_prompt = ChatPromptTemplate.from_messages(
@@ -200,17 +200,9 @@ async def websocket_endpoint(websocket: WebSocket):
                             complete_response += chunk['answer']
                             await websocket.send_json({'response': chunk['answer']})
 
-                    if context:
-                        citations = re.findall(r'\[(\d+)\]', complete_response)
-                        citation_numbers = list(map(int, citations))
-                        sources = dict()
-                        for index, doc in enumerate(context):
-                            if (index+1) in citation_numbers:
-                                sources[f"[{index+1}]"] = doc.metadata["source"]
-                        await websocket.send_json({'sources': sources})
-
                 await stream_response()
             except Exception as e:
+                print(traceback.format_exc())
                 print(f"Error during message handling: {e}")
                 await websocket.send_json({'response': "Something went wrong, Please try again." + str(e)})
     except WebSocketDisconnect: