Update app/main.py

app/main.py

@@ -1,41 +1,143 @@
-import io
-import traceback
+import os
+import json
+from typing import Literal, Dict, Any
 
-from fastapi import FastAPI, HTTPException, status
-from fastapi.concurrency import run_in_threadpool
-from pydantic import BaseModel
-from opentrons.simulate import simulate
+import httpx
+from fastapi import FastAPI, Query, HTTPException
+from fastapi.responses import JSONResponse
 
 app = FastAPI()
 
-class Protocol(BaseModel):
-    name: str
-    content: str
+CEREBRAS_API_KEY = os.getenv("CEREBRAS_API_KEY")
+CEREBRAS_API_URL = "https://api.cerebras.ai/v1/chat/completions"
 
 
-@app.get("/")
-async def root():
-    return {"message": "Opentrons simulation API is running."}
+def build_prompt(difficulty: str) -> str:
+    return f"""
+あなたはクイズ作成者です。
+難易度: {difficulty}
+
+次のフォーマットの JSON だけを返してください（前後に一切の文章を付けないこと）:
+
+{{
+  "questions": [
+    {{
+      "question": "問題文（日本語）",
+      "choices": ["選択肢1", "選択肢2", "選択肢3", "選択肢4"],
+      "correctIndex": 0
+    }},
+    {{
+      "question": "問題文（日本語）",
+      "choices": ["選択肢1", "選択肢2", "選択肢3", "選択肢4"],
+      "correctIndex": 0
+    }},
+    {{
+      "question": "問題文（日本語）",
+      "choices": ["選択肢1", "選択肢2", "選択肢3", "選択肢4"],
+      "correctIndex": 0
+    }}
+  ]
+}}
+
+制約:
+- 質問は必ず3問
+- 各問題は4択
+- テーマは一般教養レベル（雑学など）で OK
+- 難易度は easy, medium, hard に応じて調整
+"""
+
+
+async def call_cerebras_quiz_api(difficulty: str) -> Dict[str, Any]:
+    if not CEREBRAS_API_KEY:
+        raise RuntimeError("CEREBRAS_API_KEY is not set")
+
+    prompt = build_prompt(difficulty)
+
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {CEREBRAS_API_KEY}",
+    }
+
+    payload = {
+        "model": "gpt-oss-120b",
+        "messages": [
+            {"role": "system", "content": "You are a helpful quiz generator."},
+            {"role": "user", "content": prompt},
+        ],
+        "temperature": 0.7,
+        "max_tokens": 512,
+    }
+
+    async with httpx.AsyncClient(timeout=30.0) as client:
+        resp = await client.post(CEREBRAS_API_URL, headers=headers, json=payload)
+        if resp.status_code != 200:
+            raise RuntimeError(f"Cerebras API error: {resp.status_code}, {resp.text}")
+
+        data = resp.json()
+
+    content = data.get("choices", [{}])[0].get("message", {}).get("content", "")
+
+    # たまにモデルが ```json ... ``` で囲んでくる可能性があるので軽く掃除
+    cleaned = content.strip()
+    if cleaned.startswith("```"):
+        # ```json or ``` で始まるケースに対応
+        cleaned = cleaned.strip("`")
+        # 先頭行に "json" とか残っていたら落とす
+        if cleaned.startswith("json"):
+            cleaned = cleaned[len("json") :].lstrip()
 
-@app.post("/simulate")
-async def simulate_protocol(protocol: Protocol):
     try:
-        await run_in_threadpool(
-            simulate,
-            protocol_file=io.StringIO(protocol.content),
-            file_name=protocol.name
-        )
-        return {
-            "protocol_name": protocol.name,
-            "status": "success"
-        }
+        quiz = json.loads(cleaned)
+    except json.JSONDecodeError as e:
+        # デバッグ用に content を見たい場合は log に出す
+        print("Failed to parse JSON from model:", content)
+        raise RuntimeError(f"JSON parse error: {e}")
+
+    return quiz
+
+
+@app.get("/quiz")
+async def get_quiz(
+    difficulty: Literal["easy", "medium", "hard"] = Query("medium")
+):
+    """
+    You are the master of quiz. You offer 3 quizzes to a user by the difficulty that user posts (easy, medium, hard).
+    You will return  3 quizzes with the following JSON format. The following is just sample format.
+
+    format
+    ```
+    {
+      "quizzes": [
+        {
+          "id": 1,
+          "question": "question 1",
+          "choices": ["chice1", "chice2", "chice3", "chice4"],
+          "answer": "choice2"
+        },
+          {
+          "id": 2,
+          "question": "question 2",
+          "choices": ["chice1", "chice2", "chice3", "chice4"],
+          "answer": "choice4"
+        },
+          {
+          "id": 3,
+          "question": "question 3",
+          "choices": ["chice1", "chice2", "chice3", "chice4"],
+          "answer": "choice3"
+        },
+      ]
+    }
+    ```
+    """
+    try:
+        quiz = await call_cerebras_quiz_api(difficulty)
     except Exception as e:
-        raise HTTPException(
-            status_code=status.HTTP_400_BAD_REQUEST,
-            detail={
-                "status": "error",
-                "error_type": type(e).__name__,
-                "message": str(e),
-                "traceback": traceback.format_exc()
-            }
-        )
+        # 拡張側が扱いやすいように 500 + メ���セージだけ返す
+        raise HTTPException(status_code=500, detail=str(e))
+
+    # 最低限のバリデーション（壊れたJSONでないかチェック）
+    if not isinstance(quiz, dict) or "questions" not in quiz:
+        raise HTTPException(status_code=500, detail="Invalid quiz format")
+
+    return JSONResponse(content=quiz)