Fixed feedback

feedback.csv

@@ -0,0 +1,2 @@
+question,botResponse,userFeedback,feedback
+hi,Hello! How can I assist you today?,SUIII,Good

reggpt/api/router.py

@@ -4,11 +4,11 @@ from reggpt.configs.logger import logger
 from fastapi import  APIRouter, HTTPException, status
 from fastapi import HTTPException, status
 
-from reggpt.schemas.schema import UserQuery, LoginRequest, UserModel
+from reggpt.schemas.schema import UserQuery, LoginRequest, UserModel,Feedback
 from reggpt.routers.controller import get_QA_Answers, get_avaliable_models
-from reggpt.configs.api import API_ENDPOINT_LOGIN,API_ENDPOINT_CHAT, API_ENDPOINT_HEALTH, API_ENDPOINT_MODEL
+from reggpt.configs.api import API_ENDPOINT_LOGIN,API_ENDPOINT_CHAT, API_ENDPOINT_HEALTH, API_ENDPOINT_MODEL,API_ENDPOINT_FEEDBACK
 
-# logger = logging.getLogger(__name__)
+import csv, os
 
 class ChatAPI:
 
@@ -20,6 +20,9 @@ class ChatAPI:
             API_ENDPOINT_LOGIN, self.login, methods=["POST"], response_model=UserModel
         )
         self.router.add_api_route(API_ENDPOINT_CHAT, self.chat, methods=["POST"])
+        self.router.add_api_route(
+            API_ENDPOINT_FEEDBACK, self.collect_feedback, methods=["POST"]
+        )
 
     async def hello(self):
         print(API_ENDPOINT_HEALTH)
@@ -106,3 +109,20 @@ class ChatAPI:
             logger.exception(e)
             raise HTTPException(status_code=400, detail=f"Error : {e}")
 
+    async def collect_feedback(self, feedback: Feedback):
+        """
+        Route to collect feedback from the user and store it in a CSV file.
+        """
+        filename = "feedback.csv"
+        fieldnames = ["question", "botResponse", "userFeedback", "feedback"]
+
+        if not os.path.exists(filename):
+            with open(filename, mode="w", newline="") as file:
+                writer = csv.DictWriter(file, fieldnames=fieldnames)
+                writer.writeheader()
+
+        with open(filename, mode="a", newline="") as file:
+            writer = csv.DictWriter(file, fieldnames=fieldnames)
+            writer.writerow(feedback.dict())
+
+        return {"message": "Feedback received successfully"}

reggpt/configs/api.py

@@ -25,4 +25,5 @@ API_OPENAPI_URL = f"{API_ENDPOINT_PREFIX}openapi.json"
 API_ENDPOINT_HEALTH = f"{API_ENDPOINT_PREFIX}health"
 API_ENDPOINT_CHAT = f"{API_ENDPOINT_PREFIX}chat"
 API_ENDPOINT_MODEL = f"{API_ENDPOINT_PREFIX}models"
-API_ENDPOINT_LOGIN = f"{API_ENDPOINT_PREFIX}login"
+API_ENDPOINT_LOGIN = f"{API_ENDPOINT_PREFIX}login"
+API_ENDPOINT_FEEDBACK = f"{API_ENDPOINT_PREFIX}feedback"

reggpt/schemas/schema.py

@@ -54,13 +54,14 @@ class ResponseModel(BaseModel):
     source_documents: List[Document] = None
 
 
-# class Feedback(BaseModel):
-#     """
-#     Schema for collecting feedback from the user.
-#     It includes the question, bot response, and user feedback.
-#     """
 
-#     question: str
-#     botResponse: str
-#     userFeedback: str
-#     feedback: str
+class Feedback(BaseModel):
+    """
+    Schema for collecting feedback from the user.
+    It includes the question, bot response, and user feedback.
+    """
+
+    question: str
+    botResponse: str
+    userFeedback: str
+    feedback: str