Spaces:
Running
Running
File size: 2,164 Bytes
d17fdb8 bac1801 d17fdb8 bac1801 d17fdb8 bac1801 3030b52 bac1801 | 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 | import re
import httpx
from fastapi import APIRouter, HTTPException, Query
from pydantic import BaseModel
router = APIRouter(prefix="/api/interactions", tags=["interactions"])
STRING_NET = "https://string-db.org/api/json/interaction_partners"
def _sanitize_for_url(s: str) -> str:
"""Strip control characters that break URL construction."""
return re.sub(r'[\x00-\x1f\x7f-\x9f]', '', s)
class Interaction(BaseModel):
partner_gene: str
partner_protein: str
combined_score: float
nscore: float
fscore: float
pscore: float
ascore: float
escore: float
dscore: float
tscore: float
@router.get("/{gene_name}")
async def get_interactions(
gene_name: str,
species: int = Query(default=9606, description="NCBI taxon ID; 9606=human"),
limit: int = Query(default=15, ge=1, le=50),
):
gene_name = _sanitize_for_url(gene_name)
async with httpx.AsyncClient(timeout=20) as client:
r = await client.get(STRING_NET, params={
"identifiers": gene_name,
"species": species,
"limit": limit,
"caller_identity": "bio-nexus-platform",
})
if r.status_code != 200:
raise HTTPException(502, f"STRING-DB returned {r.status_code}")
data = r.json()
if not data:
raise HTTPException(404, f"No interactions found for {gene_name}")
interactions = [
Interaction(
partner_gene = item.get("preferredName_B", ""),
partner_protein = item.get("stringId_B", ""),
combined_score = item.get("score", 0),
nscore = item.get("nscore", 0),
fscore = item.get("fscore", 0),
pscore = item.get("pscore", 0),
ascore = item.get("ascore", 0),
escore = item.get("escore", 0),
dscore = item.get("dscore", 0),
tscore = item.get("tscore", 0),
)
for item in data
]
interactions.sort(key=lambda x: x.combined_score, reverse=True)
return {"gene": gene_name, "species": species, "interactions": interactions}
|