updated model.py

app.py

@@ -1,20 +1,26 @@
 import uvicorn
 from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
 from model import generate_code
 
 app = FastAPI()
 
+# Define request body schema
+class GenerateRequest(BaseModel):
+    prompt: str
+    max_tokens: int = 256  # Default value
+
 @app.get("/")
 def home():
     return {"message": "Code Generation API is running!"}
 
 @app.post("/generate")
-def generate(prompt: str, max_tokens: int = 256):
-    if not prompt:
+def generate(request: GenerateRequest):  # Expect JSON
+    if not request.prompt:
         raise HTTPException(status_code=400, detail="Prompt cannot be empty.")
 
     try:
-        code = generate_code(prompt, max_tokens)
+        code = generate_code(request.prompt, request.max_tokens)
         return {"generated_code": code}
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Internal Server Error: {str(e)}")