import requests
import streamlit as st

class ExtractorEngine:
    def __init__(self):
        self.api_url = "https://klydekushy-prospectusveritas-api.hf.space/extract"

    def extract_graph(self, text: str):
        # On envoie un morceau raisonnable pour le CPU
        payload = {"text": text[:3500], "max_tokens": 1024, "temperature": 0.1}
        try:
            response = requests.post(self.api_url, json=payload, timeout=180)
            if response.status_code == 200:
                return response.json().get("result")
            return None
        except Exception as e:
            st.error(f"Erreur : {e}")
            return None