Brushed up the backend, added writing task 1 academic prompt gen and grading ENCOA-274

ielts_be/controllers/impl/exam/__init__.py

@@ -0,0 +1,13 @@
+from .level import LevelController
+from .listening import ListeningController
+from .reading import ReadingController
+from .speaking import SpeakingController
+from .writing import WritingController
+
+__all__ = [
+    "LevelController",
+    "ListeningController",
+    "ReadingController",
+    "SpeakingController",
+    "WritingController",
+]

ielts_be/controllers/impl/exam/level.py

@@ -0,0 +1,26 @@
+from fastapi import UploadFile
+from typing import Dict, Optional
+
+from ielts_be.controllers import ILevelController
+from ielts_be.services import ILevelService
+
+
+class LevelController(ILevelController):
+
+    def __init__(self, level_service: ILevelService):
+        self._service = level_service
+
+    async def generate_exercises(self, dto):
+        return await self._service.generate_exercises(dto)
+
+    async def get_level_exam(self):
+        return await self._service.get_level_exam()
+
+    async def get_level_utas(self):
+        return await self._service.get_level_utas()
+
+    async def upload_level(self, exercises: UploadFile, solutions: Optional[UploadFile] = None):
+        return await self._service.upload_level(exercises, solutions)
+
+    async def get_custom_level(self, data: Dict):
+        return await self._service.get_custom_level(data)

ielts_be/controllers/impl/exam/listening.py

@@ -0,0 +1,39 @@
+import io
+
+from fastapi import UploadFile
+from starlette.responses import StreamingResponse, Response
+
+from ielts_be.controllers import IListeningController
+from ielts_be.services import IListeningService
+from ielts_be.dtos.listening import GenerateListeningExercises, Dialog
+
+
+class ListeningController(IListeningController):
+
+    def __init__(self, listening_service: IListeningService):
+        self._service = listening_service
+
+    async def import_exam(self, exercises: UploadFile, solutions: UploadFile = None):
+        res = await self._service.import_exam(exercises, solutions)
+        if not res:
+            return Response(status_code=500)
+        else:
+            return res
+
+    async def generate_listening_dialog(self, section_id: int, topic: str, difficulty: str):
+        return await self._service.generate_listening_dialog(section_id, topic, difficulty)
+
+    async def get_listening_question(self, dto: GenerateListeningExercises):
+        return await self._service.get_listening_question(dto)
+
+    async def generate_mp3(self, dto: Dialog):
+        mp3 = await self._service.generate_mp3(dto)
+
+        return StreamingResponse(
+            content=io.BytesIO(mp3),
+            media_type="audio/mpeg",
+            headers={
+                "Content-Type": "audio/mpeg",
+                "Content-Disposition": "attachment;filename=speech.mp3"
+            }
+        )

ielts_be/controllers/impl/exam/reading.py

@@ -0,0 +1,28 @@
+import logging
+from typing import Optional
+
+from fastapi import UploadFile, Response
+
+from ielts_be.controllers import IReadingController
+from ielts_be.services import IReadingService
+from ielts_be.dtos.reading import ReadingDTO
+
+
+class ReadingController(IReadingController):
+
+    def __init__(self, reading_service: IReadingService):
+        self._service = reading_service
+        self._logger = logging.getLogger(__name__)
+
+    async def import_exam(self, exercises: UploadFile, solutions: UploadFile = None):
+        res = await self._service.import_exam(exercises, solutions)
+        if not res:
+            return Response(status_code=500)
+        else:
+            return res
+
+    async def generate_reading_passage(self, passage: int, topic: Optional[str], word_count: Optional[int]):
+        return await self._service.generate_reading_passage(passage, topic, word_count)
+
+    async def generate_reading_exercises(self, dto: ReadingDTO):
+        return await self._service.generate_reading_exercises(dto)

ielts_be/controllers/impl/exam/speaking.py

@@ -0,0 +1,24 @@
+import logging
+
+from ielts_be.controllers import ISpeakingController
+from ielts_be.services import ISpeakingService, IVideoGeneratorService
+
+
+class SpeakingController(ISpeakingController):
+
+    def __init__(self, speaking_service: ISpeakingService, vid_gen: IVideoGeneratorService):
+        self._service = speaking_service
+        self._vid_gen = vid_gen
+        self._logger = logging.getLogger(__name__)
+
+    async def get_speaking_part(self, task: int, topic: str, second_topic: str, difficulty: str):
+        return await self._service.get_speaking_part(task, topic, second_topic, difficulty)
+
+    async def get_avatars(self):
+        return await self._vid_gen.get_avatars()
+
+    async def generate_video(self, text: str, avatar: str):
+        return await self._vid_gen.create_video(text, avatar)
+
+    async def poll_video(self, vid_id: str):
+        return await self._vid_gen.poll_status(vid_id)

ielts_be/controllers/impl/exam/writing.py

@@ -0,0 +1,19 @@
+from fastapi import UploadFile, HTTPException
+
+from ielts_be.controllers import IWritingController
+from ielts_be.services import IWritingService
+
+
+class WritingController(IWritingController):
+
+    def __init__(self, writing_service: IWritingService):
+        self._service = writing_service
+
+    async def get_writing_task_general_question(self, task: int, topic: str, difficulty: str):
+        return await self._service.get_writing_task_general_question(task, topic, difficulty)
+
+    async def get_writing_task_academic_question(self, task: int, attachment: UploadFile, difficulty: str):
+        if attachment.content_type not in ['image/jpeg', 'image/png']:
+            raise HTTPException(status_code=400, detail="Invalid file type. Only JPEG and PNG allowed.")
+
+        return await self._service.get_writing_task_academic_question(task, attachment, difficulty)