MAIRK commited on
Commit
da0cbd5
·
verified ·
1 Parent(s): 4e81ac5

Upload app.py

Browse files

# app.py
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"})

# TODO: 调用实际模型推理
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()

Files changed (1) hide show
  1. app.py +29 -0
app.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # app.py
2
+ from fastapi import FastAPI, Request
3
+ from fastapi.responses import JSONResponse, HTMLResponse
4
+ from pydantic import BaseModel
5
+ from dotenv import load_dotenv
6
+ import os
7
+
8
+ load_dotenv()
9
+ SECURE_TOKEN = os.getenv("API_KEY", "your-secret-api-key")
10
+
11
+ app = FastAPI()
12
+
13
+ class GenerateRequest(BaseModel):
14
+ prompt: str
15
+ token: str
16
+
17
+ @app.post("/generate")
18
+ async def generate(req: GenerateRequest):
19
+ if req.token != SECURE_TOKEN:
20
+ return JSONResponse(status_code=401, content={"error": "Invalid or missing token"})
21
+
22
+ # TODO: 调用实际模型推理
23
+ response = f"🤖 模拟响应: {req.prompt}"
24
+ return {"response": response}
25
+
26
+ @app.get("/chat", response_class=HTMLResponse)
27
+ async def serve_chat():
28
+ with open("chat.html", "r", encoding="utf-8") as f:
29
+ return f.read()