Upload workbook_server.py

workbook_server.py

@@ -4,6 +4,12 @@ import sqlite3
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel
+from openai import OpenAI
+import os
+
+# OpenAI APIクライアントを初期化
+client = OpenAI()
+client.api_key = os.getenv('OPENAI_API_KEY')
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
@@ -62,5 +68,28 @@ def get_questions_by_chapter(request: ChapterRequest):
     ]
     return JSONResponse(content=question_list)
 
+class HintRequest(BaseModel):
+    question_text: str
+    options: str
+    answer: str
+
+@app.post('/api/hint')
+def generate_hint(request: HintRequest):
+    prompt = f"設問: {request.question_text}\n選択肢: {request.options}\n正解: {request.answer}\nこの設問に対するヒントを生成してください。"
+    try:
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {
+                    "role": "user",
+                    "content": [{'type': 'text', 'text': prompt}],
+                },
+            ],
+        )
+        response_content = response.choices[0].message.content
+        return JSONResponse(content={"hint": response_content})
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
 if __name__ == '__main__':
     uvicorn.run(app, host="127.0.0.1", port=8000, log_level="debug")