app.py

import gradio as gr
from filter_agent import filter_math_query
from wiki_scraper import get_wikipedia_article

def math_explainer(topic):
    """Vérifie d'abord si le sujet est mathématique, corrige l'orthographe, puis récupère l'article Wikipédia entier et le résume."""

    # 🔍 Étape 1 : Vérification mathématique + Correction orthographique
    filter_result, corrected_topic = filter_math_query(topic)

    if "❌" in filter_result:  # Si la requête est rejetée
        return filter_result  # On retourne juste le message d'erreur

    # 📑 Étape 2 : Recherche de l’article entier sur Wikipédia et résumé
    return get_wikipedia_article(corrected_topic)

iface = gr.Interface(
    fn=math_explainer,
    inputs="text",
    outputs="text",
    title="Math Research AI",
    description="🚀 Math Research IA facilite la façon d'apprendre les mathématiques en rendant les concepts clés accessibles à tous grâce à des explications claires et vulgarisées.\n \n Cet agent IA spécialisé en mathématiques analyse une requête, corrige automatiquement les fautes, et extrait un article Wikipédia pour en générer un résumé simplifié et compréhensible."
)

if __name__ == "__main__":
    iface.launch()