Update app.py

app.py

@@ -4,15 +4,16 @@ from fastapi.middleware.cors import CORSMiddleware
 import os
 import google.generativeai as genai
 from typing import List, Optional
+from uuid import uuid4
 
-# Konfigurasi API Gemini
+# Konfigurasi Gemini API
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 model = genai.GenerativeModel("gemini-1.5-flash")
 
-# Inisialisasi FastAPI
+# Init FastAPI
 app = FastAPI()
 
-# Setup CORS
+# CORS biar frontend bebas akses
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -21,21 +22,54 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Model request
+# === MODEL UNTUK REQUEST ===
 class PromptRequest(BaseModel):
     prompt: str
     max_new_tokens: int = 300
     temperature: float = 0.7
     history: Optional[List[str]] = []
 
-# Endpoint untuk generate teks
+class GenerateSoalRequest(BaseModel):
+    topic: str
+    level: Optional[int] = 1
+    tipe_pertanyaan: Optional[str] = None  # "pilihan_ganda", "isian_kosong", "benar_salah"
+
+# === HELPER FUNCTION ===
+def format_prompt_soal(topic, tipe=None, level=1):
+    tipe_instruksi = f"Tipe soal yang diminta adalah '{tipe}'." if tipe else "Tipe soal boleh bebas."
+    return f"""
+Buatkan satu soal latihan grammar Bahasa Inggris untuk topik: "{topic}".
+{tipe_instruksi}
+Level kesulitan: {level}/5.
+
+Format keluaran harus dalam JSON seperti berikut:
+{{
+  "text_pertanyaan": "Tulis pertanyaan di sini...",
+  "tipe_pertanyaan": "pilihan_ganda | isian_kosong | benar_salah",
+  "opsi": {{
+    "a": "Pilihan A",
+    "b": "Pilihan B",
+    "c": "Pilihan C",
+    "d": "Pilihan D"
+  }},
+  "jawaban_benar": "b",
+  "penjelasan": "Berikan penjelasan edukatif tentang jawaban yang benar."
+}}
+
+Catatan:
+- Jika tipe soal bukan "pilihan_ganda", sesuaikan struktur 'opsi' (bisa null atau objek true/false).
+- Jangan tambahkan teks selain JSON.
+- Jangan tambahkan penjelasan tambahan di luar JSON.
+
+Mulai sekarang, buat 1 soal untuk topik di atas dan jawab dalam format JSON!
+"""
+
+# === ENDPOINT UNTUK PERBAIKAN KELUARAN MODEL (EXISTING) ===
 @app.post("/generate")
 async def generate_text(req: PromptRequest):
     try:
-        # Bentuk ulang system prompt terstruktur
         system_prompt = f"""
 Berikut ini adalah hasil dari model LLaMA:
-
 --- MULAI ---
 {req.prompt}
 --- SELESAI ---
@@ -43,33 +77,16 @@ Berikut ini adalah hasil dari model LLaMA:
 Tugas kamu adalah:
 1. Perbaiki tata bahasa dan struktur kalimat jika perlu.
 2. Ubah seluruh jawaban ke dalam Bahasa Indonesia dengan gaya bahasa edukatif dan mudah dipahami.
-3. Jika ada terdapat contoh kalimat yang di sertakan, jangan terjemahkan itu.
-4. Jangan terjemahkan kalimat atau kata yang ada tanda petik.
-5. Jika ada konsep konsep yang berhubungan dengan tata bahasa inggris jangan di terjemahkan.
-6. Jika jawaban tidak relevan atau terlalu pendek, lengkapi agar lebih informatif dan sesuai konteks grammar bahasa Inggris.
-
-Jangan ubah maksud dari jawaban, hanya sempurnakan dan terjemahkan.
-Jawab langsung tanpa menjelaskan prosesnya.
+3. Jangan terjemahkan contoh kalimat, kata yang diapit tanda petik, atau istilah grammar.
+4. Jika jawaban tidak relevan, lengkapi sesuai konteks grammar.
+Langsung berikan hasil perbaikan. Jangan jelaskan prosesnya.
 """
-
-        # Buat format pesan
-        messages = [
-            {
-                "role": "user",
-                "parts": [{"text": system_prompt.strip()}],
-            }
-        ]
-
-        # Tambahkan history jika ada
+        messages = [{"role": "user", "parts": [{"text": system_prompt.strip()}]}]
         for msg in req.history or []:
             role = "user" if msg.startswith("User:") else "model"
             text = msg.replace("User: ", "").replace("Bot: ", "")
-            messages.append({
-                "role": role,
-                "parts": [{"text": text}]
-            })
+            messages.append({"role": role, "parts": [{"text": text}]})
 
-        # Kirim ke Gemini
         result = model.generate_content(
             messages,
             generation_config={
@@ -77,13 +94,39 @@ Jawab langsung tanpa menjelaskan prosesnya.
                 "max_output_tokens": req.max_new_tokens,
             }
         )
-
         return {"generated_text": result.text}
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-# Root endpoint
+
+# === ENDPOINT BARU UNTUK GENERATE SOAL DARI AI ===
+@app.post("/generate-soal")
+async def generate_soal(req: GenerateSoalRequest):
+    try:
+        prompt = format_prompt_soal(req.topic, req.tipe_pertanyaan, req.level)
+        result = model.generate_content(
+            prompt,
+            generation_config={"temperature": 0.8, "max_output_tokens": 512}
+        )
+
+        # Bersihkan & parsing response (anggap Gemini kasih string JSON langsung)
+        import json
+        json_text = result.text.strip().split("```")[-1].strip()  # Bersihin kalau ada blok kode
+        soal_data = json.loads(json_text)
+
+        soal_data["id"] = str(uuid4())
+        soal_data["topic_id"] = req.topic
+        soal_data["level"] = req.level or 1
+        soal_data["is_ai_generated"] = True
+        soal_data["created_at"] = str(os.getenv("TZ") or "2025-07-12T00:00:00Z")
+
+        return {"soal": soal_data}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Gagal generate soal: {e}")
+
+
+# === ROOT TEST ===
 @app.get("/")
 async def root():
     return {"message": "Gemini endpoint aktif bro! 🚀"}