Upload workbook_server.py

workbook_server.py

@@ -1,3 +1,4 @@
+import base64
 import uvicorn
 from contextlib import asynccontextmanager
 import sqlite3
@@ -54,19 +55,22 @@ class ChapterRequest(BaseModel):
 def get_questions_by_chapter(request: ChapterRequest):
     db = get_db()
     cursor = db.cursor()
-    cursor.execute("SELECT question, options, answer, explain, image FROM question WHERE chapter = ? AND kind = '正誤問題'", (request.chapter_number,))
+    cursor.execute("SELECT question, options, answer, explain FROM question WHERE chapter = ? AND kind = '正誤問題'", (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_list = []
+    for row in questions:
+        image_path = os.path.join('./image', row[4])
+        with open(image_path, "rb") as image_file:
+            encoded_string = base64.b64encode(image_file.read()).decode('utf-8')
+        question_list.append({
             "question_text": row[0],
             "options": row[1],
             "answer": row[2],
             "explanation": row[3],
-            "image": row[4],
-        } for row in questions
-    ]
+            # "image": encoded_string,
+        })
     return JSONResponse(content=question_list)
 
 class HintRequest(BaseModel):