|
|
| from fastapi import FastAPI, Request
|
| from fastapi.responses import JSONResponse, HTMLResponse
|
| from pydantic import BaseModel
|
| from dotenv import load_dotenv
|
| import os
|
|
|
| load_dotenv()
|
| SECURE_TOKEN = os.getenv("API_KEY", "your-secret-api-key")
|
|
|
| app = FastAPI()
|
|
|
| class GenerateRequest(BaseModel):
|
| prompt: str
|
| token: str
|
|
|
| @app.post("/generate")
|
| async def generate(req: GenerateRequest):
|
| if req.token != SECURE_TOKEN:
|
| return JSONResponse(status_code=401, content={"error": "Invalid or missing token"})
|
|
|
|
|
| response = f"🤖 模拟响应: {req.prompt}"
|
| return {"response": response}
|
|
|
| @app.get("/chat", response_class=HTMLResponse)
|
| async def serve_chat():
|
| with open("chat.html", "r", encoding="utf-8") as f:
|
| return f.read() |
