Update app/models/schemas.py

app/models/schemas.py

@@ -1,12 +1,20 @@
-from pydantic import BaseModel
-from typing import List
-
-class QuestionRequest(BaseModel):
-    question: str
-
-class AyahResponse(BaseModel):
-    question: str
-    answer: str
-
-class AnswerListResponse(BaseModel):
-    results: List[AyahResponse]
+from pydantic import BaseModel
+from typing import List
+
+
+class QuestionRequest(BaseModel):
+    question: str
+
+
+class ReflectionResponse(BaseModel):
+    reflection: str
+
+
+class AyahResponse(BaseModel):
+    question: str
+    answer: str
+
+
+class AnswerListResponse(BaseModel):
+    reflection: ReflectionResponse
+    ayahs: List[AyahResponse]