Create app.py

app.py

@@ -0,0 +1,165 @@
+from fastapi import FastAPI, HTTPException, UploadFile, File, Form
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse
+import json
+import hashlib
+from datetime import datetime, timedelta
+import re
+import logging
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# Initialize FastAPI app
+app = FastAPI(
+    title="StudyLoop API",
+    version="1.0.0",
+    description="AI-powered study question generator"
+)
+
+# CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Simple cache
+cache = {}
+
+# Health endpoints
+@app.get("/")
+async def root():
+    return {"message": "StudyLoop API", "status": "healthy"}
+
+@app.get("/health")
+async def health():
+    return {"status": "healthy", "timestamp": datetime.now().isoformat()}
+
+# Generate mock questions
+def generate_questions(count=5):
+    questions = []
+    for i in range(count):
+        if i % 2 == 0:
+            questions.append({
+                "id": f"q_{i}",
+                "type": "multiple_choice",
+                "question": "What is the main topic?",
+                "options": ["Option A", "Option B", "Option C", "Option D"],
+                "correct_answer": "Option A",
+                "explanation": "Based on the text content.",
+                "difficulty": "easy",
+                "concept": "Main Topic"
+            })
+        else:
+            questions.append({
+                "id": f"q_{i}",
+                "type": "short_answer",
+                "question": "Explain the key concept.",
+                "correct_answer": "The key concept is explained in the text.",
+                "explanation": "Reference the source material.",
+                "difficulty": "medium",
+                "concept": "Key Concepts"
+            })
+    return questions
+
+# Main API endpoint
+@app.post("/api/generate")
+async def generate_questions_endpoint(
+    text: str = Form(None),
+    file: UploadFile = File(None),
+    session_id: str = Form(...),
+    options: str = Form('{"num_questions": 5}')
+):
+    try:
+        # Parse options
+        try:
+            opts = json.loads(options)
+            num_questions = min(opts.get("num_questions", 5), 10)
+        except:
+            num_questions = 5
+        
+        content = ""
+        
+        # Get content
+        if text:
+            content = text
+        elif file:
+            file_bytes = await file.read()
+            if len(file_bytes) > 5 * 1024 * 1024:
+                raise HTTPException(400, "File too large (max 5MB)")
+            try:
+                content = file_bytes.decode('utf-8')
+            except:
+                content = file_bytes.decode('latin-1')
+        else:
+            raise HTTPException(400, "Provide text or file")
+        
+        # Clean content
+        content = re.sub(r'\s+', ' ', content).strip()
+        
+        if len(content) < 10:
+            raise HTTPException(400, "Text too short")
+        
+        # Generate questions
+        questions = generate_questions(num_questions)
+        
+        # Extract concepts
+        concepts = list(set([q.get("concept", "General") for q in questions]))
+        
+        response = {
+            "session_id": session_id,
+            "concepts": concepts,
+            "questions": questions,
+            "generated_at": datetime.utcnow().isoformat(),
+            "question_count": len(questions)
+        }
+        
+        return JSONResponse(response)
+        
+    except HTTPException:
+        raise
+    except Exception as e:
+        logger.error(f"Error: {e}")
+        raise HTTPException(500, f"Internal error: {str(e)}")
+
+# Generate more questions
+@app.post("/api/generate-more")
+async def generate_more(
+    session_id: str = Form(...),
+    concepts: str = Form("[]")
+):
+    try:
+        # Generate 3 more questions
+        questions = generate_questions(3)
+        for q in questions:
+            q["is_followup"] = True
+        
+        return {
+            "session_id": session_id,
+            "questions": questions,
+            "generated_at": datetime.utcnow().isoformat()
+        }
+    except Exception as e:
+        raise HTTPException(500, str(e))
+
+# API info
+@app.get("/api/info")
+async def api_info():
+    return {
+        "name": "StudyLoop API",
+        "version": "1.0.0",
+        "endpoints": [
+            {"path": "/api/generate", "method": "POST"},
+            {"path": "/api/generate-more", "method": "POST"},
+            {"path": "/health", "method": "GET"},
+            {"path": "/api/info", "method": "GET"}
+        ]
+    }
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)