Update app/main.py

app/main.py

@@ -12,10 +12,11 @@ CEREBRAS_API_KEY = os.getenv("CEREBRAS_API_KEY")
 CEREBRAS_API_URL = "https://api.cerebras.ai/v1/chat/completions"
 
 
-def build_prompt(difficulty: str) -> str:
+def build_prompt(difficulty: str, language: str) -> str:
     return f"""
 あなたはクイズ作成者です。
 難易度: {difficulty}
+言語: {language}
 
 次のフォーマットの JSON だけを返してください（前後に一切の文章を付けないこと）:
 
@@ -44,6 +45,7 @@ def build_prompt(difficulty: str) -> str:
 - 各問題は4択
 - テーマは一般教養レベル（雑学など）で OK
 - 難易度は easy, medium, hard に応じて調整
+- 最終的なアウトプットは {language}に合わせて出力
 """
 
 
@@ -98,7 +100,8 @@ async def call_cerebras_quiz_api(difficulty: str) -> Dict[str, Any]:
 
 @app.get("/quiz")
 async def get_quiz(
-    difficulty: Literal["easy", "medium", "hard"] = Query("medium")
+    difficulty: Literal["easy", "medium", "hard"] = Query("medium"),
+    language: str = Query("en")
 ):
     """
     クイズを3問返すAPI。