Query Stringdb

import requests
import json

BASE_URL = "https://version-12-0.string-db.org/api"

# 1. Get interaction partners
def get_interactions(genes, species=9606, score_threshold=400):
    url = f"{BASE_URL}/json/network"
    params = {
        "identifiers": "%0d".join(genes),
        "species": species,
        "required_score": score_threshold,
        "caller_identity": "bioclaw"
    }
    r = requests.get(url, params=params)
    r.raise_for_status()
    return r.json()

# 2. Get functional enrichment
def get_enrichment(genes, species=9606):
    url = f"{BASE_URL}/json/enrichment"
    params = {
        "identifiers": "%0d".join(genes),
        "species": species,
        "caller_identity": "bioclaw"
    }
    r = requests.get(url, params=params)
    r.raise_for_status()
    return r.json()

# 3. Get interaction partners (expand network)
def get_partners(gene, species=9606, limit=10):
    url = f"{BASE_URL}/json/interaction_partners"
    params = {
        "identifiers": gene,
        "species": species,
        "limit": limit,
        "caller_identity": "bioclaw"
    }
    r = requests.get(url, params=params)
    r.raise_for_status()
    return r.json()

# 4. Download network image
def download_network_image(genes, species=9606, output_path="/workspace/group/network.png"):
    url = f"{BASE_URL}/highres_image/network"
    params = {
        "identifiers": "%0d".join(genes),
        "species": species,
        "caller_identity": "bioclaw"
    }
    r = requests.get(url, params=params)
    with open(output_path, 'wb') as f:
        f.write(r.content)
    return output_path

# Example
interactions = get_interactions(["BRCA1", "BRCA2", "TP53"])
for i in interactions[:10]:
    print(f"{i['preferredName_A']} <-> {i['preferredName_B']}  score: {i['score']}")
    print(f"  Sources: experimental={i.get('escore',0)}, database={i.get('dscore',0)}, textmining={i.get('tscore',0)}")