Update timetable creator

app.py

@@ -1,4 +1,4 @@
-# Access site: https://binkhoale1812-tutorbot.hf.space/chat 
+# Access site: https://binkhoale1812-tutorbot.hf.space 
 import os
 import time
 import uvicorn
@@ -13,13 +13,13 @@ from fastapi.middleware.cors import CORSMiddleware
 from google import genai
 from gradio_client import Client, handle_file
 
-# —————— Logging ——————
+# ———————— Logging —————————
 logging.basicConfig(level=logging.DEBUG, format="%(asctime)s — %(name)s — %(levelname)s — %(message)s", force=True)
 logger = logging.getLogger("tutor-chatbot")
 logger.setLevel(logging.DEBUG)
 logger.info("🚀 Starting Tutor Chatbot API...")
 
-# —————— Environment ——————
+# —————— Environment ———————
 gemini_flash_api_key = os.getenv("FlashAPI")
 if not gemini_flash_api_key:
     raise ValueError("❌ Missing Gemini Flash API key!")
@@ -35,7 +35,7 @@ check_system_resources()
 os.environ["OMP_NUM_THREADS"] = "1"
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
-# —————— FastAPI Setup ——————
+# —————— FastAPI Setup —————
 app = FastAPI(title="Tutor Chatbot API")
 app.add_middleware(
     CORSMiddleware,
@@ -59,7 +59,7 @@ def gemini_flash_completion(prompt, model="gemini-2.5-flash-preview-04-17", temp
         logger.error(f"❌ Gemini error: {e}")
         return "Error generating response from Gemini."
 
-# —————— Qwen 2.5 VL Client Setup ——————
+# —— Qwen 2.5 VL Client Setup —————
 qwen_client = Client("prithivMLmods/Qwen2.5-VL-7B-Instruct")
 logger.info("[Qwen] Using remote API via Gradio Client")
 
@@ -120,7 +120,7 @@ def qwen_image_summary(image_file: UploadFile, subject: str, level: str) -> str:
         raise HTTPException(500, "❌ Qwen image analysis failed")
 
 
-# —————— Unified Chat Endpoint ——————
+# ————— Unified Chat Endpoint —————
 @app.post("/chat")
 async def chat_endpoint(
     query: str = Form(""),
@@ -205,7 +205,88 @@ async def chat_endpoint(
     response_text += f"\n\n*(Response time: {end_time - start_time:.2f} seconds)*"
     return JSONResponse(content={"response": response_text})
 
-# —————— Launch Server ——————
+
+# ————— Clsr Pydantic Schema —————
+from pydantic import BaseModel, Field, validator
+from typing import Optional, List, Literal
+# Dynamic cls
+class StudyPreferences(BaseModel):
+    daysPerWeek: int = Field(..., ge=1, le=7)
+    hoursPerSession: float = Field(..., ge=0.5, le=4)
+    learningStyle: Literal["step-by-step", "conceptual", "visual"]
+# Dynamic cls
+class ClassroomRequest(BaseModel):
+    id: str
+    name: str = Field(..., min_length=2)
+    role: Literal["tutor", "student"]
+    subject: str
+    gradeLevel: str
+    textbookUrl: Optional[str] = None
+    syllabusUrl: Optional[str] = None
+    studyPreferences: StudyPreferences
+
+# —————— Time table creator ——————
+import json
+from fastapi import Body
+@app.post("/api/classrooms")
+async def create_classroom(payload: ClassroomRequest = Body(...)):
+    """
+    Generate a detailed study timetable based on classroom parameters.
+    """
+    # ---------- Build prompt for Gemini 2.5 ----------
+    prefs = payload.studyPreferences
+    prompt = f"""
+        You are an expert academic coordinator.
+
+        Create a **4-week study timetable** for a classroom with the following settings:
+
+        - Subject: {payload.subject}
+        - Grade level: {payload.gradeLevel}
+        - Study days per week: {prefs.daysPerWeek}
+        - Hours per session: {prefs.hoursPerSession}
+        - Preferred learning style: {prefs.learningStyle}
+        - Role perspective: {payload.role}
+        {"- Textbook URL: " + payload.textbookUrl if payload.textbookUrl else ""}
+        {"- Syllabus URL: " + payload.syllabusUrl if payload.syllabusUrl else ""}
+
+        Requirements:
+
+        1. Divide each week into exactly {prefs.daysPerWeek} sessions (label Day 1 … Day {prefs.daysPerWeek}).
+
+        2. For **each session**, return:
+           - `week` (1-4)
+           - `day` (1-{prefs.daysPerWeek})
+           - `durationHours` (fixed: {prefs.hoursPerSession})
+           - `topic` (max 15 words)
+           - `activities` (array of 2-3 bullet strings)
+           - `materials` (array of links/titles; include textbook chapters if URL given)
+           - `homework` (concise task ≤ 30 words)
+
+        3. **Output pure JSON only** using the schema:
+
+        ```json
+        {{
+          "classroom_id": "<same id as request>",
+          "timetable": [{{session objects as listed}}]
+        }}
+
+        Do not wrap JSON in markdown fences or commentary.
+        """
+    raw = gemini_flash_completion(prompt).strip()
+
+# ---------- Attempt to parse JSON ----------
+try:
+    timetable_json = json.loads(raw)
+except json.JSONDecodeError:
+    logger.warning("Gemini returned invalid JSON; sending raw text.")
+    return JSONResponse(content={"classroom_id": payload.id, "timetable_raw": raw})
+# Ensure id is echoed (fallback if model forgot)
+timetable_json["classroom_id"] = payload.id
+return JSONResponse(content=timetable_json)
+
+
+
+# —————— Launch Server ———————
 if __name__ == "__main__":
     logger.info("✅ Launching FastAPI server...")
     try: