feat: add v1

main.py

@@ -55,7 +55,7 @@ QA = RetrievalQA.from_chain_type(
     },
 )
 
-def sendPromptChain(QA, user_prompt):
+def sendPromptChain(QA, user_prompt, isRoom=False):
 
     res = QA(user_prompt)
 
@@ -64,6 +64,7 @@ def sendPromptChain(QA, user_prompt):
     prompt_response_dict = {
         "Prompt": user_prompt,
         "Answer": answer,
+        "Room": isRoom
     }
 
     prompt_response_dict["Sources"] = []
@@ -74,11 +75,8 @@ def sendPromptChain(QA, user_prompt):
 
     return prompt_response_dict;
 
-
 socket_manager = WebSocketManager()
 
-
-
 app = FastAPI(title="homepage-app")
 api_app = FastAPI(title="api app")
 
@@ -247,19 +245,18 @@ async def websocket_endpoint_student(websocket: WebSocket, user_id: str):
 async def websocket_endpoint_room(websocket: WebSocket, room_id: str, user_id: str):
     global QA
 
-    await socket_manager.add_user_to_room(room_id, websocket)
-
     message = {
         "message": f"Student {user_id} connected to the classroom"
     }
 
+    await socket_manager.add_user_to_room(room_id, websocket)
     await socket_manager.broadcast_to_room(room_id, json.dumps(message))
 
     try:
         while True:
             data = await websocket.receive_text()
 
-            prompt_response_dict = sendPromptChain(QA, data)
+            prompt_response_dict = sendPromptChain(QA, data, True)
 
             await socket_manager.broadcast_to_room(room_id, json.dumps(prompt_response_dict))