Upload workbook_server.py

workbook_server.py

@@ -1,8 +1,9 @@
 import uvicorn
 from contextlib import asynccontextmanager
 import sqlite3
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException
 from fastapi.responses import JSONResponse
+from pydantic import BaseModel
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
@@ -29,5 +30,19 @@ def get_chapters():
     chapter_list = [{"chapter_number": row[0], "chapter_name": row[1]} for row in chapters]
     return JSONResponse(content=chapter_list)
 
+class ChapterRequest(BaseModel):
+    chapter_number: int
+
+@app.post('/api/chapter/questions')
+def get_questions_by_chapter(request: ChapterRequest):
+    db = get_db()
+    cursor = db.cursor()
+    cursor.execute("SELECT question_id, question_text FROM questions WHERE chapter_number = ?", (request.chapter_number,))
+    questions = cursor.fetchall()
+    if not questions:
+        raise HTTPException(status_code=404, detail="Questions not found for the given chapter number")
+    question_list = [{"question_id": row[0], "question_text": row[1]} for row in questions]
+    return JSONResponse(content=question_list)
+
 if __name__ == '__main__':
     uvicorn.run(app, host="127.0.0.1", port=8000, log_level="debug")