Spaces:
Running
Running
File size: 3,229 Bytes
48edd44 f33abb4 48edd44 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 |
"""
Tolerate Space Lab - Backend
A therapeutic intervention tool for building distress tolerance
"""
import os
import httpx
from fastapi import FastAPI, HTTPException
from fastapi.staticfiles import StaticFiles
from fastapi.responses import FileResponse
from pydantic import BaseModel
from typing import List
app = FastAPI()
# Get API key from environment (set as HF Space secret)
ANTHROPIC_API_KEY = os.environ.get("anthropic_key", "")
class Message(BaseModel):
role: str
content: str
class ChatRequest(BaseModel):
messages: List[Message]
system: str
max_tokens: int = 300
class AnalysisRequest(BaseModel):
prompt: str
max_tokens: int = 1000
@app.post("/api/chat")
async def chat(request: ChatRequest):
"""Proxy chat requests to Claude API"""
if not ANTHROPIC_API_KEY:
raise HTTPException(status_code=500, detail="API key not configured")
async with httpx.AsyncClient() as client:
try:
response = await client.post(
"https://api.anthropic.com/v1/messages",
headers={
"Content-Type": "application/json",
"x-api-key": ANTHROPIC_API_KEY,
"anthropic-version": "2023-06-01"
},
json={
"model": "claude-sonnet-4-20250514",
"max_tokens": request.max_tokens,
"system": request.system,
"messages": [{"role": m.role, "content": m.content} for m in request.messages]
},
timeout=60.0
)
response.raise_for_status()
return response.json()
except httpx.HTTPStatusError as e:
raise HTTPException(status_code=e.response.status_code, detail=str(e))
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
@app.post("/api/analysis")
async def analysis(request: AnalysisRequest):
"""Proxy analysis requests to Claude API"""
if not ANTHROPIC_API_KEY:
raise HTTPException(status_code=500, detail="API key not configured")
async with httpx.AsyncClient() as client:
try:
response = await client.post(
"https://api.anthropic.com/v1/messages",
headers={
"Content-Type": "application/json",
"x-api-key": ANTHROPIC_API_KEY,
"anthropic-version": "2023-06-01"
},
json={
"model": "claude-sonnet-4-20250514",
"max_tokens": request.max_tokens,
"messages": [{"role": "user", "content": request.prompt}]
},
timeout=60.0
)
response.raise_for_status()
return response.json()
except httpx.HTTPStatusError as e:
raise HTTPException(status_code=e.response.status_code, detail=str(e))
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
# Serve static files
app.mount("/static", StaticFiles(directory="static"), name="static")
@app.get("/")
async def root():
return FileResponse("static/index.html")
|