bc-test / src /fe_handler.py
lamossta's picture
api and pages
ffcf8df
raw
history blame contribute delete
717 Bytes
"""Frontend handler — bridges the Streamlit UI with the FastAPI backend."""
import requests
BASE_URL = "http://localhost:8000"
def fix_newlines(text: str, model_name: str | None = None) -> dict:
payload = {"text": text}
if model_name:
payload["model_name"] = model_name
resp = requests.post(f"{BASE_URL}/fix-newlines", json=payload)
resp.raise_for_status()
return resp.json()
def fix_newlines_all_models(text: str) -> dict:
resp = requests.post(f"{BASE_URL}/fix-newlines-all-models", json={"text": text})
resp.raise_for_status()
return resp.json()
def health() -> dict:
resp = requests.get(f"{BASE_URL}/health")
resp.raise_for_status()
return resp.json()