feat: add backend

main.py

@@ -212,9 +212,21 @@ async def websocket_endpoint(websocket: WebSocket,  client_id: str):
 
     try:
         while True:
-            prompt = await websocket.receive_text()
-            response = QA(inputs=prompt, return_only_outputs=True, tags=f'{client_id}', include_run_info=True)
-            await websocket.send_text(f'{response}')
+            user_prompt = await websocket.receive_text()
+            response = QA(inputs=user_prompt, return_only_outputs=True, tags=f'{client_id}', include_run_info=True)
+            answer, docs = response["result"], response["source_documents"]
+
+            prompt_response_dict = {
+                "Prompt": user_prompt,
+                "Answer": answer,
+            }
+
+            prompt_response_dict["Sources"] = []
+            for document in docs:
+                prompt_response_dict["Sources"].append(
+                    (os.path.basename(str(document.metadata["source"])), str(document.page_content))
+                )
+            await websocket.send_json(prompt_response_dict)
 
     except WebSocketDisconnect:
         print('disconnect')